当我使用SPPF替换SPP时,我发现出现了以下报错:
RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 96, 8, 8] to have 1024 channels, but got 96 channels instead
我找了半天也没发现哪里有问题,所以我怀疑是SPPF固有的bug,如果遇到这个问题,请尝试将SPPF模块替换为SPP模块,问题得以解决。
[-1, 1, SPP, [1024, [5, 9, 13]]],
#[-1, 1, SPPF, [1024, 5]],
如果有人知道这个问题的原因,欢迎在讨论区给我留言,谢谢。
--------------------------------------------------------------------------------------------------------------------------------
2022/4/4
今天我突然发现是我SPPF模块忘写入yolo.py里,才导致了这个错误,并不是bug。