overwrite with WIP changes: src/dataset.py

UCPE/src/dataset.py

@@ -91,7 +91,8 @@ class PanShotDataset(Dataset):
         data = self.metas[idx].copy()
 
         video_id = data["video_id"]
-        video_path = self.data_root / "PanShot" / f"videos-{self.split}" / f"{video_id}.mp4"
+        video_subdir = getattr(self.args, "video_subdir", "videos")
+        video_path = self.data_root / "PanShot" / f"{video_subdir}-{self.split}" / f"{video_id}.mp4"
         data["video_path"] = str(video_path)
         data["result_path"] = data["video_path"] if self.result_root is None else str(self.result_root / f"{video_id}.mp4")