Yolov5添加SKNet注意力机报错

Question

Yolov5采用yolov5m-6.0版本，在SPPF前添加一个视觉注意力sknet，报错如下。已修改common.py,yolo.py和yolov5m.yaml文件。为什么会报这个错，该怎么修改呢

Answer 1

从报错信息来看,可能有以下几个原因:

1. SKNet模块的参数初始化有问题。SKNet模块需要初始化一些参数,如果初始化错误会导致报错。可以 double check SKNet的参数初始化。
2. SKNet模块的forward函数实现有问题。forward函数实现不正确也会导致报错。可以检查SKNet的forward函数实现是否正确。
3. SKNet模块的输出 не匹配Yolov5的输入格式。Yolov5的输入是一个list,每个元素是一层的输出,SKNet的输出如果不在list中,也会导致报错。可以将SKNet的输出放在list中,作为Yolov5的输入。
4. SKNet模块和Yolov5的其它部分连接不正确。除了forward函数,还需要正确连接SKNet模块和Yolov5的输入、输出。如果连接不正确也会导致报错。
5. 有其他实现错误。也有可能是其他实现细节导致的报错,需要仔细检查代码。
   总之,添加第三方模块到Yolov5,需要确保:
6. 模块参数初始化正确
7. forward函数实现正确
8. 输入输出格式匹配Yolov5
9. 与Yolov5其他部分连接正确
10. 无其他实现错误