diff --git a/video_llama/datasets/datasets/webvid_datasets.py b/video_llama/datasets/datasets/webvid_datasets.py index aaf6b9d6..4c624db6 100644 --- a/video_llama/datasets/datasets/webvid_datasets.py +++ b/video_llama/datasets/datasets/webvid_datasets.py @@ -40,7 +40,7 @@ def __init__(self, vis_processor, text_processor, vis_root, ann_root): self.frm_sampling_strategy = 'headtail' def _get_video_path(self, sample): - rel_video_fp = os.path.join(sample['page_dir'], str(sample['videoid']) + '.mp4') + rel_video_fp = os.path.join(str(sample['page_dir']), str(sample['videoid']) + '.mp4') full_video_fp = os.path.join(self.vis_root, rel_video_fp) return full_video_fp