diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index 24384512a3230d2b32929815b19bf99da6c8bd1d..49ef4710eafc076da70926d91ee7436292bed605 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/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()