diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index ea21c8ba9a56b23114bc6055401d227914f7f42d..e5db230d11bcefc69858efd35a8ce089ee67489b 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -55,7 +55,7 @@ label_params = ['refl_0_65um_nom', 'temp_11_0um_nom', 'cld_temp_acha', 'cld_pres DO_ZERO_OUT = False -data_idx, label_idx = 1, 1 +data_idx, label_idx = 4, 4 data_param = data_params[data_idx] label_param = label_params[label_idx] @@ -226,7 +226,8 @@ class SRCNN: # 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 data_param != 'cloud_fraction': + data = normalize(data, data_param, mean_std_dct, add_noise=True, noise_scale=0.005) if label_param != 'cloud_fraction': label = normalize(label, label_param, mean_std_dct)