专栏算法工具链yolov5的onnx转换问题

yolov5的onnx转换问题

已解决
taogui2022-05-05
273
3

地平线文档关于yolov5的onnx格式要求

实际模型的onnx

地平线需求的onnx格式

问题1：如果不按照文档中，对onnx修改，直接用咱推理代码是否可行？我这边测试了一些，可推理，类别是对的，不过框是乱的

问题2：能否提供对于开源yolo5的onnx修改code，借鉴一下

算法工具链
评论3
0/1000
  • Horizon_guosheng
    Lv.1
    2022-05-07
    0
    0
  • 颜值即正义
    Lv.2

    你好，你可以看的文档版本应该比较老了，新的版本对这部分内容进行了修正，提供了源码更改说明，和 Github 相比，唯一差别就是把一个 premute(Transpose) 的操作放到后处理了

    2022-05-09
    0
    0
  • Horizon_guosheng
    Lv.1
    2022-09-27
    0
    0