YOLO-MST: Multiscale deep learning method for infrared small target detection based on super-resolution and YOLO

被引:0
作者
Yue, Taoran [1 ,2 ]
Lu, Xiaojin [2 ]
Cai, Jiaxi [2 ]
Chen, Yuanping [1 ,2 ]
Chu, Shibing [1 ,2 ]
机构
[1] Jiangsu Univ, Sch Phys & Elect Engn, Zhenjiang 212013, Peoples R China
[2] Jiangsu Engn Res Ctr Quantum Percept & Intelligent, Zhenjiang 212013, Peoples R China
基金
中国国家自然科学基金;
关键词
Deep learning; Infrared remote sensing; Small target recognition; Super-resolution; YOLO;
D O I
10.1016/j.optlastec.2025.112835
中图分类号
O43 [光学];
学科分类号
070207 ; 0803 ;
摘要
With the advancement of aerospace technology and the increasing demands of military applications, the development of low false-alarm and high-precision infrared small target detection algorithms has emerged as a key focus of research globally. However, the traditional model-driven method is not robust enough when dealing with features such as noise, target size, and contrast. The existing deep-learning methods have limited ability to extract and fuse key features, and it is difficult to achieve high-precision detection in complex backgrounds and when target features are not obvious. To solve these problems, this paper proposes a deep-learning infrared small target detection method that combines image super-resolution technology with multi-scale observation. First, the input infrared images are preprocessed with super-resolution and multiple data enhancements are performed. Secondly, based on the YOLOv5 model, we proposed a new deep-learning network named YOLO-MST. This network includes replacing the SPPF module with the self-designed MSFA module in the backbone, optimizing the neck, and finally adding a multi-scale dynamic detection head to the prediction head. By dynamically fusing features from different scales, the detection head can better adapt to complex scenes. The mAP@0.5 detection rates of this method on three datasets IRIS, SIRST and SIRST+ reach 99.5%, 96.4% and 91.4% respectively, more effectively solving the problems of missed detection, false alarms, and low precision.
引用
收藏
页数:10
相关论文
共 52 条
  • [51] Lightweight Hot-Spot Fault Detection Model of Photovoltaic Panels in UAV Remote-Sensing Image
    Zheng, Qiuping
    Ma, Jinming
    Liu, Minghui
    Liu, Yuchen
    Li, Yanxiang
    Shi, Gang
    [J]. SENSORS, 2022, 22 (12)
  • [52] Zhu XZ, 2021, Arxiv, DOI arXiv:2010.04159