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

img


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

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

您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!
PS:问答VIP年卡 【限时加赠:IT技术图书免费领】,了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632

你好请问你的问题解决了吗

你好,您
的问题解决了吗

请问您这个问题解决没