南京邮电大学学报(自然科学版)

作者：尹　芹,方　晖,王金东,王　侃,晏天文,霍智勇

单位：中兴通讯股份有限公司多媒体视讯产品部，江苏 南京 210000中兴通讯股份有限公司多媒体视讯产品部，江苏 南京 210000中兴通讯股份有限公司多媒体视讯产品部，江苏 南京 210000中兴通讯股份有限公司多媒体视讯产品部，江苏 南京 210000南京邮电大学 通信与信息工程学院，江苏 南京 210003南京邮电大学 通信与信息工程学院，江苏 南京 210003

关键词：注意力机制; 小目标检测; 自注意力；通道注意力；空间注意力

基金：中兴通讯研究基金资助项目

小目标检测是计算机视觉领域具有挑战性的问题。空间注意力和通道注意力机制的使用提高了目标检测网络的均值平均精度，但捕获小物体上下文信息的能力仍然有限，并且在小目标和大中型目标的检测精度上存在差距，难以感知小物体的位置。算法构建了一种基于通道自注意力机制(Channel Self-Attention， CSA)的算法模块，将输入特征映射压缩后，运用自注意力机制建立特征通道间相关性，自适应地重新优化特征通道的响应，提升了捕获小物体远距离上下文信息的能力，从而提高了对小目标的检测精度。 实验结果表明，在几乎不增加计算成本的情况下，CSA块能够为现有目标检测网络带来性能改进。在 PASCAL VOC2007数据集上，采用通道自注意力机制的RetinaNet的 mAP值分别比原始RetinaNet的mAP值高3.11个百分点。使用通道自注意力机制的MobileNetv2比原始的MobileNetv2 mAP值提高3.05个百分点。

来源：2022年第4期

《南京邮电大学学报(自然科学版)》期刊编辑部

查看南京邮电大学学报(自然科学版)杂志2022年第4期