Skip to content

IOU based object tracking (SORT, ByteTrack) Python library implemented in Rust

License

Notifications You must be signed in to change notification settings

PaulKlinger/ioutrack

Repository files navigation

IOU Track

Actions Status PyPI

Python package for IOU-based tracking (SORT & ByteTrack) written in Rust.

from ioutrack import Sort

tracker = Sort(max_age=5, min_hits=2)

#                   xmin ymin xmax ymax score
boxes_0 = np.array([[10., 60., 50., 95., 0.8],...])
tracks_0 = tracker.update(boxes_0)

#                            xmin ymin xmax ymax track_id
assert tracks_0 == np.array([[10., 60., 50., 95., 1.],...])

Demo video: https://youtu.be/BLMnY8K9HBE
Code to generate the demo video is in the demo folder.

Roughly 30x faster than python/numpy implementation.