快速使用Yolov5_DeepSort_Pytorch

jupiter
2022-03-16 / 0 评论 / 753 阅读 / 正在检测是否收录...
温馨提示:
本文最后更新于2022年03月16日,已超过822天没有更新,若内容或图片失效,请留言反馈。

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行人模型的方法
0

评论 (0)

打卡
取消