Laser & Optoelectronics Progress, Volume. 58, Issue 16, 1610010(2021)

Vehicle Recognition Method Based on Improved YOLOv3 Algorithm

Yongshun Wang, Wenjie Jia*, Chenfei Wang, and Hui Song
Author Affiliations
  • School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou, Gansu 730070, China
  • show less

    An improved YOLOv3 algorithm that detects target vehicles is proposed to address the problems of low detection accuracy of small targets and poor robustness of systems in target vehicle detection. First, the proposed algorithm introduces the dilated convolution into the downsampling layer of the YOLOv3 algorithm, improving the resolution of the feature maps and detection effect of small targets. Second, to address the problem of small target recognition in vehicle images, the proposed algorithm increases the three detection scales of YOLOv3 to four in addition to connecting and fusing the information with different scales, and the improved feature pyramid structure further improves small target detection. Finally, using Complete IoU (CIoU) as the loss function makes the target frame regression more stable, and there is no divergence in training. The KITTI dataset test results show that the improved YOLOv3 algorithm can achieve high detection accuracy. The proposed algorithm improves the average detection accuracy by 4.6%, and the detection rate is approximately 44.1 frame/s. On the premise of improving the accuracy, the proposed algorithm maintains a high detection rate.

    Tools

    Get Citation

    Copy Citation Text

    Yongshun Wang, Wenjie Jia, Chenfei Wang, Hui Song. Vehicle Recognition Method Based on Improved YOLOv3 Algorithm[J]. Laser & Optoelectronics Progress, 2021, 58(16): 1610010

    Download Citation

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

    Category: Image Processing

    Received: Nov. 17, 2020

    Accepted: Dec. 17, 2020

    Published Online: Aug. 19, 2021

    The Author Email: Jia Wenjie (361658066@qq.com)

    DOI:10.3788/LOP202158.1610010

    Topics