基于改进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%。此外,与目前主流算法对比,本文算法提高了红外目标的检测精度,验证了算法改进的有效性。

     

    Abstract: 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.

     

/

返回文章
返回