基于YOLOv5行人车辆跟踪检测识别计数系统源码+数据集，实现出/入分别计数

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

下载地址：基于YOLOv5行人车辆跟踪检测识别计数系统源码+数据集

• 实现了 出/入 分别计数。
• 默认是 南/北 方向检测，若要检测不同位置和方向，可在 main.py 文件第13行和21行，修改2个polygon的点。
• 默认检测类别：小客车、大客车、小货车、中货车、大货车、集装箱车。
• 检测类别可在 detector.py 文件第60行修改。

运行环境

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

如何运行

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/1geqjht-no0iyzQ88JQopwA 密码: i6cs

   capture = cv2.VideoCapture('./video/test.mp4')
   

9. 运行程序

   python main.py
   

引用

• https://github.com/Sharpiless/Yolov5-deepsort-inference
• https://github.com/ultralytics/yolov5/
• https://github.com/ZQPei/deep_sort_pytorch
  =======
  下载地址：基于YOLOv5行人车辆跟踪检测识别计数系统源码+数据集