yolov5 deepsort 行人 车辆 跟踪 检测 计数
实现了 出/入 分别计数。
默认是 南/北 方向检测,若要检测不同位置和方向,可在 main.py 文件第13行和21行,修改2个polygon的点。
默认检测类别:行人、自行车、小汽车、摩托车、公交车、卡车。
检测类别可在 detector.py 文件第60行修改。
视频
bilibili
运行环境
python 3.6+,pip 20+
pytorch
pip install -r requirements.txt
如何运行
下载代码
$ git clone https://github.com/dyh/unbox_yolov5_deepsort_counting.git
因此repo包含weights及mp4等文件,若 git clone 速度慢,可直接下载zip文件:
进入目录
$ cd unbox_yolov5_dee
2021-09-26 14:35:53
83.75MB
Python
1