一、本文介绍
作为入门性篇章,这里介绍了SEAttention注意力在YOLOv8中的使用。包含SEAttention原理分析,SEAttention的代码、SEAttention的使用方法、以及添加以后的yaml文件及运行记录。
二、SEAttention原理分析
SEAttention官方论文地址:SE文章
SEAttention注意力机制(挤压-激励):一种通道注意力,核心思想:
- Squeeze:通过全局平均池化(Global Average Pooling)将特征图的空间维度(高和宽)压缩成一个值,从而获得每个通道的全局表示。
- Excitation:使用一个包含非线性激活函数的全连接层网络来学习每个通道的重要性权重。
- Reweight:将学习到的通道权重应用于原始特征图,强化有用特征并抑制无关特征。