diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py index c91458838b698534baa050b838ab19ac6fb3bcd4..22c3de10d19afd7c157c242cdc1cafd2a043e7e8 100644 --- a/modules/deeplearning/unet_l1b_l2.py +++ b/modules/deeplearning/unet_l1b_l2.py @@ -244,7 +244,8 @@ class UNET: data_norm.append(tmp) data = np.stack(data_norm, axis=3) - label = scale(label, label_param, mean_std_dct) + if label_param != 'cloud_fraction': + label = scale(label, label_param, mean_std_dct) if is_training and DO_AUGMENT: data_ud = np.flip(data, axis=1)