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