YOLOv8改进 - 注意力篇 - 引入CoordAtt注意力机制

#YOLO# #目标检测# #计算机视觉#

一、本文介绍

作为入门性篇章，这里介绍了CoordAtt注意力在YOLOv8中的使用。包含CoordAtt原理分析，CoordAtt的代码、CoordAtt的使用方法、以及添加以后的yaml文件及运行记录。

二、CoordAtt原理分析

CoordAtt官方论文地址：文章

CoordAtt官方代码地址：代码

​

CoordAtt注意力机制：在本文中，我们提出了一种新的移动网络注意机制，将位置信息嵌入到通道注意中，我们称之为“坐标注意”。与通过二维全局池化将特征张量转换为单个特征向量的通道注意不同，坐标注意将通道注意分解为两个一维特征编码过程，分别沿着两个空间方向聚合特征。这样可以在一个空间方向上捕获远程依赖关系，同时在另一个空间方向上保持精确的位置信息。然后将得到的特征图分别编码为一对方向感知和位置敏感的注意图，它们可以互补地应用于输入特征图，以增强感兴趣对象的表示。

三、相关代码：

CoordAtt注意力的代码，如下。

class CoordAtt(nn.