From 85546ce5cea3c6501d705e149cfad6c6b6b5cabb Mon Sep 17 00:00:00 2001 From: Victor Joos Date: Fri, 16 Feb 2024 10:35:39 +0100 Subject: [PATCH] Correct tracker_state when not loading from file or gt --- tracklab/datastruct/tracker_state.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tracklab/datastruct/tracker_state.py b/tracklab/datastruct/tracker_state.py index 0bb7f4c7..844eae03 100644 --- a/tracklab/datastruct/tracker_state.py +++ b/tracklab/datastruct/tracker_state.py @@ -293,9 +293,10 @@ def load(self): return self.json_detections_pred[ self.json_detections_pred.video_id == self.video_id ] - image_preds = self.image_metadatas[self.image_metadatas.video_id == self.video_id] + video_detections = pd.DataFrame() + video_image_preds = self.image_metadatas[self.image_metadatas.video_id == self.video_id] if self.load_from_groundtruth: - image_preds = self.image_pred[self.image_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(): with self.zf["load"].open(f"{self.video_id}.pkl", "r") as fp: @@ -306,7 +307,7 @@ def load(self): if f"{self.video_id}_image.pkl" in self.zf["load"].namelist(): with self.zf["load"].open(f"{self.video_id}_image.pkl", "r") as fp_image: video_image_preds = merge_dataframes( - pickle.load(fp_image), image_preds + pickle.load(fp_image), video_image_preds )[self.load_columns["image"]] else: video_image_preds = self.image_metadatas[