齐天小圣66 2023-05-27 20:59 采纳率: 50%
浏览 71
已结题

yolov5转onnx验证出错

改过结构的yolov5转onnx,转换成功,但是用detect.py --weights best.onnx验证的时候出现下面报错:
IndexError: The shape of the mask [3, 40, 2] at index 2 does not match the shape of the indexed tensor [3, 40, 6] at index 2
是输出张量不匹配,请问要怎么修改呢?看哪一个文件修改?

  • 写回答

6条回答 默认 最新

  • 这一次有糖 2023-05-27 21:08
    关注
    获得2.40元问题酬金

    输入输出张量图像检查过了吗,可以去看看输出节点的定义--在模型的ONNX文件里

    评论 编辑记录

报告相同问题?

问题事件

  • 系统已结题 6月4日
  • 创建了问题 5月27日