GML-YOLO: a lightweight infrared small target detection algorithm

作者：Lin Jiang, Yixuan Shen, Mei Da, Jue Hu, Zhijian Zhang · 发表于：Measurement Science and Technology · 年份：2025 · DOI：10.1088/1361-6501/adf2d0 · 被引用次数：10 · 研究领域：Infrared Target Detection Methodologies

Abstract Infrared imaging technology captures the thermal radiation emitted by targets to form images, enabling the filtration of redundant information in complex road scenes and thus facilitating pedestrian and vehicle monitoring. However, the existing infrared target detection models suffer from inadequate accuracy, prone to false detections and missed detections in complex scenarios such as nighttime and adverse weather conditions, posing threats to traffic safety and intelligent driving. Moreover, these models typically have a large number of parameters and rely on high-performance GPUs, which increases hardware costs and restricts their deployment. Additionally, their slow detection speed makes it difficult to meet real-time requirements. In response to the aforementioned issues, this paper proposes a lightweight infrared small target detection algorithm: GML-YOLO. Firstly, we designed a lightweight backbone network, ghost-hierarchical geometry network, to improve feature extraction efficiency, enabling accurate and real-time feature extraction. Secondly, we incorporated adaptive downsampling and attention mechanisms in the network fusion part, replacing the simple concatenation used in traditional detectors. This design effectively integrates shallow and deep information. In addition, we have also designed the cross stage partial-mixed local channel attention module. This module innovatively reworks the original C2f module by integrating a hybrid attention mechanism, ef...