Laser & Infrared, Volume. 55, Issue 3, 452(2025)

Lightweight infrared and visible image detection methods for UAV perspectives

JIANG Xing-guo1,2, WANG Yao1, LIN Guo-jun1,2, SUN Xiao1, DIAO Hao-jie1, and LI Ming1
Author Affiliations
  • 1School of Automation and Information Engineering, Sichuan University of Science and Engineering, Yibin 644000, China
  • 2Artificial Intelligence of Key Laboratory of Sichuan Province, Yibin 644001, China
  • show less

    Aiming at the UAV aerial photography viewpoint target detection spatial scale change is large, the object pixels account for a small proportion, and the algorithm deployment edge computing platform storage space occupies a large proportion of the problem. In this paper, based on the YOLOv8n network structure, an improved aerial photography viewpoint lightweight small target detection method DSF-YOLO-P algorithm is proposed. Firstly, the backbone network C2f module is integrated with FasterNet to form the Faster-C2f lightweight module to ensure that the model achieves network lightweighting and improves the detection speed without affecting the detection accuracy. Then, a new 160×160 prediction head is added and the network channels are reconfigured to improve the accuracy and robustness of the model for small target detection. The improved DSF-YOLO algorithm improves the accuracy by 2.5% and 0.6% on the visible dataset VisDrone2019 and infrared dataset HIT-UAV, respectively, and reduces the number of parameters by 10%. Finally, the DSF-YOLO algorithm is subjected to the dependency graph pruning operation to reduce the redundant parameters of the model without affecting the model performance. The pruned DSF-YOLO-P algorithm achieves the same accuracy and reduces the computational effort and number of parameters by 45% and 26%, respectively, compared with the DSF-YOLO algorithm on the VisDrone2019 dataset. The experimental results fully demonstrate the effectiveness of the DSF-YOLO-P algorithm in detecting small targets in the aerial view of UAVs.

    Tools

    Get Citation

    Copy Citation Text

    JIANG Xing-guo, WANG Yao, LIN Guo-jun, SUN Xiao, DIAO Hao-jie, LI Ming. Lightweight infrared and visible image detection methods for UAV perspectives[J]. Laser & Infrared, 2025, 55(3): 452

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Jun. 19, 2024

    Accepted: Apr. 23, 2025

    Published Online: Apr. 23, 2025

    The Author Email:

    DOI:10.3969/j.issn.1001-5078.2025.03.020

    Topics