YOLOv8添加注意力机制方法【MHSA、ShuffleAttention、SiAM】

  • 开发
  • 70

YOLOv8添加注意力机制方法

文章目录

注意力机制示例1：MHSA

1、 注意力机制文件放在nn文件夹下
2、 task.py文件导入注意力机制文件
在这里插入图片描述
3、 task.py中parse_model函数添加elif语句：
在这里插入图片描述
在这里插入图片描述
注意是in
4、 改模型配置文件

原模型（左图）和添加注意力机制后的（右图）
在这里插入图片描述
这里的参数要看MHSA的参数要求
该注意力机制属于有参数的

注意力机制示例2：ShuffleAttention

1、 nn文件夹下添加.py文件，task.py文件导入ShuffleAttention
2、 parse_model函数添加elif语句，由于该注意力机制也接收通道数为参数，所以方法和MHSA相同，如图所示：
在这里插入图片描述
3、看注意力机制源码的参数
在这里插入图片描述

注意力机制示例3：SiAM

1、 该注意力机制不接受参数，故直接在task.py中导入即可
在这里插入图片描述
2、 修改yaml文件后即可运行
在这里插入图片描述

阅读全部

相关推荐

  9. YOLOv8改进 | 注意力机制 | 添加混合局部通道注意力——MLCA【原理讲解】

    2024-03-16 03:46:04       26 阅读

最近更新

  3. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-03-16 03:46:04       94 阅读

  4. Could not load dynamic library ‘cudart64_100.dll‘

    2024-03-16 03:46:04       100 阅读

  9. 在Django里面运行非项目文件

    2024-03-16 03:46:04       82 阅读

  19. Python语言-面向对象

    2024-03-16 03:46:04       91 阅读

  20. 如何分清楚常见的 Git 分支管理策略Git Flow、GitHub Flow 和 GitLab Flow

    2024-03-16 03:46:04       85 阅读

热门阅读

  2. ORM Bee数据源的使用, Bee有自己的数据源吗?

    2024-03-16 03:46:04       43 阅读

  5. 一文了解前端面试重点--闭包

    2024-03-16 03:46:04       49 阅读

  9. 专升本英语语法学习

    2024-03-16 03:46:04       45 阅读

  12. C++语言学习（一）—— 认识C++语言

    2024-03-16 03:46:04       33 阅读

  17. Docker学习

    2024-03-16 03:46:04       37 阅读

  19. 系统重装后git客户端的设置恢复

    2024-03-16 03:46:04       41 阅读

  23. 正则表达式中元字符的使用

    2024-03-16 03:46:04       43 阅读

  28. 登录态数据抓取：Python爬虫携带Cookie与Session的应用技巧

    2024-03-16 03:46:04       45 阅读

  29. 空气质量指数（AQI）计算小结

    2024-03-16 03:46:04       40 阅读