yolov7添加注意力机制

yolov7结构图
yolov7添加注意力机制_第1张图片
方法:直接在common里改,在相关的后面加上就行

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

1、目的:在三个输出地方添加注意力
yolov7添加注意力机制_第2张图片

yolov7.yaml文件,换成其他模块
注意力链接
yolov7添加注意力机制_第3张图片

2、models下建SE.py
3、common.py下,先找class Conv,再复制一份修改,把模块导进来
yolov7添加注意力机制_第4张图片

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

yolov7添加注意力机制_第5张图片

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

你可能感兴趣的:(yolov7,YOLO)