改过结构的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是输出张量不匹配,请问要怎么修改呢?看哪一个文件修改?
收起
输入输出张量图像检查过了吗,可以去看看输出节点的定义--在模型的ONNX文件里
报告相同问题?