yolov7结构图
在这里插入图片描述
方法:直接在common里改,在相关的后面加上就行

1、接受通道数的注意力机制

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添加名字
在这里插入图片描述

2、不接受通道的注意力机制

比如说simAm
1、models下建立文件夹,放进去
2、common.py导入模块名
在这里插入图片描述
3、common.py下改,默认的可以不用动
在这里插入图片描述

3、加在连接之后

四个连接之后加
在这里插入图片描述
1、common.py找到Concat类,复制一份,加一个参数channel
不用接受通道数的可以不加channel参数
在这里插入图片描述
2、yolo.py,加这些
在这里插入图片描述

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐