登录Anaconda官网下载适合自己版本的anaconda
安装后,在命令窗口输入conda -V
检测Anaconda版本,查看自己的环境是否安装成功
为方便环境的管理,打开Anaconda Prompt,创建pytorch虚拟环境
conda create -n pytorch python=3.6
可以根绝自己的python版本进行更换
激活虚拟环境
conda activate pytorch
登录pytorch官网
根据自己情况填写后,获取相应的安装指令。自己的虚拟环境中通过安装指令安装pytorch。
我选择的是没有cpu的pytorch1.8.1。安装指令是
pip3 install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio===0.8.1 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html
在Github中clone项目Yolov5_DeepSort_Pytorch,因国内访问Github比较慢,CSDN下载链接。
在Anaconda Prompt下,通过cd命令进入到项目文件夹下,执行安装指令
pip install -r require ments.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
下载YOLOV5,在Github的DeepSort-Object-Tracking-PyTorch项目中将YOLOV5文件夹复制到Yolov5_DeepSort_Pytorch项目中(原本的Yolov5_DeepSort_Pytorch项目中yolov5文件夹是空的没有相应的文件),CSDN下载链接
下载ckpt.t7
将ckpt.t7放到\deep_sort_pytorch\deep_sort\deep\checkpoint
下载一个最小的模型yolov5s.pt。如果速度太慢,请转yolov5s.pt,将模型文件复制到\yolov5\weights
路径
可以自己找行人视频也在、可以根据下载链接下载本项目视频,将测试视频置于项目路径
在Anaconda Prompt中运行命令
python track.py --source ./Demo.mp4 --show-vid --save-vid --yolo_weights ./yolov5/weights/yolov5s.pt