yolov5 deepsort 行人 车辆 跟踪 检测 计数
下载地址:基于YOLOv5行人车辆跟踪检测识别计数系统源码+数据集
实现了 出/入 分别计数。默认是 南/北 方向检测,若要检测不同位置和方向,可在 main.py 文件第13行和21行,修改2个polygon的点。默认检测类别:小客车、大客车、小货车、中货车、大货车、集装箱车。检测类别可在 detector.py 文件第60行修改。
运行环境
python 3.6+,pip 20+pytorchpip 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版本,在 /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 /whl/torch_stable.html
安装软件包
$ pip install -r requirements.txt
在 main.py 文件中第66行,设置要检测的视频文件路径,默认为 ‘./video/test.mp4’
140MB的测试视频可以在这里下载:/s/1geqjht-no0iyzQ88JQopwA 密码: i6cs
capture = cv2.VideoCapture('./video/test.mp4')
运行程序
python main.py
引用
/Sharpiless/Yolov5-deepsort-inference/ultralytics/yolov5//ZQPei/deep_sort_pytorch=======
下载地址:基于YOLOv5行人车辆跟踪检测识别计数系统源码+数据集