diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index 9f6a45e7ef525a3c21ff0a2cd0c36c0a662ccafe..df2f53f358d08ba8e701e5f5175c59fda6a52d0a 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -715,7 +715,7 @@ 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 = [f.replace('mres', 'ires') for f in valid_data_files] self.setup_test_pipeline(valid_data_files, valid_label_files) self.build_model()