/detectron2-deepsort-pytorch

MOT tracking using deepsort and yolov3 with pytorch

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Deep Sort with PyTorch

Detectron2 version of DeepSORT PyTorch

Demo

  1. Clone this repository: git clone --recurse-submodules https://github.com/sayef/detectron2-deepsort-pytorch.git
  2. Install detectron2: cd detectron2-deepsort-pytorch and pip install -e detectron2/
  3. Install deepsort requirements: pip install -r requirements.txt
  4. Run the demo: python demo_detectron2_deepsort.py path/to/example_video.avi

Notes:

  1. Try out different detectron2 models: Change the configs in ___init__ of detectron2_detection.py
  2. Regarding any issues of detectron2, please refer to Detectron2 repository.

References and Credits:

  1. Pytorch implementation of deepsort with Yolo3
  2. Facebook Detectron2
  3. Deepsort
  4. SORT