YOLOv5——SPPF（bug）

当我使用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。