专栏算法工具链yolov5在J6M平台前后处理与demo一致的情况推理框多且杂乱

yolov5在J6M平台前后处理与demo一致的情况推理框多且杂乱

已解决
zzp2026-01-15
60
6

yolov5模型结构和demo示例的结构完全一致,python推理bc或onnx都有结果输出。板端运行框很多且杂乱,用demo给的onnx转hbm板端测试前后处理确定没有问题(letterbox输入和反量化因子都没问题),demo的转的hbm框推理正常。文件压缩包里有对应的模型及转换推理文件,可以帮忙看看是什么原因吗?

 

工具链版本:Horizon OpenExplorer 3.0.31(release_date:202501)

 

模型见附件中的onnx文件,详细的工具链量化结果见百度网盘:

链接: https://pan.baidu.com/s/1OgTd0YfDyeFFxLy2i-n-9w?pwd=v9bc 提取码: v9bc --来自百度网盘超级会员v5的分享

算法工具链
征程6
评论3
0/1000
  • zzp
    Lv.1

    另外,锚点和类别数都设置没问题,与python推理的一致。

    2026-01-15
    0
    0
  • YCJ
    Lv.4

    收到,我这边先帮你复现一下问题

    2026-01-16
    0
    0
  • zzp
    Lv.1
    排查后发现与stride有关,官方提供的demo是yolov5x,最后一个维度是255:(80+4+1)x3,但是stride的倒数第2个维度是1024,对应256,所以后处理代码每次循环后data加了1。
    同理,我的模型最后一个维度是45:(10+4+1)x3,stride的倒数第二个维度是256,对应的是64,所以后处理代码每次循环后要加19,做维度对齐。
    官方的80个类别部分输出信息如下:

    自己训练的10个类别的打印部分信息如下:
    后处理代码修改,做维度对齐
    2026-01-16
    0
    3
    • YCJ回复zzp:

      收到,意思是已经解决?

      2026-01-16
      0
    • zzp回复YCJ:

      嗯嗯,已解决,也是看了您写的 J6 部署踩坑记:一次由 stride 引发的推理不一致问题,找到了原因

      2026-01-16
      0
    • YCJ回复zzp:
      good_org.svg
      2026-01-16
      0