diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index 50626c18e37f9f76c2b5c4f8ac9c5bc601b99598..97e079ab130236109722b45e16da27e084cb1087 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -225,7 +225,11 @@ class ESPCN: # if label_param != 'cloud_fraction': # label = scale(label, label_param, mean_std_dct) - data = normalize(data, data_param, mean_std_dct, add_noise=True, noise_scale=0.005) + if is_training: + data = normalize(data, data_param, mean_std_dct, add_noise=True, noise_scale=0.005) + else: + data = normalize(data, data_param, mean_std_dct) + if label_param != 'cloud_fraction': label = normalize(label, label_param, mean_std_dct)