From 6e2bd5f7e70d2707b146c5c72145cdb4c77c6b16 Mon Sep 17 00:00:00 2001 From: Anirudh Thatipelli Date: Fri, 5 Jan 2024 11:33:21 -0800 Subject: [PATCH] Fix ptv_dataset_helper * Loading the "clips" in annotation directly. --- ego4d_forecasting/datasets/ptv_dataset_helper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ego4d_forecasting/datasets/ptv_dataset_helper.py b/ego4d_forecasting/datasets/ptv_dataset_helper.py index eb2143b..b7ea133 100644 --- a/ego4d_forecasting/datasets/ptv_dataset_helper.py +++ b/ego4d_forecasting/datasets/ptv_dataset_helper.py @@ -358,7 +358,7 @@ def clip_recognition_dataset( if g_pathmgr.isfile(data_path): try: with g_pathmgr.open(data_path, "r") as f: - annotations = json.load(f) + annotations = json.load(f)['clips'] except Exception: raise FileNotFoundError(f"{data_path} must be json for Ego4D dataset")