Lightweight Design for Infrared Dim and Small Target Detection in Complex Environments

被引:0
|
作者
Chang, Yan [1 ]
Ma, Decao [1 ]
Ding, Yao [1 ]
Chen, Kefu [1 ]
Zhou, Daming [2 ]
机构
[1] PLA Rocket Force Univ Engn, Xian 710025, Peoples R China
[2] Northwestern Polytech Univ, Sch Astronaut, Xian 710072, Peoples R China
关键词
object detection; lightweight; small infrared targets; attention mechanism; NETWORK; MODEL;
D O I
10.3390/rs16203761
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
In the intricate and dynamic infrared imaging environment, the detection of infrared dim and small targets becomes notably challenging due to their feeble radiation intensity, intricate background noise, and high interference characteristics. To tackle this issue, this paper introduces a lightweight detection and recognition algorithm, named YOLOv5-IR, and further presents an even more lightweight version, YOLOv5-IRL. Firstly, a lightweight network structure incorporating spatial and channel attention mechanisms is proposed. Secondly, a detection head equipped with an attention mechanism is designed to intensify focus on small target information. Lastly, an adaptive weighted loss function is devised to improve detection performance for low-quality samples. Building upon these advancements, the network size can be further compressed to create the more lightweight YOLOv5-IRL version, which is better suited for deployment on resource-constrained mobile platforms. Experimental results on infrared dim and small target detection datasets with complex backgrounds indicate that, compared to the baseline model YOLOv5, the proposed YOLOv5-IR and YOLOv5-IRL detection algorithms reduce model parameter counts by 42.9% and 45.6%, shorten detection time by 13.6% and 16.9%, and enhance mAP0.5 by 2.4% and 1.8%, respectively. These findings demonstrate that the proposed algorithms effectively elevate detection efficiency, meeting future demands for infrared dim and small target detection.
引用
收藏
页数:24
相关论文
共 50 条
  • [31] Dim-small moving target detection in infrared image sequences
    Zhang Q.
    Cai J.
    Zhang Q.
    Qiangjiguang Yu Lizishu/High Power Laser and Particle Beams, 2011, 23 (12): : 3312 - 3316
  • [32] Infrared dim and small target detection based on U-Transformer
    Lin, Jian
    Zhang, Kai
    Yang, Xi
    Cheng, Xiangzheng
    Li, Chenhui
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2022, 89
  • [33] Guided Attention and Joint Loss for Infrared Dim Small Target Detection
    Tong, Yunfei
    Liu, Jing
    Fu, Zhiling
    Wang, Zhe
    Yang, Hai
    Niu, Saisai
    Tan, Qinyan
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [34] Target-Focused Enhancement Network for Distant Infrared Dim and Small Target Detection
    Tong, Yunfei
    Leng, Yue
    Yang, Hai
    Wang, Zhe
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2024, 62
  • [35] The optimal design of top-hat morphological filter and application to infrared small dim target detection
    School of Information and Communication Engineering, Harbin Engineering University, Harbin 150001, China
    J. Inf. Comput. Sci., 2008, 6 (2393-2398):
  • [36] Fast Infrared Dim and Small Target Tracking
    Qian, Kun
    Zhou, Huixin
    Rong, Shenghui
    Wang, Bingjian
    Yin, Shiming
    Zeng, Qingjie
    AOPC 2015: IMAGE PROCESSING AND ANALYSIS, 2015, 9675
  • [37] Infrared Dim and Small Target Detection Based on the Improved Tensor Nuclear Norm
    Fan, Xiangsuo
    Wu, Anqing
    Chen, Huajin
    Huang, Qingnan
    Xu, Zhiyong
    APPLIED SCIENCES-BASEL, 2022, 12 (11):
  • [38] Infrared dim and small target detection network based on spatial attention mechanism
    Ju M.-R.
    Luo H.-B.
    Liu G.-Q.
    Liu Y.-P.
    Guangxue Jingmi Gongcheng/Optics and Precision Engineering, 2021, 29 (04): : 843 - 853
  • [39] Infrared Small Dim Target detection Based on Weighted Nuclear Norm Minimization
    Wang, Huan
    Lou, Jing
    Zhang, Congcong
    Liu, Yong
    2017 IEEE 7TH ANNUAL INTERNATIONAL CONFERENCE ON CYBER TECHNOLOGY IN AUTOMATION, CONTROL, AND INTELLIGENT SYSTEMS (CYBER), 2017, : 388 - 393
  • [40] Infrared dim and small target detection based on YOLO-IDSTD algorithm
    Jiang X.
    Cai W.
    Yang Z.
    Xu P.
    Jiang B.
    Hongwai yu Jiguang Gongcheng/Infrared and Laser Engineering, 2022, 51 (03):