1.项目地址

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行人模型的方法
Last modification:March 16th, 2022 at 03:08 pm
如果觉得我的文章对你有用,请随意赞赏