Laser & Optoelectronics Progress, Volume. 58, Issue 16, 1610010(2021)
Vehicle Recognition Method Based on Improved YOLOv3 Algorithm
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.
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
Category: Image Processing
Received: Nov. 17, 2020
Accepted: Dec. 17, 2020
Published Online: Aug. 19, 2021
The Author Email: Jia Wenjie (361658066@qq.com)