From fd1681d584c7d9e169f435c8d6af5e71e1f29433 Mon Sep 17 00:00:00 2001 From: Victor Joos Date: Mon, 19 Feb 2024 14:41:14 +0100 Subject: [PATCH 1/2] Fix loading from GT only --- tracklab/datastruct/tracker_state.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tracklab/datastruct/tracker_state.py b/tracklab/datastruct/tracker_state.py index 844eae03..d54e40f3 100644 --- a/tracklab/datastruct/tracker_state.py +++ b/tracklab/datastruct/tracker_state.py @@ -296,6 +296,7 @@ def load(self): video_detections = pd.DataFrame() video_image_preds = self.image_metadatas[self.image_metadatas.video_id == self.video_id] if self.load_from_groundtruth: + video_detections = self.detections_pred[self.detections_pred.video_id == self.video_id] video_image_preds = self.image_pred[self.image_pred.video_id == self.video_id] if self.load_file is not None: if f"{self.video_id}.pkl" in self.zf["load"].namelist(): From a96d7ddeb49f8594260fc31ec37cd1dc475dcc77 Mon Sep 17 00:00:00 2001 From: Victor Joos Date: Mon, 19 Feb 2024 16:15:47 +0100 Subject: [PATCH 2/2] allow lack of labels in other sets than challenge --- tracklab/wrappers/datasets/soccernet/soccernet_game_state.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tracklab/wrappers/datasets/soccernet/soccernet_game_state.py b/tracklab/wrappers/datasets/soccernet/soccernet_game_state.py index 176e0efc..31debf44 100644 --- a/tracklab/wrappers/datasets/soccernet/soccernet_game_state.py +++ b/tracklab/wrappers/datasets/soccernet/soccernet_game_state.py @@ -111,7 +111,7 @@ def video_dir_to_dfs(args): video_level_categories = [] video_folder_path = os.path.join(dataset_path, video_folder) if os.path.isdir(video_folder_path): - if split == "challenge": + if not (Path(video_folder_path) / "Labels-GameState.json").exists(): img_folder_path = os.path.join(video_folder_path, 'img1') video_id = str(int(video_folder.split('-')[-1])) video_metadata = { @@ -225,7 +225,7 @@ def load_set(dataset_path, nvid=-1, vids_filter_set=None): annotations_pitch_camera_list.append(result["annotations_pitch_camera"]) categories_list += result["video_level_categories"] - if split == "challenge": + if len(categories_list) == 0: video_metadata = pd.DataFrame(video_metadatas_list) image_metadata = pd.concat(image_metadata_list, ignore_index=True) detections = None