diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 94aa3b3675bc0b4ce1aac20213b7c1d0cfb60362..cd22b735cf9df5e9bd98056e52d6e2abb63a7aea 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -251,8 +251,7 @@ class SRCNN: # label = input_data[:, label_idx, 3:131:2, 3:131:2] label = input_data[:, label_idx, 3:131, 3:131] if label_param != 'cloud_fraction': - #label = normalize(label, label_param, mean_std_dct) - label = np.where(np.isnan(label), 0, label) + label = normalize(label, label_param, mean_std_dct) else: label = np.where(np.isnan(label), 0, label) label = np.expand_dims(label, axis=3)