Spaces:

leoandeol
/

cods-live

Sleeping

Léo Andéol commited on 19 days ago

Commit

9c3ff92

unverified ·

1 Parent(s): fcce8eb

Files changed (1) hide show

dataset.py CHANGED Viewed

@@ -47,7 +47,7 @@ class DatasetWrapper(ClassificationDataset):
     def __getitem__(self, idx):
         item = self.dataset[idx]
-        print(item)
         img, label = item["image"], item["label"]
         if img.mode != "RGB":
@@ -55,4 +55,4 @@ class DatasetWrapper(ClassificationDataset):
         # img = Image.open(path)
         if self.transforms:
             img = self.transforms(img)
-        return img, label

     def __getitem__(self, idx):
         item = self.dataset[idx]
+        # print(item)
         img, label = item["image"], item["label"]
         if img.mode != "RGB":
         # img = Image.open(path)
         if self.transforms:
             img = self.transforms(img)
+        return idx, img, label

More