tracklab.wrappers.dataset.soccernet package
Submodules
tracklab.wrappers.dataset.soccernet.soccernet_game_state module
- class tracklab.wrappers.dataset.soccernet.soccernet_game_state.SoccerNetGameState(dataset_path: str, nvid: int = -1, vids_dict: list | None = None, *args, **kwargs)[source]
Bases:
TrackingDataset
- tracklab.wrappers.dataset.soccernet.soccernet_game_state.dict_to_df_detections(annotation_dict, categories_list)[source]
- tracklab.wrappers.dataset.soccernet.soccernet_game_state.download_dataset(dataset_path, splits=('train', 'valid', 'test', 'challenge'))[source]
tracklab.wrappers.dataset.soccernet.soccernet_mot module
- class tracklab.wrappers.dataset.soccernet.soccernet_mot.SoccerNetMOT(dataset_path: str, nvid: int = -1, vids_dict: list | None = None, *args, **kwargs)[source]
Bases:
TrackingDataset