|
--- |
|
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} |
|
} |
|
``` |