From 81a10937c73f68eeddb863221c61fe6a60a1cca2 Mon Sep 17 00:00:00 2001 From: Peter Lin Date: Thu, 13 Jan 2022 22:36:37 -0800 Subject: [PATCH] Update youtubevis.py --- dataset/youtubevis.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dataset/youtubevis.py b/dataset/youtubevis.py index babbf0e..8abe40a 100644 --- a/dataset/youtubevis.py +++ b/dataset/youtubevis.py @@ -38,7 +38,7 @@ class YouTubeVISDataset(Dataset): self.index = [] for video_id in self.videos.keys(): - for frame in range(len(self.videos[video_id])): + for frame in range(len(self.videos[video_id]['file_names'])): self.index.append((video_id, frame)) def __len__(self): @@ -120,4 +120,4 @@ class YouTubeVISAugmentation: imgs = F.hflip(imgs) segs = F.hflip(segs) - return imgs, segs \ No newline at end of file + return imgs, segs