Laser & Infrared, Volume. 54, Issue 1, 122(2024)

Low altitude lightweight infrared weak small target detection algorithm

ZHANG Shang1,2, HUANG Jun-feng1,2, WANG Heng-tao1,2、*, CHEN Yong-lin1,2, and WANG Kang1,2
Author Affiliations
  • 1Hubei Province Engineering Technology Research Center for Construction Quality Testing Equipment, China Three Gorges University, Yichang 443002, China
  • 2College of Computer and Information, China Three Gorges University, Yichang 443002, China
  • show less

    Accurate infrared small and weak target detection is the key to real-time monitoring, tracking, and guidance. Infrared weak and small targets have problems of high detection difficulty, high false detection, and serious missed detection. In this paper, an ultra-lightweight infrared dim small target detection algorithm SL-YOLO is proposed to improve the real-time performance and detection accuracy of infrared dim small target detection algorithms. Firstly, the downsampling scheme is redesigned to adjust the network architecture for the infrared image feature information to solve the problem of feature gradient reduction and feature disappearance for infrared weak targets. Then, a network model pruning algorithm is designed to integrate pruning algorithm with network structure, removing redundant parameters, and improving detection speeds. Finally, the SIoU Varifocal loss function is designed to equalise the positive and negative samples with overlapping losses while weighting the positive samples to solve the problem of background interference. The experimental results show that the detection accuracy is improved to 96.4% and 98.1% under the SIRST and IDSAT datasets, respectively. The model volume and computational complexity can be compressed to 190 kB and 0.9 GFLOPs, and the inference speed is reduced to less than 3 ms. Comparing with the mainstream algorithms, the improved algorithm has achieved good results in terms of detection accuracy, model volume; computational complexity. It can meet the real-time detection requirements.

    Tools

    Get Citation

    Copy Citation Text

    ZHANG Shang, HUANG Jun-feng, WANG Heng-tao, CHEN Yong-lin, WANG Kang. Low altitude lightweight infrared weak small target detection algorithm[J]. Laser & Infrared, 2024, 54(1): 122

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Feb. 21, 2023

    Accepted: Apr. 22, 2025

    Published Online: Apr. 22, 2025

    The Author Email: WANG Heng-tao (1248558938@qq.com)

    DOI:10.3969/j.issn.1001-5078.2024.01.018

    Topics