snapshot...

modules/deeplearning/cloud_opd_srcnn_abi.py

@@ -662,8 +662,8 @@ class SRCNN:
         # train_label_files = glob.glob(directory+'train*ires*.npy')
         # valid_label_files = glob.glob(directory+'valid*ires*.npy')
 
-        train_label_files = [f.replace('mres', 'ires'), f in train_data_files]
-        valid_label_files = [f.replace('mres', 'ires'), f in valid_data_files]
+        train_label_files = [f.replace('mres', 'ires') for f in train_data_files]
+        valid_label_files = [f.replace('mres', 'ires') for f in valid_data_files]
         self.setup_pipeline(train_data_files, train_label_files, valid_data_files, valid_label_files, num_data_samples)
 
         self.build_model()
@@ -676,7 +676,7 @@ class SRCNN:
 
         valid_data_files = glob.glob(directory + 'valid*mres*.npy')
         # valid_label_files = glob.glob(directory + 'valid*ires*.npy')
-        valid_label_files = [f.replace('mres', 'ires'), f in valid_data_files]
+        valid_label_files = [f.replace('mres', 'ires') for f in valid_data_files]
         self.setup_test_pipeline(valid_data_files, valid_label_files)
 
         self.build_model()