基于yolov5和deepsort的行人车辆的检测跟踪和计数（课程设计）

Posted 2023-01-02 踟蹰横渡口,彳亍上滩舟。

- 实现了行人出入分别计数，沿着图像上下方向检测。

- 检测类别：行人、自行车、小汽车、摩托车、公交车、卡车。

-支持yolov5s.pt yolov5x.pt yolov5m.pt yolov5l.pt模型

代码、模型和demo视频下载地址：下载地址

环境配置：

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

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

运行命令：

python main.py

部分源码解读：

import numpy as np

import tracker
from detector import Detector
import cv2

if __name__ == \'__main__\'