一、本文介绍
本文记录的是利用GAM(Global Attention Mechanism)注意力机制改进YOLOv10的特征提取部分。GAM通过全局注意力机制实现高效的特征增强。
二、GAM注意力机制介绍
2.1 设计出发点
传统注意力机制计算复杂度高,GAM通过高效的全局注意力机制实现特征增强。
2.2 模块结构
GAM块:
- 通道注意力:学习通道重要性权重
- 空间注意力:学习空间位置重要性权重
- 全局融合:融合通道和空间注意力
三、GAM注意力机制的实现代码
importtorchimporttorch.nnasnnclassGAMAttention