Packaged version of the Strongsort repository
Project description
Overview
This repo is a packaged version of the StrongSort algorithm.
Installation
pip install strongsort
Detection Model + StrongSort
from strong_sort import StrongSORT
tracker = StrongSORT(model_weights='model.pth', device='cuda')
pred = model(img)
for i, det in enumerate(pred):
det[i] = tracker[i].update(detection, im0s)
Citations
@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}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
strongsort-0.2.1.tar.gz
(61.2 kB
view hashes)