Electric Vehicle Traffic Flow Detection Algorithm based on Improved YOLOv5s

Main Article Content

Guanfang Zuo
Fang Zhao
Tao Huang
Sicheng Wang
Sirui Gu

Abstract

In the view of uneven distribution of electric vehicle charging piles in some areas, resulting in a shortage of charging resources. The flow of electric vehicles in the area can be detected, so that charging piles can be reasonably deployed according to the traffic flow. Therefore, we propose a traffic flow detection algorithm based on improved CAR-YOLOv5s algorithm. In terms of the backbone network, an improved MobileNetv3 architecture is used to balance the speed and precision of the model. A ghost lightweight module is introduced into the neck network to improve the detection speed of the model. After the model is trained and verified, it is combined with DeepSort to track, detect and count the driving electric vehicles and detect the traffic flow of electric vehicles. The improved CAR-YOLOv5s model parameters are reduced by 3.43M and the floating-point calculation is reduced by 10.2G. The tracking effect was tested on the GPU with an average speed of 20.8FPS, which is 27.18% faster than before the improvement. The experimental results show that the improved algorithm is suitable for real-time tracking and detection of electric vehicles on traffic roads.

Article Details

Section
Articles