1.项目地址

• https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch

2.运行步骤

2.1 clone项目+ 配置环境

git clone --recurse-submodules https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch.git

pip install -r requirements.txt

2.2训练YOLOv5子模块

参考资料：快速使用YOLOv5进行训练VOC格式的数据集

2.3 安装Torchreid

cd deep_sort/deep
rm -r reid
git clone https://github.com/KaiyangZhou/deep-person-reid.git
mv deep-person-reid reid
cd reid
python setup.py develop

2.4 调用

python track.py --source vid.mp4 --yolo_model yolov5s.pt --deep_sort_model osnet_x1_0 --save-txt --save-vid

参考资料

1. https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch
2. https://github.com/KaiyangZhou/deep-person-reid.git
3. 新版Yolov5_DeepSort_Pytorch使用ZQPei行人模型的方法