K510开发 AI部署章节 模型转换失败 一直报错

[wedong]

这里，我用了老师提供的虚拟机进行模型转换和代码编译。对于Yolov5的环境，我是自己之前搭建的环境，也按照老师的要求在转onnx时候修改了yolo.py中的forward函数，但是模型转换过程中依旧出现了问题
这个是转的yolov5.onnx，最后是255通道
这个是转的我自己模型的best-sim.onnx

这个地方就不太理解为什么通道数变成了24
nncase转换模型就是这样

object_detect我修改了标签数量和名称
运行出错

应该是大小不匹配不问题，但是不知道怎么解决

[100ASK_RKQ]

请严格按照下面文章中进行修改模型文件yolo.py

YOLOV5端侧部署 | 百问网

[wedong]

你好，老师，我是按照这个修改的。因为我前边已经自己部署了yolov5的环境，我就修改了yolo.py中的forward。而且这个地方yolov5s.pt转换的都是正常的，请问是怎么回事

[100ASK_RKQ]

通过报错信息，确定是yolo.py没有正确修改，请你仔细检查。或者你也可以使用下面我提供的yolo.py文件
yolo.zip (4.9 KB)

[wedong]

你好，老师，就是出现这个错误

[100ASK_RKQ]

这个属于您修改yolo.py导致的问题，出问题的地方可能是以下机制情况

[wedong]

yolo.py就是用你发的替代的啊，怎么错误了

[100ASK_RKQ]

给您分享一个我这边验证过的源码包：
yolov5-6.0-k510.zip (958.2 KB)
如果还是出现问题，那可能需要您自行排查问题了

[wedong]

老师，我是用的这个源码包转换，不过我的模型是在我之前的yolov5中训练的，你的意思是我要在这个环境中重新训练嘛

[100ASK_RKQ]

不需要重新训练，使用修改了yolo.py的源码导出模型即可

[wedong]

是这样的，我用的这个，就报上边的错误

[100ASK_RKQ]

使用我这边的提供的源码进行测试

[wedong]

你好，老师还是这个错误，我感觉这个是yolov5不匹配的问题嘛

[100ASK_RKQ]

我没遇到过这种问题，我这边可以正常运行的，你可以尝试你在Ubuntu下运行。

[wedong]

老师，我这边yolov5也是可以的，我自己的模型运行会出现上边的错误，链接: 百度网盘 请输入提取码 提取码: 7tnb ，这个是我的模型best.pt

[100ASK_RKQ]

您这边建立的模型有问题

[wedong]

老师，请问这个模型的建立的有什么问题啊，我是用正常的yolov5训练的

[100ASK_RKQ]

你是否使用过yolov5s.pt模型进行转换和部署呢？

[wedong]

还没有使用过yolov5s.pt进行部署过

[wedong]

老师，你好，我全部按照你的Ai部署流程来，这里依旧是24