这里,我用了老师提供的虚拟机进行模型转换和代码编译。对于Yolov5的环境,我是自己之前搭建的环境,也按照老师的要求在转onnx时候修改了yolo.py中的forward函数,但是模型转换过程中依旧出现了问题
这个是转的yolov5.onnx,最后是255通道
这个是转的我自己模型的best-sim.onnx
这个地方就不太理解为什么通道数变成了24
nncase转换模型就是这样
object_detect我修改了标签数量和名称
运行出错
应该是大小不匹配不问题,但是不知道怎么解决
这里,我用了老师提供的虚拟机进行模型转换和代码编译。对于Yolov5的环境,我是自己之前搭建的环境,也按照老师的要求在转onnx时候修改了yolo.py中的forward函数,但是模型转换过程中依旧出现了问题
这个是转的yolov5.onnx,最后是255通道
这个是转的我自己模型的best-sim.onnx
请严格按照下面文章中进行修改模型文件yolo.py
你好,老师,我是按照这个修改的。因为我前边已经自己部署了yolov5的环境,我就修改了yolo.py中的forward。而且这个地方yolov5s.pt转换的都是正常的,请问是怎么回事
通过报错信息,确定是yolo.py没有正确修改,请你仔细检查。或者你也可以使用下面我提供的yolo.py文件
yolo.zip (4.9 KB)
yolo.py就是用你发的替代的啊,怎么错误了
给您分享一个我这边验证过的源码包:
yolov5-6.0-k510.zip (958.2 KB)
如果还是出现问题,那可能需要您自行排查问题了
老师,我是用的这个源码包转换,不过我的模型是在我之前的yolov5中训练的,你的意思是我要在这个环境中重新训练嘛
不需要重新训练,使用修改了yolo.py的源码导出模型即可
是这样的,我用的这个,就报上边的错误
使用我这边的提供的源码进行测试
老师,我这边yolov5也是可以的,我自己的模型运行会出现上边的错误,链接: 百度网盘 请输入提取码 提取码: 7tnb ,这个是我的模型best.pt
老师,请问这个模型的建立的有什么问题啊,我是用正常的yolov5训练的
你是否使用过yolov5s.pt模型进行转换和部署呢?
还没有使用过yolov5s.pt进行部署过
老师,你好,我全部按照你的Ai部署流程来,这里依旧是24