kadirnar commited on
Commit
4b268e6
1 Parent(s): 6d1e6ee

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +35 -0
README.md CHANGED
@@ -1,3 +1,38 @@
1
  ---
2
  license: gpl-3.0
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: gpl-3.0
3
+ tags:
4
+ - object-detection
5
+ - computer-vision
6
+ - sort
7
+ - tracker
8
+ - strongsort
9
  ---
10
+
11
+ ### Model Description
12
+ [StrongSort](https://arxiv.org/abs/2202.13514): Make DeepSORT Great Again
13
+ <img src="https://raw.githubusercontent.com/dyhBUPT/StrongSORT/master/assets/MOTA-IDF1-HOTA.png" width="1000"/>
14
+
15
+ ### Installation
16
+ ```
17
+ pip install strongsort
18
+ ```
19
+
20
+ ### Tracker
21
+ ```python
22
+ from strong_sort import StrongSORT
23
+
24
+ tracker = StrongSORT(model_weights='model.pt', device='cuda')
25
+ pred = model(img)
26
+ for i, det in enumerate(pred):
27
+ det[i] = tracker[i].update(detection, im0s)
28
+ ```
29
+
30
+ ### BibTeX Entry and Citation Info
31
+ ```
32
+ @article{du2022strongsort,
33
+ title={Strongsort: Make deepsort great again},
34
+ author={Du, Yunhao and Song, Yang and Yang, Bo and Zhao, Yanyun},
35
+ journal={arXiv preprint arXiv:2202.13514},
36
+ year={2022}
37
+ }
38
+ ```