专栏算法工具链Yolov5 v2.0 onnx在x3上推理时出了很多框

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

ziyifly2023-08-24
37
3

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

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

算法工具链
评论3
0/1000
  • ziyifly
    Lv.1

    原模型：yolov5_emotion.onnx

    链接：https://pan.baidu.com/s/1AnTqMbLuPjYgcNdJ8-LOWw

    提取码：j7dl

    转换后模型：yolov5s_emotions_nv12.bin

    链接：https://pan.baidu.com/s/1GdTb6VwDTPWZPzPJYqRLpg

    提取码：5tju

    2023-08-24
    0
    0
  • ziyifly
    Lv.1
    2023-08-24
    0
    0
  • ziyifly
    Lv.1

    已解决：

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

    不清楚什么原因

    2023-08-24
    0
    0