Technical Paper
Research on Vehicle Type Recognition Based on Improved YOLOv5 Algorithm
2024-04-09
2024-01-1992
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.