YOLOv5 + DeepSORT 实现 车辆/行人检测与跟踪

git clone https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch.git
cd Yolov5_DeepSort_Pytorch

安装依赖

pip install -r requirements.txt

用一个视频测试行人或车辆追踪：

python track.py --source video.mp4 --yolo-model yolov5s.pt --classes 0 2 5 7

参数说明：

• --source: 视频路径，也可以是摄像头（如 0）

• --yolo-model: 使用的 YOLOv5 模型权重

• --classes: 只检测特定类别（可选）

  • 0 = person

  • 2 = car

  • 5 = bus

  • 7 = truck

例如，仅跟踪人和车：

python track.py --source 0 --yolo-model yolov5s.pt --classes 0 2

你也可以在 track.py 中修改以下逻辑：

for det in detections: # det 里