1. 算法设计:基于 MCA 的 YOLOv8 优化
1.1 总体架构
YOLOv8 的优化算法在原有架构的基础上,引入了 MCA 模块,以增强特征提取能力和目标定位精度。MCA 模块被嵌入到 YOLOv8 的主干网络(Backbone)和特征金字塔网络(FPN)中,用于捕捉多维度的上下文信息。
1.2 MCA 模块设计
MCA 模块的核心思想是通过多维度的注意力机制(如通道注意力、空间注意力和尺度注意力)来增强特征表示。其结构如下:
-
通道注意力(Channel Attention):通过全局平均池化和全连接层计算通道权重,增强重要通道的特征。
-
空间注意力(Spatial Attention):通过卷积操作计算空间权重,关注目标区域。
-
尺度注意力(Scale Attention):通过多尺度特征融合,增强对不同尺度目标的检测能力。
MCA 模块的输出是上述三种注意力机制的加权融合结果。