Skip to content

jamjamjon/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.

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Rust 87.0%
  • Python 13.0%