专栏算法工具链自己训练模型YOLOv5模型onnx转换完成，04_inference单张图片推理

自己训练模型YOLOv5模型onnx转换完成，04_inference单张图片推理

已解决
zhengsiyu2022-08-05
140
5

在单张图片推理过程中报错，具体报错内容：

查看别人帖子说需要修改后处理中output的reshape，更改依照01check中输出的output的reshape，但是执行check没有output的信息，无法修改。

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

    您好，示例里的yolov5模型输出大小如下图所示，后处理也只能适配示例模型。如果您对模型做了修改，可以直接用自己训练模型时前后处理代码，或者以及自己模型的输出大小把示例的后处理代码做一下适配。

    2022-08-05
    0
    3
    • zhengsiyu回复颜值即正义:

      我的onnx模型输出是这样的：

      如果修改后处理的代码是修改这个位置的参数吗？需要怎样修改呢，小白，底子略浅，希望能详细指点一下。

      2022-08-05
      0
    • zhengsiyu回复颜值即正义:

      还有模型输出只能是思维的数据吗？我使用的OE包版本是1.10.7的。

      2022-08-05
      0
    • 颜值即正义回复zhengsiyu:

      您好，BPU的输出只能是4维的数据哈

      2022-08-08
      0
  • 地平线小旭日
    Lv.1

    您好，您的问题我们已收到，技术同事稍后将会为您解答哈~

    2022-08-05
    0
    0