论文所提的Coordinate注意力
很简单,可以灵活地插入到经典的移动网络中,而且几乎没有计算开销。大量实验表明,Coordinate注意力不仅有益于ImageNet分类,而且更有趣的是,它在下游任务(如目标检测和语义分割)中表现也很好。本文结合目标检测任务应用
- 应
专栏读者
的要求,写一篇关于YOLOv10+CA(Coordinate attention) 注意力机制
的改进 - 重点:有不少读者已经反映该专栏的改进 在自有数据集上
有效涨点
!!!同时COCO也能涨点
文章目录
- 一、Coordinate Attention论文理论部分
- Coordinate Attention介绍
- Coordinate Attention设计
- Coordinate Attention Block
- 论文实验
- 二、结合YOLOv10 改进代码
- 2.1 网络配置
- 2.2 核心代码
- 2.3 运行
一、Coordinate Attention论文理论部分
最近对移动网络设计的研究已经证明了通道注意力的显着效果(例如, Squeeze-and-Excitation 注意)用于提升模型性能,但它们通常忽略位置信息,这对于生成空间选择性注意图很重要。在本文中,我们提出了一种新的移动网络注意机制,将位置信息嵌入到通道注意中,我们称之为“坐标注意力”。与通过 2D 全局池化将特征张量转换为单个特征向量的通道注意不同,坐标注意将通道注意分解为两个 1D 特征编码过程,分别沿两个空间方向聚合特征。通过这种方式,可以沿一个空间方向捕获远程依赖关系,同时可以沿另一个空间方向保留精确的位置信息。然后将生成的特征图分别编码为一对方向感知和位置敏感的注意力图,这些注意力图可以互补地应用于输入特征图以增强感