首页|基于改进Faster R-CNN的红外目标检测算法

基于改进Faster R-CNN的红外目标检测算法

Infrared target detection algorithm based on improved Faster R-CNN

扫码查看
为提升红外目标的检测精度,提出了一种引入频域注意力机制的Faster R-CNN红外目标检测算法.首先,针对红外图像边缘模糊和噪声问题,设计了一种并行的图像增强预处理结构;其次,在Faster R-CNN中引入频域注意力机制,设计了一种新型红外目标检测主干网络;最后,引入路径增强金字塔结构,融合多尺度特征进行预测,利用底层网络丰富的位置信息,提升检测精度.在红外飞机的数据集上进行实验,结果表明,改进后的Faster R-CNN目标检测框架比以ResNet50为主干的算法的AP提升了 7.6%.此外,与目前主流算法对比,本文算法提高了红外目标的检测精度,验证了算法改进的有效性.
In order to improve the detection accuracy of infrared targets,a Faster R-CNN infrared target detection algorithm introducing a frequency domain attention mechanism was proposed.Firstly,a parallel image enhancement preprocessing structure was designed to address the issues of edge blur and noise in infrared images.Secondly,a frequency domain attention mechanism was introduced into Faster R-CNN,and a new infrared target detection backbone network was designed.Finally,a path enhanced pyramid structure was introduced to fuse multi-scale features for prediction,and the rich location information of the underlying network was utilized to improve detection accuracy.The experiment was conducted on a dataset of infrared aircraft.The results show that the AP of improved Faster R-CNN target detection framework is 7.6%higher than that of the algorithm with ResNet50 as the main stem.In addition,compared with current mainstream algorithms,the proposed algorithm improves the detection accuracy of infrared targets and verifies the effectiveness of the algorithm improvement.

infrared target detectionimage enhancementFaster R-CNNfrequency domain attention mechanismmulti-scale feature fusion

汪西晨、彭富伦、李业勋、张俊举

展开 >

南京理工大学 电子工程与光电技术学院,江苏 南京 210094

西安应用光学研究所,陕西 西安 710065

江苏北方湖光光电有限公司,江苏 无锡 214100

红外目标检测 图像增强 Faster R-CNN 频域注意力机制 多尺度特征融合

国家自然科学基金

61971386

2024

应用光学
中国兵工学会 中国兵器工业第二0五研究所

应用光学

CSTPCD北大核心
影响因子:0.517
ISSN:1002-2082
年,卷(期):2024.45(2)
  • 17