报错The size of tensor a (80) must match the size of tensor b (56) at non-singleton dimension 3

Question

yolov5的detect.py运行报错，出了什么问题呢，求大佬解释

Answer 1

错误分析：这里面说的是要求输入的权重是[64，64,1，1]，其中第二个维度的要求是64，但是给定的输入却是[16, 48, 40, 40]，导致输入的第二个维度48和64不匹配。
一般出错都是在卷积层的输出的维度匹配不上下一个卷积层要求的输出维度，这时候我们要仔细检查上一层卷积的out_channel是否和下一层卷积的in_channel一模一样。
当然还要注意concatenation拼接等操作引起的维度变化。