代码拉取完成,页面将自动刷新
bilibili
下载代码
$ git clone https://github.com/dyh/unbox_yolov5_deepsort_counting.git
因此repo包含weights及mp4等文件,若 git clone 速度慢,可直接下载zip文件:https://github.com/dyh/unbox_yolov5_deepsort_counting/archive/main.zip
进入目录
$ cd unbox_yolov5_deepsort_counting
创建 python 虚拟环境
$ python3 -m venv venv
激活虚拟环境
$ source venv/bin/activate
升级pip
$ python -m pip install --upgrade pip
安装pytorch
根据你的操作系统、安装工具以及CUDA版本,在 https://pytorch.org/get-started/locally/ 找到对应的安装命令。我的环境是 ubuntu 18.04.5、pip、CUDA 11.0。
$ pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
安装软件包
$ pip install -r requirements.txt
在 main.py 文件中第66行,设置要检测的视频文件路径,默认为 './video/test.mp4'
140MB的测试视频可以在这里下载:https://pan.baidu.com/s/1geqjht-no0iyzQ88JQopwA 密码: i6cs
capture = cv2.VideoCapture('./video/test.mp4')
运行程序
python main.py
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。