https://github.com/kadirnar/yolov5-strongsort

Minimal PyTorch implementation of YOLOv5 and StrongSort
https://github.com/kadirnar/yolov5-strongsort

artificial-intelligence computer-vision deep-learning deep-sort object-detection object-tracker pytorch strongsort

Last synced: 12 days ago
JSON representation

Minimal PyTorch implementation of YOLOv5 and StrongSort

• Host: GitHub
• URL: https://github.com/kadirnar/yolov5-strongsort
• Owner: kadirnar
• License: gpl-3.0
• Created: 2022-02-08T21:19:50.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2022-09-13T20:06:08.000Z (almost 2 years ago)
• Last Synced: 2024-05-01T16:28:37.574Z (about 2 months ago)
• Topics: artificial-intelligence, computer-vision, deep-learning, deep-sort, object-detection, object-tracker, pytorch, strongsort
• Language: Python
• Homepage:
• Size: 36.6 MB
• Stars: 62
• Watchers: 3
• Forks: 15
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Lists

• awesome-yolo-object-detection - kadirnar/yolov5-strongsort - strongsort?style=social"/> : Minimal PyTorch implementation of YOLOv5 and [StrongSORT](https://github.com/dyhBUPT/StrongSORT). (Applications)

README

        




  Yolov5-StrongSort: YOLOv5-Pip + StrongSort





    





## 
Overview


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

### Installation

```

git clone https://github.com/kadirnar/yolov5-strongsort

cd yolov5-strongsort

pip install -r requirements.txt

```

### Yolov5 Model + StrongSort Prediction

```

python yolo_tracker.py --video_path test.mp4 --model_path yolov5m.pt --device cuda:0 --confidence_threshold 0.5 --image_size 640 --config_path osnet_x0_25_market1501.pt

```

## Citations

```bibtex

@article{du2022strongsort,

  title={Strongsort: Make deepsort great again},

  author={Du, Yunhao and Song, Yang and Yang, Bo and Zhao, Yanyun},

  journal={arXiv preprint arXiv:2202.13514},

  year={2022}

}

```

### Reference:

 - [Yolov5-Pip](https://github.com/fcakyon/yolov5-pip)

 - [Strongsort](https://github.com/dyhBUPT/StrongSORT)