YOLOv8添加注意力机制

添加方法：

notes:有时需要在这里面进行修改
/home/ailab/anaconda3/lib/python3.11/site-packages/ultralytics-8.0.230-py3.11.egg/ultralytics

1、conv代码中添加注意力代码（本文添加的是GAM注意力机制）

2、注册及引用注意力代码

1、ultralytics/nn/modules/init.py代码中

2、ultralytics/nn/modules/tasks.py代码中

tasks代码中写入调入方式

        # """**************add Attention***************"""
        elif m in {GAM_Attention}:
            c1, c2 = ch[f], args[0]
            if c2 != nc:  # if not output
                c2 = make_divisible(min(c2, max_channels) * width, 8)
            args = [c1, c2, *args[1:]]

1
2
3
4
5
6
7

添加位置如下：
参考1：

参考2：推荐

3、yaml文件修改

建议复制一份yaml文件进行修改

修改yolov8-GAM.yaml文件

# Ultralytics YOLO 
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/413608