Skip to content

tudobf/unbox_yolov5_deepsort_counting

This branch is up to date with dyh/unbox_yolov5_deepsort_counting:main.

Folders and files

NameName
Last commit message
Last commit date

Latest commit

May 27, 2021
391f724 · May 27, 2021

History

12 Commits
Feb 12, 2021
Feb 12, 2021
Apr 13, 2021
Apr 13, 2021
Feb 12, 2021
Apr 13, 2021
Feb 12, 2021
May 27, 2021
May 27, 2021
Feb 12, 2021
May 27, 2021
Feb 12, 2021
Apr 13, 2021
May 27, 2021

Repository files navigation

yolov5 deepsort 行人 车辆 跟踪 检测 计数

  • 实现了 出/入 分别计数。
  • 显示检测类别。
  • 默认是 南/北 方向检测,若要检测不同位置和方向,可在 main.py 文件第13行和21行,修改2个polygon的点。
  • 默认检测类别:行人、自行车、小汽车、摩托车、公交车、卡车。
  • 检测类别可在 detector.py 文件第60行修改。

视频

bilibili

bilibili

运行环境

  • python 3.6+,pip 20+
  • pytorch
  • pip install -r requirements.txt

如何运行

  1. 下载代码

    $ 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

  2. 进入目录

    $ cd unbox_yolov5_deepsort_counting
    
  3. 创建 python 虚拟环境

    $ python3 -m venv venv
    
  4. 激活虚拟环境

    $ source venv/bin/activate
    
  5. 升级pip

    $ python -m pip install --upgrade pip
    
  6. 安装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
    
  7. 安装软件包

    $ pip install -r requirements.txt
    
  8. 在 main.py 文件中第66行,设置要检测的视频文件路径,默认为 './video/test.mp4'

    140MB的测试视频可以在这里下载:https://pan.baidu.com/s/1qHNGGpX1QD6zHyNTqWvg1w 提取码: 8ufq

    capture = cv2.VideoCapture('./video/test.mp4')
    
  9. 运行程序

    python main.py
    

使用框架

About

yolov5 deepsort 行人 车辆 跟踪 检测 计数

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.8%
  • Other 1.2%