|
|
@ -38,7 +38,7 @@ class YouTubeVISDataset(Dataset):
|
|
|
|
|
|
|
|
|
|
|
|
self.index = []
|
|
|
|
self.index = []
|
|
|
|
for video_id in self.videos.keys():
|
|
|
|
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))
|
|
|
|
self.index.append((video_id, frame))
|
|
|
|
|
|
|
|
|
|
|
|
def __len__(self):
|
|
|
|
def __len__(self):
|
|
|
@ -120,4 +120,4 @@ class YouTubeVISAugmentation:
|
|
|
|
imgs = F.hflip(imgs)
|
|
|
|
imgs = F.hflip(imgs)
|
|
|
|
segs = F.hflip(segs)
|
|
|
|
segs = F.hflip(segs)
|
|
|
|
|
|
|
|
|
|
|
|
return imgs, segs
|
|
|
|
return imgs, segs
|
|
|
|