基于深度学习的行人目标检测与跟踪  

Pedestrian detection and tracking based on deep learning

作　　者：王芳苏 张硕 沈永良[1] WANG Fang-Su;ZHANG Shuo;SHEN Yong-Liang(School of Electronic Engineering,Heilongjiang University,Harbin 150080,China)

机构地区：[1]黑龙江大学电子工程学院,哈尔滨150080

出　　处：《黑龙江大学工程学报（中英俄文）》2023年第3期61-67,共7页Journal of Engineering of Heilongjiang University

基　　金：黑龙江省自然科学基金项目(LH2020F046)。

摘　　要：传统的目标检测跟踪方法存在实时性低、准确性差等问题。采用TBD(Tracking by Detection)方式,基于目标检测算法YOLOv5和多目标跟踪算法DeepSort,提出一种深度学习的多目标检测与跟踪方法。通过K-means聚类算法得出最佳聚类anchor大小对输入的视频帧中不同大小的目标进行检测和跟踪。基于残差结构的卷积神经网络构建行人重识别网络,在Market-1501数据集上离线训练提取行人特征。实验结果表明,算法在Crowdhuman数据集的测试集上的检测精度可达83.6,验证了检测算法的准确性;将算法应用在连续的视频帧上,可为同一个目标分配并保持相同的身份ID,在保证跟踪实时性的同时,验证了算法的可行性。Traditional target detection and tracking methods have problems such as low real-time performance and poor accuracy.Using TBD(Tracking by Detection,TBD)method,based on the target detection algorithm YOLOv5 and the multi-target tracking algorithm DeepSort,a multi-target detection and tracking method under deep learning is proposed.The K-means clustering algorithm is used to obtain the optimal cluster anchor size to detect and track targets of different sizes in the input video frame.A pedestrian re-identification network is built based on the convolutional neural network of the residual structure,and the pedestrian features are extracted by offline training on the Market-1501 dataset.The experimental results show that the detection accuracy of the algorithm on the test set of the Crowdhuman dataset can reach 83.3%,which verifies the accuracy of the detection algorithm;applying the algorithm to continuous video frames can assign and maintain the same ID for the same target,while ensuring the real-time tracking,the feasibility of the algorithm is verified.

关 键 词：深度学习 YOLOv5 多目标跟踪 行人重识别 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]