sportslabkit.sot#

Overview#

Classes#
`SingleObjectTracker`	Helper class that provides a standard way to create an ABC using
`HungarianTracker`	Helper class that provides a standard way to create an ABC using
`MeanShiftTracker`	Helper class that provides a standard way to create an ABC using
`MinimumCostFlowTracker`	Helper class that provides a standard way to create an ABC using

Classes#

class sportslabkit.sot.SingleObjectTracker(target, window_size=1, step_size=None, pre_init_args={}, post_init_args={})[source]#

Bases: abc.ABC

Helper class that provides a standard way to create an ABC using inheritance.

Overview

Methods#
`pre_initialize`(**kwargs)	-
`post_initialize`(**kwargs)	-
`update_tracklet_observations`(states)	-
`update`(current_frame)	abc -
`process_sequence_item`(sequence)	-
`track`(sequence)	-
`pre_track`()	-
`post_track`()	-
`reset`()	-
`check_required_types`(target)	-
`check_updated_state`(state)	-
`create_hparam_dict`()	-
`tune_hparams`(frames, ground_truth_positions, n_trials, hparam_search_space, metric, verbose, return_study)	-

Members

pre_initialize(**kwargs)#

post_initialize(**kwargs)#

update_tracklet_observations(states: dict[str, Any])#

abstract update(current_frame: Any) → dict[str, Any]#

process_sequence_item(sequence: Any)#

track(sequence: Iterable[Any] | np.ndarray) → sportslabkit.Tracklet#

pre_track()#

post_track()#

reset()#

check_required_types(target: dict[str, Any])#

check_updated_state(state: dict[str, Any])#

create_hparam_dict()#

tune_hparams(frames, ground_truth_positions, n_trials=100, hparam_search_space=None, metric=iou_scores, verbose=False, return_study=False)#

class sportslabkit.sot.HungarianTracker(target, initial_frame, detection_model=None, image_model=None, motion_model=None, matching_fn=None)[source]#

Bases: sportslabkit.sot.base.SingleObjectTracker

Helper class that provides a standard way to create an ABC using inheritance.

Overview

Methods#
`pre_initialize`(initial_frame, detection_model, image_model, motion_model, matching_fn)	-
`update`(current_frame)	-

Members

pre_initialize(initial_frame, detection_model, image_model, motion_model, matching_fn)#

update(current_frame)#

class sportslabkit.sot.MeanShiftTracker(target, initial_frame, bins=16, max_iterations=10, termination_eps=1, *args, **kwargs)[source]#

Bases: sportslabkit.sot.base.SingleObjectTracker

Helper class that provides a standard way to create an ABC using inheritance.

Overview

Attributes#
`required_keys`	-

Methods#
`pre_initialize`(initial_frame, bins, max_iterations, termination_eps)	-
`update`(current_frame)	-

Members

required_keys = ['box']#

pre_initialize(initial_frame, bins=16, max_iterations=10, termination_eps=1)#

update(current_frame)#

class sportslabkit.sot.MinimumCostFlowTracker(target, initial_frame, detection_model=None, image_model=None, motion_model=None, matching_fn=None, window_size=10)[source]#

Bases: sportslabkit.sot.base.SingleObjectTracker

Helper class that provides a standard way to create an ABC using inheritance.

Overview

Methods#
`pre_initialize`(initial_frame, detection_model, image_model, motion_model, matching_fn)	-
`update`(sequence)	-

Members

pre_initialize(initial_frame, detection_model, image_model, motion_model, matching_fn)#

update(sequence)#