基于稀疏注意力的红外弱小目标检测方法

Infrared Weak Target Detection Method Based on Sparse Attention

  • 摘要: 针对复杂背景下红外弱小目标像素占比少,细节纹理特征匮乏导致特征提取困难、检测率低、虚警率高的问题,提出一种基于稀疏注意力和多尺度特征融合的红外弱小目标检测网络。该网络利用Resnest的分割注意力提取不同尺度特征,引入Biformer注意力模块学习目标与背景之间的远程关系,采用融合模块将高、低层特征进行融合,经过Head模块输出检测结果二值图。实验结果表明,本文方法在IoU和Fmeasure这两项指标中均取得最优,与DNANet方法相比,所提方法的交并比(IoU)提高3.9%、Fmeasure提高5.6%;与ABCNet方法相比,所提方法的IoU提高5.8%、Fmeasure提高10%;并且在不同复杂背景下均可有效检测出红外弱小目标,体现良好的鲁棒性和适应性,可以有效应用于复杂背景中的红外弱小目标检测。

     

    Abstract: In this study, a novel weak infrared small target detection network based on sparse attention and multiscale feature fusion is proposed to address the challenges of low pixel occupancy and limited texture features for weak infrared small targets within complex backgrounds, leading to difficulties in feature extraction, low detection rates, and high false alarm rates. The network utilizes the segmentation attention of ResNest to extract features at different scales. A BiFormer attention module is introduced to learn the distant relationships between targets and backgrounds. Furthermore, a fusion module is employed to merge both high- and low-level features, with the final detection results represented as a binary image through a head module. The experimental results demonstrate that the proposed method achieves the best performance in terms of both Intersection over Union (IoU) and F-measure. Compared with the dense nested attention network (DNANet), the proposed method improved the IoU by 3.9% and F-measure by 5.6%. Compared with the attentive bilateral contextual network (ABCNet), the proposed method improved the IoU by 5.8% and F-measure by 10%. Moreover, the proposed approach exhibited robustness and adaptability in effectively detecting small weak infrared targets in diverse, complex backgrounds. This method is applicable to weak infrared small-target detection in complex backgrounds, exhibiting superior performance.

     

/

返回文章
返回