求助yolov5部署问题

新手使用 RDK X3 MD 104032,在部署yolov5时,总是有错误。

1.在yolov5 v2.0 和yolov5 v6.2下训练出的pt文件,用下面命令转onnx。

python export.py --data tempdata.yaml --weight runs/train/exp3/weights/best.pt --include onnx --opset 11

2.构建docker环境,检验并转化bin文件,得到bin文件,转换过程见附件,感觉转换正常。

3.但最后在调用部署推理时,总是出现附件图片中问题:

推理得到的输出 model_output[0].buffer.size总是201600,在后期处理reshape时出错误。

按理yolov5输出大小应该是25200,为什么会变成201600呢?到底哪里出问题了。

看论坛中也有用户出过这样问题,不知如何解决的。

您好,地平线工具链在持续迭代优化,为了给您提供更好的服务,希望您能抽出3分钟左右的时间,将您在使用工具链期间的感受和建议告诉我们,您的宝贵意见对我们很重要,非常感谢!-
问卷链接:地平线算法工具链使用满意度反馈

您好,模型输出为1x25200x8x1==201600,是无法reshape成1x80x80x3x8==153600的,这种情况下建议学习了解一下yolov5的后处理原理,修改一下模型的后处理代码,对齐模型输出的shape

可以参考这篇文章 【模型提速】如何在X3pi使用yolov5模型50ms推理 (horizon.cc)