snapshot...

modules/deeplearning/cloud_opd_srcnn_abi.py

@@ -658,8 +658,12 @@ class SRCNN:
     def run(self, directory, ckpt_dir=None, num_data_samples=50000):
         train_data_files = glob.glob(directory+'train*mres*.npy')
         valid_data_files = glob.glob(directory+'valid*mres*.npy')
-        train_label_files = glob.glob(directory+'train*ires*.npy')
-        valid_label_files = glob.glob(directory+'valid*ires*.npy')
+
+        # 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]
         self.setup_pipeline(train_data_files, train_label_files, valid_data_files, valid_label_files, num_data_samples)
 
         self.build_model()
@@ -671,7 +675,8 @@ class SRCNN:
         self.num_data_samples = 1000
 
         valid_data_files = glob.glob(directory + 'valid*mres*.npy')
-        valid_label_files = glob.glob(directory + 'valid*ires*.npy')
+        # valid_label_files = glob.glob(directory + 'valid*ires*.npy')
+        valid_label_files = [f.replace('mres', 'ires'), f in valid_data_files]
         self.setup_test_pipeline(valid_data_files, valid_label_files)
 
         self.build_model()