Journal of Optoelectronics · Laser, Volume. 35, Issue 11, 1166(2024)

Lightweight night driving infrared image target detection algorithm

CHEN Yifang1,2,3, ZHANG Shang2,3, and RAN Xiukang1,2,3
Author Affiliations
  • 1College of Electrical Engineering and New Energy, China Three Gorges University, Yichang, Hubei 443002, China
  • 2Hubei Province Engineering Technology Research Center for Construction Quality Testing Equipment, China Three Gorges University, Yichang, Hubei 443002, China
  • 3College of Computer and Information Technology,China Three Gorges University, Yichang, Hubei 443002, China
  • show less

    To solve these problems, such as large amount of calculation, lack of ability of generalization and poor detection performance, a light-weight night driving infrared image target detection algorithm is proposed in this paper. The algorithm first utilizes the Ghost structure as the backbone network to reduce the amount of model calculation. Then, the bidirectional feature gramid network (BIFPN) structure and coordinate attention (CA) mechanism are introduced in the neck to improve the model detection effect. Finally, the Focal-EIOU and Mish functions are used as the loss function and activation function of the algorithm to improve the convergence speed and regression accuracy. The experimental results show that the improved algorithm has significantly improved compared with YOLOv3-tiny in all aspects. Compared with YOLOv5, the accuracy has increased to 88.9%, the model volume has been reduced by 24.09%, the number of parameters has been reduced by 25.07%, and the amount of calculation has been reduced by 28.48%, the detection accuracy is improved in the two categories of person and bicycle. A balance between detection accuracy and model complexity is achieved.

    Tools

    Get Citation

    Copy Citation Text

    CHEN Yifang, ZHANG Shang, RAN Xiukang. Lightweight night driving infrared image target detection algorithm[J]. Journal of Optoelectronics · Laser, 2024, 35(11): 1166

    Download Citation

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

    Category:

    Received: Mar. 30, 2023

    Accepted: Dec. 31, 2024

    Published Online: Dec. 31, 2024

    The Author Email:

    DOI:10.16136/j.joel.2024.11.0153

    Topics