Research on Vehicle Type Recognition Based on Improved YOLOv5 Algorithm

2024-01-1992

04/09/2024

Features
Event
WCX SAE World Congress Experience
Authors Abstract
Content
As a key technology of intelligent transportation system, vehicle type recognition plays an important role in ensuring traffic safety,optimizing traffic management and improving traffic efficiency, which provides strong support for the development of modern society and the intelligent construction of traffic system. Aiming at the problems of large number of parameters, low detection efficiency and poor real-time performance in existing vehicle type recognition algorithms, this paper proposes an improved vehicle type recognition algorithm based on YOLOv5. Firstly, the lightweight network model MobileNet-V3 is used to replace the backbone feature extraction network CSPDarknet53 of the YOLOv5 model. The parameter quantity and computational complexity of the model are greatly reduced by replacing the standard convolution with the depthwise separable convolution, and enabled the model to maintain higher accuracy while having faster reasoning speed. Secondly, the attention mechanism in MobileNet-V3 is improved, and a more efficient coordinate attention module is embedded to enhance the model ‘s attention to key features, so as to further improve the accuracy of vehicle type recognition. Finally, in order to better integrate different levels of features, the weighted bidirectional feature pyramid network BiFPN structure is introduced. By processing the top-down and bottom-up paths at the same time, the feature maps of different resolutions are fused, so that the model can obtain richer feature information at multiple scales, thereby improving detection efficiency and accuracy. The experimental results show that the improved algorithm has a vehicle type recognition accuracy of 96% on the BIT-Vehicle dataset. Compared with the original YOLOv5 model, the number of model parameters is reduced by 47%, and the detection speed is increased by 54.6%. Therefore, the improved algorithm in this paper can better meet the requirements of small parameter quantity and high detection efficiency, and is suitable for real-time vehicle type recognition.
Meta TagsDetails
DOI
https://doi.org/10.4271/2024-01-1992
Pages
8
Citation
Liu, X., "Research on Vehicle Type Recognition Based on Improved YOLOv5 Algorithm," SAE Technical Paper 2024-01-1992, 2024, https://doi.org/10.4271/2024-01-1992.
Additional Details
Publisher
Published
Apr 09
Product Code
2024-01-1992
Content Type
Technical Paper
Language
English