YOLO11改进|注意力机制篇|引入轴向注意力Axial Attention
目录
-
- 一、【Axial Attention】注意力机制
-
- 1.1【Axial Attention】注意力介绍
- 1.2【Axial Attention】核心代码
- 二、添加【Axial Attention】注意力机制
-
- 2.1STEP1
- 2.2STEP2
- 2.3STEP3
- 2.4STEP4
- 三、yaml文件与运行
-
- 3.1yaml文件
- 3.2运行成功截图
一、【Axial Attention】注意力机制
1.1【Axial Attention】注意力介绍
下图是【Axial Attention】的结构图,让我们简单分析一下运行过程和优势
处理过程:
- 按通道进行处理:
- 图片展示了对图像不同颜色通道(如绿色 G 和蓝色 B)的分离处理。这一步表示每个通道单独执行处理流程,确保每个通道的特征得到充分学习。