YOLOv11 添加注意力机制 !
视频教程地址-哔哩哔哩
文章目录
- YOLOv11 添加注意力机制 !
- 注意力机制介绍
- 注意力机制的分类
- 1. SE 注意力模块
- 1.1 原理
- 1.2 代码
- 2. CBAM 注意力模块
- 2.1 原理
- 2.2 代码
- 3. ECA 注意力模块
- 3.1 原理
- 3.2 代码
- 4. CA 注意力模块
- 4.1 原理
- 4.2 代码
- 5. 添加方式💡
- 5.1 检测任务
- 5.2 分类任务
- 5.3 分割任务
- 5.4 关键点任务
- 6. 20+种注意力机制代码汇总添加
- 第一步
- 第二步
- 第三步
- 模板1-Backbone
- 模板2-针对小目标层添加
- 模板3-针对大目标层添加
- 模板4-Neck
注意力机制介绍
注意力机制(Attention Mechanism)源于对人类视觉的研究。在认知科学中,由于信息处理的瓶颈,人类会选择性地关注所有信息的一部分,同时忽略其他可见的信息。为了合理利用有限的视觉信息处理资源,人类需要选择视觉区域中的特定部分,然后集中关注它。例如,人们在阅读时,通常只有少量要被读取的词会被关注和处理。综上,注意力机制主要有两个方面:决定需要关注输入的哪部分;分配有限的信息处理资源给重要的部分。这几年有关attention的论文与日俱增,下图就显示了在包括CVPR、ICCV、ECCV、NeurIPS、ICML和ICLR在内的顶级会议中,与attention相关的论文数量的增加量。下面我将会分享 Yolov8 如何添加注意力机制!