File size: 827 Bytes
6d1e6ee 4b268e6 6d1e6ee 4b268e6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
---
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}
}
``` |