yolov5 训练出来的权重做检测,运行detect.py出错,请问这是什么原因呢?

yolov5 训练出来的权重做检测,运行detect.py出错

img

你输入的数据维度有问题,注意卷积的时候数据的尺寸变化,自己计算一下大概就知道怎么回事了,具体计算方法可以借鉴这个:
https://blog.csdn.net/qq_28531269/article/details/122311631?spm=1001.2014.3001.5502

如有帮助,请采纳,多谢!

博主解决了吗