基于YOLOv5行人车辆跟踪检测识别计数系统源码+数据集,实现出/入分别计数
yolov5 deepsort 行人 车辆 跟踪 检测 计数
运行环境
- python 3.6+,pip 20+
- pytorch
- pip install -r requirements.txt
如何运行
- 进入目录
$ 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