animecomico / yolov5-deepsort

Minimal PyTorch implementation of YOLOv5 and DeepSort

Yolov5-DeepSort: Minimal YOLOv5 + Deep Sort

Overview

This repo is a shortened version of yolov5 codes and added deep sort algorithm.

Installation

git clone https://github.com/kadirnar/yolov5-lite
cd yolov5-lite
pip install -r requirements.txt

Yolov5 Model + Deep Sort Prediction

python detect.py --source ../video/uav.mp4 --yolo_model yolov5x6.pt --deep_sort_model osnet_x1_0 --show-vid

Citations

@misc{yolov5deepsort2020,
    title={Real-time multi-object tracker using YOLOv5 and deep sort},
    author={Mikel Broström},
    howpublished = {\url{https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch}},
    year={2020}
}

Reference:

• YOLOv5
• Deep Sort