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 条
  • [41] Infrared Dim Small Target Detection Method Based on Enhanced Local Contrast
    Yuan Ming
    Song Yansong
    Zhang Ziqi
    Zhao Xin
    Zhao Bo
    LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (04)
  • [42] A Fast Non-parametric Detection Method for Infrared Small Dim Target
    Zhang, Kun
    Sun, Peng
    Li, Xinguo
    2019 IEEE INTERNATIONAL CONFERENCE ON INDUSTRIAL TECHNOLOGY (ICIT), 2019, : 1425 - 1429
  • [43] Infrared dim and small target detection based on the human visual attention mechanism
    Deng, Yaping
    Wang, Min
    Lecture Notes in Electrical Engineering, 2015, 334 : 1153 - 1161
  • [44] A Dynamic Queue Based Pipeline Filter for Infrared Dim Small Target Detection
    Wang, Changcheng
    Shen, Yuheng
    Zhang, Di
    Cai, Yunze
    2015 34TH CHINESE CONTROL CONFERENCE (CCC), 2015, : 3770 - 3775
  • [45] Novel Asymmetric Pyramid Aggregation Network for Infrared Dim and Small Target Detection
    Lv, Guangrui
    Dong, Lili
    Liang, Junke
    Xu, Wenhai
    REMOTE SENSING, 2022, 14 (22)
  • [46] Review on Dim Small Target Detection Technologies in Infrared Single Frame Images
    Wang Haoxian
    Dong Heng
    Zhou Zhiquan
    LASER & OPTOELECTRONICS PROGRESS, 2019, 56 (08) : 1DUMM
  • [47] Infrared small and dim target detection based on weighted nuclear norm minimization
    Zhang C.
    Wang H.
    Lou J.
    Huazhong Keji Daxue Xuebao (Ziran Kexue Ban)/Journal of Huazhong University of Science and Technology (Natural Science Edition), 2017, 45 (10): : 31 - 37
  • [48] Infrared dim small target detection based on visual saliency and local entropy
    Zhao Peng-peng
    Li Shu-zhong
    Li Xun
    Luo Jun
    Chang Kai
    CHINESE OPTICS, 2022, 15 (02): : 267 - 275
  • [49] Detection Algorithm of Dim and Small Infrared Target Based on Temporal χ2 Test
    Sun Lihui
    Jin Sumei
    Zhang Ruisheng
    PROCEEDINGS OF THE 2009 2ND INTERNATIONAL CONGRESS ON IMAGE AND SIGNAL PROCESSING, VOLS 1-9, 2009, : 3539 - 3542
  • [50] A Lightweight Infrared Small Target Detection Network Based on Target Multiscale Context
    Ma, Tianlei
    Yang, Zhen
    Liu, Benxue
    Sun, Siyuan
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20