Update dataset/datasets.py

dataset/datasets.py

@@ -21,12 +21,12 @@ class TrainDataset(torch.utils.data.Dataset):
 
     def __getitem__(self, idx):
         image_name = self.data[idx]
-
-    try:
-        color_img = plt.imread(os.path.join(self.data_path, "color", image_name))
-    except SyntaxError:
-        print(f"Archivo {image_name} no es un PNG válido. Saltando...")
-        return None  # O alguna otra acción que prefieras
+        
+        try:
+            color_img = plt.imread(os.path.join(self.data_path, 'color', image_name))
+        except SyntaxError:
+            print(f"Archivo {image_name} no es un PNG válido. Saltando...")
+            return None  # O alguna otra acción que prefieras
 
         if self.mults_amount > 1:
             mult_number = np.random.choice(range(self.mults_amount))