Update youtubevis.py

PeterL1n · Jan 14, 2022 · 81a1093 · 81a1093
1 parent f8a26e2
commit 81a1093
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dataset/youtubevis.py b/dataset/youtubevis.py
@@ -38,7 +38,7 @@ def __init__(self, videodir, annfile, size, seq_length, seq_sampler, transform=N
 
         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 @@ def __call__(self, imgs, segs):
             imgs = F.hflip(imgs)
             segs = F.hflip(segs)
 
-        return imgs, segs
+        return imgs, segs