首页|基于注意力与量化感知的航拍红外目标检测

基于注意力与量化感知的航拍红外目标检测

扫码查看
原文链接
  • NETL
  • NSTL
  • 万方数据
针对航拍场景下红外目标对比度低、识别精度差、检测难度大等问题,提出一种基于注意力与量化感知的航拍红外目标检测算法.首先,利用DCNv2 替代ELAN模块中的 3×3 卷积,构建了DC-ELAN模块,有效提升了模型捕捉局部和全局特征的能力,进而强化了网络的特征表达能力;其次,通过巧妙地将SE注意力机制融入SPPCSPC模块和ELAN模块中,设计出了SE-SPPCSPC模块和SE-ELAN模块,有助于增强特征图的空间自注意力,模型能够更好地关注目标区域;此外,引入QARepVGG模块,提升模型的量化感知能力并增强其对量化误差的鲁棒性;最后,引入DyHead模块,该模块可以根据输入图像的不同动态调整检测头,提高模型对不同大小、形状目标的检测能力,从而进一步提高红外目标检测的准确性和鲁棒性.实验结果表明,相较于原模型,改进后的YOLOv7-tiny模型在计算量未增长的情况下,mAP@0.5值提升了3.4%,mAP@0.5:0.95值提升了4.8%,显著提高了模型检测精度.
Aerial Infrared Target Detection Based on Attention and Quantization Awareness
Aiming at the problems of low contrast,poor recognition accuracy,and difficult detection of infrared targets in aerial scenes,this study proposes an aerial infrared target detection algorithm based on attention and quantization awareness.Firstly,the DC-ELAN module is constructed by using DCNv2 to replace the 3×3 convolution in the ELAN module,which effectively improves the ability of the model to capture local and global features,and then strengthens the feature representation ability of the network.Secondly,by cleverly integrating the SE attention mechanism into the SPPCSPC module and the ELAN module,the SE-SPPCSPC module and the SE-ELAN module are designed,which helps to enhance the spatial self-attention of the feature map,and the model can better focus on target areas.In addition,the QARepVGG module is introduced to improve the quantization awareness of the model and enhance its robustness to quantization errors.Finally,the DyHead module is introduced,which can dynamically adjust the detection head according to different input images,improve the detection ability of the model to targets of different sizes and shapes,and further improve the accuracy and robustness of infrared target detection.Experimental results show that compared with the original model,the improved YOLOv7-tiny model has 3.4%and 4.8%increases in mAP@0.5 and mAP@0.5:0.95 values without increasing the amount of calculation,which significantly improves model detection accuracy.

infrared target detectiondeformable convolutionattention mechanismquantization-aware trainingtarget detection head

周进、裴晓芳

展开 >

南京信息工程大学 电子与信息工程学院,南京 210044

无锡学院 电子信息工程学院,无锡 214105

红外目标检测 可变形卷积 注意力机制 量化感知训练 目标检测头

国家自然科学基金青年项目苏高教会"高质量公共课教学改革研究"专项课题高校哲学社会科学研究一般项目江苏职业教育研究立项课题一般项目无锡学院教改课题无锡学院教改课题2023江苏省大学生创新创业训练计划

422050782022JDKT1382022SJYB0979XHYBLX2023282XYJG2023002XYJG2023023202313982007Z

2024

10.15888/j.cnki.csa.009699

计算机系统应用
中国科学院软件研究所

计算机系统应用

CSTPCD
影响因子：0.449
ISSN：1003-3254
年,卷(期)：2024.33(11)