Pedestrian Target Tracking Based On DeepSORT With YOLOv5

被引:21
作者
Gai, Yuqiao [1 ]
He, Weiyang [2 ]
Zhou, Zilong [3 ]
机构
[1] Nanchang Univ, Sch Software, Nanchang, Jiangxi, Peoples R China
[2] Fujian Univ Technol, Coll Ecol Environm & Urban Construct, Fuzhou, Peoples R China
[3] Tongji Univ, Sch Software Engn, Shanghai, Peoples R China
来源
2021 2ND INTERNATIONAL CONFERENCE ON COMPUTER ENGINEERING AND INTELLIGENT CONTROL (ICCEIC 2021) | 2021年
关键词
YOLOv5; DeepSORT; deep learning; MOT;
D O I
10.1109/ICCEIC54227.2021.00008
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Pedestrian target tracking is an important problem in the field of computer vision. To address low tracking accuracy and tracking errors in pedestrian target tracking. This paper developed a YOLOv5-based DeepSORT pedestrian target tracking algorithm (YOLOv5-DeepSORT), which introduces the high-performing YOLOv5 algorithm into the DeepSORT algorithm, which detects the tracking video frame by frame, and then predicts the target position using Kalman filtering while matching the same target using the Hungarian algorithm. To verify the performance of the YOLOv5-DeepSORT algorithm, this paper compares it with the YOLOv3-DeepSORT algorithm for experiments. The results show that the YOLOv5-DeepSORT algorithm can detect and track pedestrians well. Compared with the YOLOv3-DeepSORT algorithm, the MOTA of the YOLOv5-DeepSORT algorithm is improved by 9.9%, the number of false detections and missed detections is significantly reduced, and the number of target ID switching is decreased by 23 times, which proves the effectiveness of the method in this paper.
引用
收藏
页码:1 / 5
页数:5
相关论文
共 7 条
[1]  
Barron J L., 1992, COMP VIS PATT REC 19
[2]  
Bewley A, 2016, IEEE IMAGE PROC, P3464, DOI 10.1109/ICIP.2016.7533003
[3]  
Chen J.Q., 2021, Acta Metrologica Sinica, V06, P718
[4]  
Chen Y., 2020, Computer Application Research, V37, P311
[5]  
Chen Z.Q., 2021, J. Guilin Univ. Electron. Technol, V41, P140
[6]  
WANG Z, 2007, CAMSHIFT GUIDED PART, P301
[7]  
Wojke N, 2017, IEEE IMAGE PROC, P3645, DOI 10.1109/ICIP.2017.8296962