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

ziyifly · 2023 年8 月 24 日 03:37

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

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

ziyifly · 2023 年8 月 24 日 06:03

已解决：

#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导出时打开，训练时屏蔽

不清楚什么原因

ziyifly · 2023 年8 月 24 日 04:05

ziyifly · 2023 年8 月 24 日 03:41

原模型：yolov5_emotion.onnx

提取码：j7dl

转换后模型：yolov5s_emotions_nv12.bin

提取码：5tju