自训练出来的yolov5模型转om文件与yolov5官网提供的yolov5模型转om,模型结构存在差异,不能正常推理
收藏回复举报
自训练出来的yolov5模型转om文件与yolov5官网提供的yolov5模型转om,模型结构存在差异,不能正常推理
t('forum.solved') 已解决
发表于2024-12-20 17:52:39
0 查看

1、代码链接:

https://gitee.com/ascend/modelzoo-GPL/tree/master/built-in/PyTorch/Official/cv/object_detection/Yolov5_for_PyTorch_v6.0

2、环境配置:

pytorch环境和 Ascend910环境:

image.png

image.png

推理环境:

cke_4058.png

3、自己训练的pt文件链接如下:

链接:https://pan.baidu.com/s/1P-o-ZUZ8EzfY7teTLh4dgA  

提取码:8zjd

4、问题:

    1、希望博主用我链接里的pt文件转一下om文件,发我。

我要发帖子