Laser & Infrared, Volume. 55, Issue 5, 789(2025)
Infrared dim small target detection method based on Yolov8
To address the issue of the lack of targeted detection of infrared dim small targets by existing deep learning network structures, a method for recognizing infrared dim small targets based on an improved YOLOv8, named UT-YOLOv8 (YOLOv8 enhanced with UniRepLK Block and Triplet Attention) is proposed. In this method, a triplet attention mechanism is introduced in the detection head at the output end of the feature fusion network. Additionally, new small target detection layers and detection heads are added within the feature fusion network, and large kernel convolutions are incorporated within the spatial pyramid pooling layer of the feature extraction network. These enhancements are tailored to the imaging characteristics of infrared dim small targets. Validated on real infrared image data, the experimental results indicate that the UT-YOLOv8 algorithm, while maintaining high detection speed, has effectively enhanced the network's precision in recognizing infrared weak small targets, achieving a mean Average Precision mAP@0.5 of 95.9%.
Get Citation
Copy Citation Text
LI Xue-feng, LI Ning, WU Di, YU Xiang-yue, GUO Yong-qiang. Infrared dim small target detection method based on Yolov8[J]. Laser & Infrared, 2025, 55(5): 789
Category:
Received: Jun. 24, 2024
Accepted: Jul. 11, 2025
Published Online: Jul. 11, 2025
The Author Email: LI Ning (119124328@qq.com)