sportslabkit.mot.base#

Overview#

Classes#

Callback

Base class for creating new callbacks.

MultiObjectTracker

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

Function#

with_callbacks(func)

Decorator for wrapping methods that require callback invocations.

Classes#

class sportslabkit.mot.base.Callback[source]#

Base class for creating new callbacks.

This class defines the basic structure of a callback and allows for dynamic method creation for handling different events in the Trainer’s lifecycle.

__getattr__(name

str) -> callable: Returns a dynamically created method based on the given name.

class sportslabkit.mot.base.MultiObjectTracker(window_size=1, step_size=None, max_staleness=5, min_length=5, callbacks=None)[source]#

Bases: abc.ABC

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

Overview

Methods#

update_tracklet(tracklet, states)

-

update(current_frame, trackelts)

abc -

process_sequence_item(sequence)

-

track(sequence)

-

track_sequence(sequence)

-

cleanup_tracklets(tracklets)

-

increment_staleness(tracklets)

-

reset_staleness(tracklets)

-

pre_track()

-

post_track()

-

reset()

-

check_updated_state(state)

-

create_tracklet(state)

-

to_bbdf()

Create a bounding box dataframe.

separate_stale_tracklets(unassigned_tracklets)

-

create_hparam_dict()

-

get_new_hyperparameters(hparams, trial)

-

apply_hyperparameters(params)

-

tune_hparams(frames_list, bbdf_gt_list, n_trials, hparam_search_space, verbose, return_study, use_bbdf, reuse_detections, sampler, pruner)

-

Members

update_tracklet(tracklet: sportslabkit.Tracklet, states: dict[str, Any])[source]#
abstract update(current_frame: Any, trackelts: list[sportslabkit.Tracklet]) tuple[list[sportslabkit.Tracklet], list[dict[str, Any]]][source]#
process_sequence_item(sequence: Any)[source]#
track(sequence: Iterable[Any] | np.ndarray) sportslabkit.Tracklet[source]#
track_sequence(sequence)[source]#
cleanup_tracklets(tracklets)[source]#
increment_staleness(tracklets)[source]#
reset_staleness(tracklets)[source]#
pre_track()[source]#
post_track()[source]#
reset()[source]#
check_updated_state(state: dict[str, Any])[source]#
create_tracklet(state: dict[str, Any])[source]#
to_bbdf()[source]#

Create a bounding box dataframe.

separate_stale_tracklets(unassigned_tracklets)[source]#
create_hparam_dict()[source]#
get_new_hyperparameters(hparams, trial)[source]#
apply_hyperparameters(params)[source]#
tune_hparams(frames_list, bbdf_gt_list, n_trials=100, hparam_search_space=None, verbose=False, return_study=False, use_bbdf=False, reuse_detections=False, sampler=None, pruner=None)[source]#

Functions#

sportslabkit.mot.base.with_callbacks(func)[source]#

Decorator for wrapping methods that require callback invocations.

Parameters:

func (callable) – The method to wrap.

Returns:

The wrapped method.

Return type:

callable