Spaces:

naver
/

SuperFeatures

Build error

YannisK commited on Apr 20, 2022

Commit

3a6d8a8

1 Parent(s): 26bd1c5

edits

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,7 +48,9 @@ class ImgDataset(data.Dataset):
     def __getitem__(self, index):
         img = self.images[index]
-        return  self.transform(img.thumbnail((self.imsize, self.imsize), Image.ANTIALIAS))
     def __len__(self):

     def __getitem__(self, index):
         img = self.images[index]
+        img.thumbnail((self.imsize, self.imsize), Image.ANTIALIAS)
+        print(img.size)
+        return  self.transform(img)
     def __len__(self):