运行报错RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1]

Question

运行yolov5中的train.py后出现报错RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 5, 8, 8] to have 1024 channels, but got 5 channels instead

请问这个应该怎么修改

Answer 1

把你的yaml贴出来，看下你的CFP修改了什么，应该是通道数目设置不对。或者你自己按shift+ F9，打上断点debug一下，计算一下每一层网络输入输出设置，看下哪一层网络设置问题