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

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

一、本文介绍

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

二、CAA原理分析

CAA官方论文地址：文章

CAA官方代码地址：代码

​

CAA注意力机制：CAA旨在掌握远距离之间的上下文相互依存关系像素，同时增强中心特征。CAA模块通过池化、卷积和Sigmoid等操作，生成了一个加权注意力图，使得特征图中的不同区域能够根据其全局上下文信息进行加权，从而对不同重要性的特征进行选择性加强或抑制。

三、相关代码：

CAA注意力的代码，如下：

from typing import Optional
from mmcv.cnn i