Laser & Infrared, Volume. 54, Issue 2, 295(2024)

Research on pedestrian target detection in lightweight infrared images based on YOLOv5s

HU Yan1, ZHAO Yu-hang1、*, HU Hao-bing1, GONG Yin2, and SUN Huan-yu3
Author Affiliations
  • 1School of Aviation Electronics and Electrical Engineering, China Civil Aviation Flight Academy, Guanghan 618307, China
  • 2Tibet Airlines Co., Ltd., Chengdu 610202, China
  • 3Shandong Airlines Co., Ltd., Jinan 250107, China
  • show less

    Pedestrian recognition based on infrared images is an important component of modern security systems. In scenarios with limited computing resources, it is often difficult to balance the detection accuracy and deployment difficulty due to the influence of model size in infrared pedestrian detection algorithms. In response to this issue, a lightweight object detection algorithm based on YOLOv5s is proposed in this paper. Firstly, the MobileNetv3 lightweight feature extraction network is introduced and deep separable convolution is used to reduce the model size, making it easier to deploy to CPU devices. Secondly, the nearest neighbor interpolation upsampling method is replaced with CARAFE (Content-Aware ReAssembly of FEatures) which significantly improves the image reconstruction effect. Finally, EIOU Loss is used as the loss function of the bounding box to improve the regression performance of the model. Additionally, tests are conducted on the sampled LLVIP infrared pedestrian image dataset and the results show that for pedestrian targets in infrared images, the model size is reduced by 80.6% and the number of parameters is reduced by 82.8% while maintaining a high detection accuracy (AP=95.4%); and the inference speed is improved by 43.3% when using a CPU platform for inference, and the performance of detecting multi-scale targets is improved. The above two results validate the effectiveness of the algorithm.

    Tools

    Get Citation

    Copy Citation Text

    HU Yan, ZHAO Yu-hang, HU Hao-bing, GONG Yin, SUN Huan-yu. Research on pedestrian target detection in lightweight infrared images based on YOLOv5s[J]. Laser & Infrared, 2024, 54(2): 295

    Download Citation

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

    Category:

    Received: Apr. 23, 2023

    Accepted: Jun. 4, 2025

    Published Online: Jun. 4, 2025

    The Author Email: ZHAO Yu-hang (1127095630@qq.com)

    DOI:10.3969/j.issn.1001-5078.2024.02.020

    Topics