yolov7结构图

方法:直接在common里改,在相关的后面加上就行
1、目的:在三个输出地方添加注意力

yolov7.yaml文件,换成其他模块
注意力链接

2、models下建SE.py
3、common.py下,先找class Conv,再复制一份修改,把模块导进来

4、yolo.py文件,导入进来,添加

5、yolov7.yaml,修改模块名字

6、全都变成conv_ATT,倒时候换的时候只用换common.py的这里

7、还有第三个地方没加
8、common.py找到sppcspc,复制一份在下面

9、yolo.py添加名字

比如说simAm
1、models下建立文件夹,放进去
2、common.py导入模块名

3、common.py下改,默认的可以不用动

四个连接之后加

1、common.py找到Concat类,复制一份,加一个参数channel
不用接受通道数的可以不加channel参数

2、yolo.py,加这些
