An Infrared Small Target Detection Method Based on Attention Mechanism

被引:8
作者
Wang, Xiaotian [1 ,2 ]
Lu, Ruitao [3 ]
Bi, Haixia [1 ]
Li, Yuhai [4 ]
机构
[1] Northwestern Polytech Univ, Unmanned Syst Res Inst, Xian 710072, Peoples R China
[2] Xi An Jiao Tong Univ, Sch Informat & Commun Engn, Xian 710049, Peoples R China
[3] Rocket Force Univ Engn, Dept Missile Engn, Xian 710025, Peoples R China
[4] Natl Key Lab Electromagnet Space Secur, Tianjin 300308, Peoples R China
关键词
infrared small target detection; feature frequency domain fusion; attention mechanism; feature weighting adjustment; LOCAL CONTRAST METHOD; MODEL; DIM;
D O I
10.3390/s23208608
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
The human visual attention system plays an important role in infrared target recognition because it can quickly and accurately recognize infrared small targets and has good scene adaptability. This paper proposes an infrared small target detection method based on an attention mechanism, which consists of three modules: a bottom-up passive attention module, a top-down active attention module, and decision feedback equalization. In the top-down active attention module, given the Gaussian characteristics of infrared small targets, the idea of combining knowledge-experience Gaussian shape features is applied to implement feature extraction, and quaternion cosine transform is performed to achieve multi-dimensional fusion of Gaussian shape features, thereby achieving complementary fusion of multi-dimensional feature information. In the bottom-up passive attention module, considering that the difference in contrast and motion between the target and the background can attract attention easily, an optimal fast local contrast algorithm and improved circular pipeline filtering are adopted to find candidate target regions. Meanwhile, the multi-scale Laplacian of the Gaussian filter is adopted to estimate the optimal size of the infrared small target. The fast local contrast algorithm based on box filter acceleration and structure optimization is employed to extract local contrast features, and candidate target regions can be obtained by using an adaptive threshold. Besides, the mean gray, target size, Gaussian consistency, and circular region constraint are used in pipeline filtering to extract motion regions, and the false-alarm rate is reduced effectively. Finally, decision feedback equalization is adopted to obtain real targets. Experiments are conducted on some real infrared images involving complex backgrounds with sea, sky, and ground clutters, and the experimental results indicate that the proposed method can achieve better detection performance than conventional baseline methods, such as RLCM, ILCM, PQFT, MPCM, and ADMD. Also, mathematical proofs are provided to validate the proposed method.
引用
收藏
页数:18
相关论文
共 36 条
[1]  
Achanta R, 2009, PROC CVPR IEEE, P1597, DOI 10.1109/CVPRW.2009.5206596
[2]   Content based image retrieval with sparse representations and local feature descriptors: A comparative study [J].
Celik, Ceyhun ;
Bilge, Hasan Sakir .
PATTERN RECOGNITION, 2017, 68 :1-13
[3]   A Local Contrast Method for Small Infrared Target Detection [J].
Chen, C. L. Philip ;
Li, Hong ;
Wei, Yantao ;
Xia, Tian ;
Tang, Yuan Yan .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2014, 52 (01) :574-581
[4]  
Chen QW, 2019, AAAI CONF ARTIF INTE, P9925
[5]   An effective infrared small target detection method based on the human visual attention [J].
Chen, Yuwen ;
Song, Bin ;
Wang, Dianjun ;
Guo, Linghua .
INFRARED PHYSICS & TECHNOLOGY, 2018, 95 :128-135
[6]   An infrared small target detection algorithm based on high-speed local contrast method [J].
Cui, Zheng ;
Yang, Jingli ;
Jiang, Shouda ;
Li, Junbao .
INFRARED PHYSICS & TECHNOLOGY, 2016, 76 :474-481
[7]   Attentional Local Contrast Networks for Infrared Small Target Detection [J].
Dai, Yimian ;
Wu, Yiquan ;
Zhou, Fei ;
Barnard, Kobus .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (11) :9813-9824
[8]   Reweighted Infrared Patch-Tensor Model With Both Nonlocal and Local Priors for Single-Frame Small Target Detection [J].
Dai, Yimian ;
Wu, Yiquan .
IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2017, 10 (08) :3752-3767
[9]   Robust Infrared Maritime Target Detection Based on Visual Attention and Spatiotemporal Filtering [J].
Dong, Lili ;
Wang, Bin ;
Zhao, Ming ;
Xu, Wenhai .
IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2017, 55 (05) :3037-3050
[10]   IRSDT: A Framework for Infrared Small Target Tracking with Enhanced Detection [J].
Fan, Jun ;
Wei, Jingbiao ;
Huang, Hai ;
Zhang, Dafeng ;
Chen, Ce .
SENSORS, 2023, 23 (09)