CoolFace
Modelpublic

kadirnar/strongsort

sourceHugging Facegpl-3.0updated 4y agoView on Hugging Face
1likes
Model Card

Model Description

StrongSort: 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}
}