sportslabkit.mot.bytetrack#

Overview#

Classes#

class sportslabkit.mot.bytetrack.BYTETracker(detection_model=None, image_model=None, motion_model=None, first_matching_fn: sportslabkit.matching.MotionVisualMatchingFunction = MotionVisualMatchingFunction(motion_metric=IoUCMM(use_pred_box=True), motion_metric_gate=0.2, visual_metric=CosineCMM(), visual_metric_gate=0.2, beta=0.5), second_matching_fn=SimpleMatchingFunction(metric=IoUCMM(use_pred_box=True), gate=0.9), detection_score_threshold=0.6, window_size: int = 1, step_size: int | None = None, max_staleness: int = 5, min_length: int = 5, callbacks=None)[source]#

Bases: sportslabkit.mot.base.MultiObjectTracker

BYTE tracker from https://arxiv.org/pdf/2110.06864.pdf

Overview

Methods#

update(current_frame, tracklets)

-

Members

update(current_frame, tracklets)[source]#