Update youtubevis.py

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
+        return imgs, segs