Laser & Infrared, Volume. 55, Issue 1, 130(2025)

Infrared pedestrian vehicle detection algorithm based on improved YOLOV8

QIN Hai-yang1,2, TAN Gong-quan1,2、*, DENG Hao1,2, WANG Yao1,2, CAI Da-yang1,2, and WEN Li1,2
Author Affiliations
  • 1School of Automation and Information Engineering, Sichuan University of Science & Engineering, Yibin 644000, China
  • 2Artificial Intelligence Key Laboratory of Sichuan Province, Sichuan University of Science & Engineering, Yibin 644000, China
  • show less

    Given that infrared pedestrian-vehicle images are difficult to detect due to their low resolution, poor quality, and high noise, an infrared image pedestrian and vehicle target detection algorithm based on YOLOV8 is proposed, namely PSWG-YOLO. For the YOLOv8n network, a 160×160 maximum feature map P2 is added to improve the model's detection ability of pedestrian small targets. At the same time, the SPD-Conv part is used to replace the stride-2 convolutional layer of the original network to improve the feature extraction capability of low-resolution images. In addition, the loss function is replaced with WIoU to optimize the model's processing of low-quality images. Finally, the Ghost module is introduced to reduce model complexity. The experimental results show that the improved PSWG-YOLO algorithm significantly reduces the model volume and parameter amount while ensuring high detection accuracy. Compared with the original YOLOv8n algorithm, the P, R, and mAP@0.5 on the public infrared data set FLIR_v2 are increased by 1.6 %, 6.3 %, and 7.2 % respectively, and the number of parameters is reduced by 16 %, and the model size is reduced by 15.8 %, which improves the accuracy of the pedestrian-vehicle detection in infrared scenarios and is easy to deploy.

    Tools

    Get Citation

    Copy Citation Text

    QIN Hai-yang, TAN Gong-quan, DENG Hao, WANG Yao, CAI Da-yang, WEN Li. Infrared pedestrian vehicle detection algorithm based on improved YOLOV8[J]. Laser & Infrared, 2025, 55(1): 130

    Download Citation

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

    Category:

    Received: Apr. 1, 2024

    Accepted: Mar. 13, 2025

    Published Online: Mar. 13, 2025

    The Author Email: TAN Gong-quan (tgq77@163.com)

    DOI:10.3969/j.issn.1001-5078.2025.01.019

    Topics