Abstract:
To address the challenges of low resolution, blurry imaging, and noise susceptibility in infrared small-target detection, this paper proposes a multiscale feature fusion infrared small-target detection algorithm, YOLO-InfraRed Tiny Detect (YOLO-IRTD), based on YOLOv8n. The algorithm first reconstructs the feature fusion network of YOLOv8 and introduces a feature-focused diffusion pyramid network to enhance the network's ability to extract small target features. It also designs an adaptive dynamic detection head that promotes information interaction between the classification and localization branches, improving the detection accuracy while reducing the model complexity. Finally, the SimAM attention mechanism is integrated into the Star Block to form the Star-SimAM module, which replaces the C2f module in the backbone network, further enhancing feature extraction capabilities while reducing the parameter count of the model. The experimental results show that the YOLO-IRTD model reduced the parameter count by 20%, with a weight size of only 5 MB. It achieved a mean average precision (mAP) of 90.4% on the HIT-UAV dataset, an improvement of 3.6% over the base model and outperforming YOLO series models by 1%-4%. On the FLIR dataset, the mAP improved by 1.2%. The model demonstrates excellent generalization performance, and its overall metrics are among the best in current mainstream detection models.