新手使用 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呢?到底哪里出问题了。
看论坛中也有用户出过这样问题,不知如何解决的。