| --- |
| license: gpl-3.0 |
| tags: |
| - object-detection |
| - computer-vision |
| - sort |
| - tracker |
| - strongsort |
| --- |
| |
| ### Model Description |
| [StrongSort](https://arxiv.org/abs/2202.13514): Make DeepSORT Great Again |
| <img src="https://raw.githubusercontent.com/dyhBUPT/StrongSORT/master/assets/MOTA-IDF1-HOTA.png" width="1000"/> |
|
|
| ### Installation |
| ``` |
| pip install strongsort |
| ``` |
|
|
| ### Tracker |
| ```python |
| from strong_sort import StrongSORT |
| |
| tracker = StrongSORT(model_weights='model.pt', device='cuda') |
| pred = model(img) |
| for i, det in enumerate(pred): |
| det[i] = tracker[i].update(detection, im0s) |
| ``` |
|
|
| ### BibTeX Entry and Citation Info |
| ``` |
| @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} |
| } |
| ``` |
