Journal of Computer Applications ›› 2025, Vol. 45 ›› Issue (1): 284-291.DOI: 10.11772/j.issn.1001-9081.2024010102

• Multimedia computing and computer simulation • Previous Articles     Next Articles

Infrared small target detection method based on information compensation

Boran YANG1, Suzhen LIN1(), Dawei LI2, Xiaofei LU3, Chenhui CUI1   

  1. 1.College of Data Science and Technology,North University of China,Taiyuan Shanxi 030051,China
    2.School of Electrical and Control Engineering,North University of China,Taiyuan Shanxi 030051,China
    3.Jiuquan Satellite Launch Center,Jiuquan Gansu 735000,China
  • Received:2024-01-29 Revised:2024-03-21 Accepted:2024-03-22 Online:2024-05-09 Published:2025-01-10
  • Contact: Suzhen LIN
  • About author:YANG Boran, born in 1999, M. S. candidate. His research interests include infrared small target detection.
    LI Dawei, born in 1980, Ph. D., associate professor. His research interests include pattern recognition, machine learning, image processing.
    LU Xiaofei, born in 1981, Ph. D. His research interests include measurement data processing.
    CUI Chenhui, born in 1998, M. S. candidate. His research interests include image processing, target tracking.
  • Supported by:
    Fundamental Research Program of Shanxi Province(202303021211147);Shanxi Province Patent Transformation Special Program(202302001)

基于信息补偿的红外弱小目标检测方法

杨博然1, 蔺素珍1(), 李大威2, 禄晓飞3, 崔晨辉1   

  1. 1.中北大学 计算机科学与技术学院,太原 030051
    2.中北大学 电气与控制工程学院,太原 030051
    3.酒泉卫星发射中心,甘肃 酒泉 735000
  • 通讯作者: 蔺素珍
  • 作者简介:杨博然(1999—),男,山西临汾人,硕士研究生,CCF会员,主要研究方向:红外弱小目标检测;
    李大威(1980—),男,河北衡水人,副教授,博士,主要研究方向:模式识别、机器学习、图像处理;
    禄晓飞(1981—),男,甘肃酒泉人,博士,主要研究方向:测量数据处理;
    崔晨辉(1998—),男,山西晋城人,硕士研究生,CCF会员,主要研究方向:图像处理、目标跟踪。
  • 基金资助:
    山西省基础研究计划项目(202303021211147);山西省专利转化专项计划项目(202302001)

Abstract:

An infrared small target method based on information compensation was proposed to address the problem that infrared small targets are prone to losing texture detail information during network iteration, which decreased accuracy of target localization and contour segmentation. Firstly, Image Feature Extraction (IFE) module was used to encode shallow details and deep semantic features of infrared image. Secondly, a Multi-level Information Compensation (MIC) module was constructed to perform information compensation to down-sampled features in the encoding stage by aggregating features from adjacent levels. Thirdly, Global Target Response (GTR) module was introduced to compensate the limitation of convolutional locality by incorporating global contextual information of feature map. Finally, Asymmetric Cross-Fusion (ACF) module was constructed to fuse shallow and deep features, thereby preserving texture and positional information during target decoding, thus achieving detection of infrared small targets. Experimental results of training and testing on publicly available NUAA-SIRST (Nanjing University of Aeronautics and Astronautics-Single-frame InfraRed Small Target) and NUDT-SIRST (National University of Defense Technology-Single-frame InfraRed Small Target) mixed datasets show that compared to methods such as UIUNet (U-Net in U-Net Network), LSPM (Local Similarity Pyramid Modules), and DNANet (Dense Nested Attention Network), the proposed method achieves improvements of 9.2, 8.9, and 5.5 percentage points in Intersection over Union (IoU), respectively, and 6.0, 5.4, and 3.1 percentage points in F1-Score, respectively. The above demonstrates that the proposed method enables accurate detection and effective segmentation of small targets in complex infrared background images.

Key words: target detection, infrared small target, information compensation, Global Target Response (GTR), Asymmetric Cross-Fusion (ACF)

摘要:

针对红外弱小目标容易在网络迭代过程中损失纹理细节信息,从而导致目标定位和轮廓分割的准确性下降的问题,提出一种基于信息补偿的红外弱小目标检测方法。首先,利用图像特征提取(IFE)模块编码红外源图像的浅层细节及深层语义特征;其次,构建多级信息补偿(MIC)模块通过聚合相邻级别的特征对编码阶段下采样后的特征进行信息补偿;随后,引入全局目标响应(GTR)模块联合特征图的全局上下文信息对卷积局部性的限制进行补偿;最后,构建非对称交叉融合(ACF)模块对浅层和深层特征进行融合,以实现目标解码时纹理信息与位置信息的保留,进而完成对红外弱小目标的检测。在公开的NUAA-SIRST (Nanjing University of Aeronautics and Astronautics-Single-frame InfraRed Small Target)和NUDT-SIRST (National University of Defense Technology-Single-frame InfraRed Small Target)混合数据集上训练和测试的实验结果表明,与UIUNet (U-Net in U-Net Network)、LSPM (Local Similarity Pyramid Modules)和DNANet (Dense Nested Attention Network)等方法相比,所提方法在交并比(IoU)上分别提高了9.2、8.9和5.5个百分点,在F1分数(F1-Score)上分别提高了6.0、5.4和3.1个百分点。以上表明所提方法对红外复杂背景图像中的弱小目标可以实现准确检测和有效分割。

关键词: 目标检测, 红外弱小目标, 信息补偿, 全局目标响应, 非对称交叉融合

CLC Number: