专栏算法工具链yolov5转onnx

yolov5转onnx

已解决
慌慌张张不嚣张2023-03-23
99
7

请问yolov5进行模型转换时，转onnx时为什么要去除4维到5维的reshape，将后续结构放入到后处理中运行？

算法工具链
征程3
评论2
0/1000
  • 颜值即正义
    Lv.2

    你好，通过对比发现，将红框部分移至后处理中会更加高效

    2023-03-23
    0
    5
    • 慌慌张张不嚣张回复颜值即正义:

      请问模型转换过程中对张量的计算有限制吗？例如是否能进行5维张量的计算

      2023-03-23
      0
    • 颜值即正义回复慌慌张张不嚣张:

      XJ3一些算子是支持非四维的，建议通过具体模型转换实操，比如5维的模型输入也是支持的，可以参考链接：https://developer.horizon.ai/forumDetail/146177165367615508

      2023-03-23
      0
    • x3sdb回复颜值即正义:

      请问，这个图片中的文档在哪里可以看到呢

      2023-04-18
      0
    • 颜值即正义回复x3sdb:

      文档是会和对应的OE一起发布的呀，发布页面如下：https://developer.horizon.ai/forumDetail/136488103547258769

      2023-04-19
      0
    • x3sdb回复颜值即正义:

      好的谢谢！

      2023-04-19
      0
  • 颜值即正义
    Lv.2
    2023-04-24
    0
    0