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

process_trackeval_results(results, dataset_config, eval_config)[source]
save_for_eval(detections: DataFrame, image_metadatas: DataFrame, video_metadatas: DataFrame, save_folder: str, bbox_column_for_eval='bbox_ltwh', save_classes=False, is_ground_truth=False, save_zip=True)[source]

Save predictions in MOT Challenge format.

static soccernet_encoding(dataframe: DataFrame, supercategory)[source]
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_game_state.extract_category(attributes)[source]
tracklab.wrappers.dataset.soccernet.soccernet_game_state.load_set(dataset_path, nvid=-1, vids_filter_set=None)[source]
tracklab.wrappers.dataset.soccernet.soccernet_game_state.read_json_file(file_path)[source]
tracklab.wrappers.dataset.soccernet.soccernet_game_state.transform_bbox_image(row)[source]
tracklab.wrappers.dataset.soccernet.soccernet_game_state.video_dir_to_dfs(args)[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

tracklab.wrappers.dataset.soccernet.soccernet_mot.load_set(dataset_path, nvid=-1, vids_filter_set=None)[source]
tracklab.wrappers.dataset.soccernet.soccernet_mot.read_ini_file(file_path)[source]
tracklab.wrappers.dataset.soccernet.soccernet_mot.read_motchallenge_formatted_file(file_path)[source]