fix data load 'None'

Author: likyoo

change_detection_pytorch/datasets/LEVIR_CD.py

@@ -50,12 +50,12 @@ def __getitem__(self, idx):
             img1, img2, filename = self.prepare_img(idx)
             transformed_data = self.transform(image=img1, image_2=img2)
             img1, img2 = transformed_data['image'], transformed_data['image_2']
+            return img1, img2, filename
         else:
             img1, img2, ann, filename = self.prepare_img_ann(idx)
             transformed_data = self.transform(image=img1, image_2=img2, mask=ann)
             img1, img2, ann = transformed_data['image'], transformed_data['image_2'], transformed_data['mask']
-
-        return img1, img2, ann, filename
+            return img1, img2, ann, filename
 
 
 if __name__ == "__main__":

change_detection_pytorch/datasets/SVCD.py

@@ -49,9 +49,9 @@ def __getitem__(self, idx):
             img1, img2, filename = self.prepare_img(idx)
             transformed_data = self.transform(image=img1, image_2=img2)
             img1, img2 = transformed_data['image'], transformed_data['image_2']
+            return img1, img2, filename
         else:
             img1, img2, ann, filename = self.prepare_img_ann(idx)
             transformed_data = self.transform(image=img1, image_2=img2, mask=ann)
             img1, img2, ann = transformed_data['image'], transformed_data['image_2'], transformed_data['mask']
-
-        return img1, img2, ann, filename
+            return img1, img2, ann, filename