Skip to content

Commit

Permalink
Correct tracker_state when not loading from file or gt
Browse files Browse the repository at this point in the history
  • Loading branch information
victorjoos committed Feb 16, 2024
1 parent b48fd86 commit 85546ce
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tracklab/datastruct/tracker_state.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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[
Expand Down

0 comments on commit 85546ce

Please sign in to comment.