Fix filename without suffix

src/data/_AirChange.py

@@ -59,6 +59,9 @@ class _AirChangeDataset(CDDataset):
         label = (label / 255.0).astype(np.uint8)    # To 0,1
         return label if self.phase == 'train' else self.cropper(label)
 
+    def get_name(self, index):
+        return '{loc}-{id}-cm.bmp'.format(loc=self.LOCATION, id=index)
+
     @staticmethod
     def _bmp_loader(bmp_path_wo_ext):
         # Case insensitive .bmp loader

src/data/__init__.py

-from os.path import join, expanduser, basename, exists
+from os.path import join, expanduser, basename, exists, splitext
 
 import torch
 import torch.utils.data as data
@@ -41,7 +41,7 @@ class CDDataset(data.Dataset):
         if self.phase == 'train':
             return t1, t2, label
         else:
-            return basename(self.label_list[index]), t1, t2, label
+            return self.get_name(index), t1, t2, label
 
     def _read_file_paths(self):
         raise NotImplementedError
@@ -52,6 +52,9 @@ class CDDataset(data.Dataset):
     def fetch_image(self, image_path):
         return default_loader(image_path)
 
+    def get_name(self, index):
+        return splitext(basename(self.label_list[index]))[0]+'.bmp'
+
     def preprocess(self, t1, t2, label):
         if self.transforms[0] is not None:
             # Applied on all