diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 0a46be6c7ef34e184cd6d3619dbcf8f2b2de5999..cd7d453432fca7748726a18fe999020e69666cb0 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -47,8 +47,8 @@ f.close() mean_std_dct.update(mean_std_dct_l1b) mean_std_dct.update(mean_std_dct_l2) -params = ['temp_11_0um_nom', 'temp_12_0um_nom', 'cloud_fraction'] -data_params = ['temp_11_0um_nom', 'temp_12_0um_nom'] +params = ['temp_11_0um_nom', 'temp_12_0um_nom', 'refl_0_65um_nom', 'cloud_fraction'] +data_params = ['temp_11_0um_nom', 'temp_12_0um_nom', 'refl_0_65um_nom'] label_params = ['cloud_fraction'] @@ -203,7 +203,7 @@ class SRCNN: f = files[k] nda = np.load(f) data_s.append(nda[0:len(data_params), :, :]) - label_s.append(nda[2, :, :]) + label_s.append(nda[3, :, :]) data = np.concatenate(data_s) label = np.concatenate(label_s)