代码

https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch
链接：https://pan.baidu.com/s/1QctrRuIYowR1lq14V3Q4SQ
提取码：9780

环境

conda create -n py38_torch python=3.8
# CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
pip install -r requirements.txt

使用

python track.py --source ./test.mp4 --view-vid --save-vid --weights ./yolov5/weights/yolov5l.pt

效果