Yolov5 v2.0 onnx在x3上推理时出了很多框

用户您好,请详细描述您所遇到的问题,这会帮助我们快速定位问题~

1.芯片型号:例如X3、X3派、J5等

2.天工开物开发包OpenExplorer版本:例如XJ3_OE_2.4.2,XJ3_OE_1.15.2,J5_OE_1.1.40等

3.问题定位:例如模型转换、板端部署等

4.问题具体描述:请提供运行的命令、报错信息,如果可以的话,可在附件中提供模型供技术支持进行复现

Yolov5 v2.0 onnx模型

转换正常(图03_build.sh),推理没有报错,但是出了很多框;

见图 demo1 和 demo2:

后推理文件见:postprocess.py

PS: 在PC上推理正常:见图: PC_OK

03_build.sh.png-
demo.jpg-
demo2.jpg-
PC_OK.jpg-
postprocess.py

已解决:

#x[i] = x[i].view(bs, self.na, self.no, ny, nx).permute(0, 1, 3, 4, 2).contiguous()

x[i] = x[i].permute(0, 2, 3, 1).contiguous() # x3导出时打开,训练时屏蔽

不清楚什么原因

原模型:yolov5_emotion.onnx

链接:百度网盘-链接不存在

提取码:j7dl

转换后模型:yolov5s_emotions_nv12.bin

链接:百度网盘-链接不存在

提取码:5tju