yolo项目改进_添加CABM机制

134 阅读1分钟

CBAM是由通道注意力机制和空间注意力机制组成。

空间注意力可使神经网络更加关注图像中对分类起决定作用的像素区域而忽略无关紧要的区域,通道注意力则用于处理特征图通道的分配关系,同时对两个维度进行注意力分配,增强了注意力机制对模型性能的提升效果。 image.png

通道注意力模块

空间注意力模块