- Conference Article
- 10.1145/3584376.3584479
Intelligent vehicle detection and tracking system based on improved artificial intelligence algorithm
- Dec 16, 2022
- Yuxin Liu + 1 more +1
Aiming at the problems of low detection and tracking accuracy, poor robustness and inaccurate vehicle flow statistics of multi-vehicle types in surveillance video, an artificial intelligence-based vehicle detection and tracking method was proposed. This method reconstructed the feature extraction network of the target detection model YOLOv5s to enhance the extraction of important features of the target and improve the detection accuracy of the detector. Firstly, Swin Transformer module is introduced into the Backbone network to replace part of C3 modules in the original algorithm, so as to enhance the global modeling ability of the model, better capture the context feature information, and expand the model's receptive field. Then, by comparing different attention mechanisms, NAM attention is selected to access the Neck network to enhance the cross-dimensional interaction between the channel and spatial dimensions of information, reduce information loss and enhance the performance of the network. Finally, the feature extraction network part of the DeepSort algorithm was optimized and retrained on the vehicle rerecognition data set to make it more suitable for vehicle tracking. The experimental results show that the accuracy of vehicle detection by the improved YOLOv5s is increased by 1.93 percentage points compared with the original algorithm. Combined with the DeepSort algorithm, the accuracy of vehicle flow statistics under different lighting conditions such as daytime, evening and night can reach 97.2%, 97.5% and 80.4% respectively.
Read more