当前位置:   article > 正文

yolov7添加注意力机制

yolov7添加注意力机制

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,加这些
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/698077
推荐阅读
相关标签
  

闽ICP备14008679号