融合注意力机制的多尺度红外目标检测

作者：李向荣; 孙立辉^*

来源：红外技术, 2023, 45(07): 746-754.

• 红外图像
• 目标检测
• YOLOv4
• 通道注意力机制

针对红外图像存在细节纹理特征差、对比度低、目标检测效果差等问题，基于YOLOv4(You Only Look Once version 4)架构提出了一种融合通道注意力机制的多尺度红外目标检测模型。该模型首先通过降低主干特征提取网络深度，减少了模型参数。其次，为补充浅层高分辨率特征信息，重新构建多尺度特征融合模块，提高了特征信息利用率。最后在多尺度加强特征图输出前，融入通道注意力机制，进一步提高红外特征提取能力，降低噪声干扰。实验结果表明，本文算法模型大小仅为YOLOv4的28.87%，对红外目标的检测精度得到了明显提升。

• 单位
  河北经贸大学