yolov5转onnx验证出错

Question

改过结构的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
是输出张量不匹配，请问要怎么修改呢？看哪一个文件修改？

Answer 1

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