Vehicle Multi-Object Detection and Tracking Algorithm Based on Improved You Only Look Once 5s Version and DeepSORT
Source abstract
Abstract
The increasing popularity of vehicles has led to traffic congestion and frequent traffic accidents. Intelligent transportation technology is an effective solution to this problem. In order to improve the accuracy and effectiveness of vehicle detection and tracking, this paper combined the improved YOLOv5s model with the optimized DeepSORT tracking algorithm to detect and track vehicles on traffic roads. Firstly, in the detection model of YOLOv5s, the Attention-based Intra-scale Feature Interaction (AIFI) module is introduced to detect vehicles more quickly and accurately. Secondly, the Kalman filtering (KF) algorithm of DeepSORT is optimized to improve the accuracy of predictions of the vehicle state by using the width to replace the length-to-width ratio of the vehicle prediction box in the original KF algorithm. Finally, in the re-recognition network of DeepSORT, the original Convolutional Neural Network (CNN) model is replaced by an improved ResNet36 as the backbone network for feature extraction. The experimental results show that, compared with the original algorithm, when examining the performance of the improved algorithm in terms of target detection, the recall rate, average accuracy (mAP), and detection speed, are increased by 7.7%, 15.5%, and 14.2%, respectively; in terms of multi-object tracking performance, such as multi-object tracking precision (MOTP) and multi-object tracking accuracy (MOTA), improvements of 14.84% and 9.62%, respectively, are obtained and the total number of times a trajectory is fragmented (Frag) is reduced by 32.52%.These results indicate that the proposed algorithm can meet the requirements of accuracy, real-time detection, and stable vehicle detection and tracking on traffic roads.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Video Surveillance and Tracking Methods · Advanced Neural Network Applications · Vehicle License Plate Recognition
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Physics and Electronics, Nanning Normal University, Nanning 530100, China
Guangxi Key Laboratory of Functional Information Materials and Intelligent Information Processing, Nanning 530001, China
Funding
Guangxi Science and Technology Program · National Natural Science Foundation of China · Natural Science Foundation of Guangxi Province
References (22)
Show source-provided references
1. Hedges Company (2023, April 15). How Many Cars Are There in the World. Available online: https://hedgescompany.com/blog/2021/06/how-many-cars-are-there-in-the-world/.
2. Traffic anomaly detection based on multi-target tracking
3. Qiao, P. (2019). Research on Traffic Flow Detection Based on Deep Learning and Edge Task Offloading. [Master’s Thesis, Xidian University].
4. Aharon, N., Orfaig, R., and Bobrovsky, B.Z. (2022). BoT-SORT: Robust Associations Multi-Pedestrian Tracking. arXiv.
5. Online Multiple Object Tracking Based on State Prediction and Motion Structure10.3724/SP.J.1089.2018.16263 ↗
6. Bashar, M.K., Islam, S., Hussain, K.K., Hasan, B., Rahman, A., and Kabir, H. (2022). Multiple Object Tracking in Recent Times: A Literature Review. arXiv.
7. Multi-target tracking of person based on deep learning10.32604/csse.2023.038154 ↗
8. Multi-objective algorithm based on YOLOv5+DeepSort
9. Bao, J., Dong, Y.C., and Liu, H.Z. (2019, January 1–2). Survey of Object Tracking Algorithms Based on Deep-Sort. In Network Application Branch of China Computer Users Association. Proceedings of the 23rd Annual Conference on New Network Technology and Application in 2019 of the Network Application Branch of the China Computer Users Association, Enshi, China.
10. Research on Pedestrian Tracking Method Based on Improved DeepSort
11. An improved DeepSORT mine personnel tracking algorithm
12. Stable and real-time pedestrian tracking method based on improved DeepSORT under complex background10.37188/CJLCD.2022-0350 ↗
13. Research on Dynamic Vehicle Multi-Object Tracking Method Based on DeepSort
14. Pedestrian multi-target tracking algorithm based on improved DeepSort
15. Road Target Tracking Algorithm Based on Improved YOLOv5 and DeepSort
16. Liu, L. (2019). Research on Traffic Flow Statistics of Intelligent Transportation Based on YOLO Network. [Master’s Thesis, Xi’an University of Science and Technology].
17. Redmon, J., Divvala, S., Girshick, R., and Farhadi, A. (July, January 26). You only look once: Unified, real-time object detection. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA.
18. Lv, W.Y., Xu, S.L., Zhao, Y., Wang, G.Z., Wei, J.M., Cui, C., Du, Y.N., Dang, Q.Q., and Liu, Y. (2023). DETRs Beat YOLOs on real-time object detection. arXiv.
19. Ashish, V., Noam, S., Niki, P., Jakob, U., Llion, J., Aidan, N.G., Lukasz, K., and Illia, P. (2017). Attention is all you need. arXiv.
20. Wojke, N., Bewley, A., and Paulus, D. (2017, January 17–20). Simple online and real time tracking with a deep association metric. Proceedings of the 2017 IEEE International Conference on Image Processing (ICIP), Beijing, China.10.1109/ICIP.2017.8296962 ↗
21. Bewley, A., Ge, Z., Ott, L., Ramos, F., and Upcroft, B. (2016, January 25–28). Simple online and real time tracking. Proceedings of the 2016 IEEE International Conference on Image Processing (ICIP), Phoenix, AZ, USA.10.1109/ICIP.2016.7533003 ↗
22. FasterR-CNN: Towards real-time object detection with region proposal networks10.1109/TPAMI.2016.2577031 ↗