diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index f0df6282ad9f2a5e2ee2194e8f5286d8fe8dd39b..b963d5f240b4c8df274e9073523f39fc85222655 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -698,6 +698,10 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): grd_b = normalize(grd_b, 'refl_0_65um_nom', mean_std_dct) grd_c = resample_2d_linear_one(x, y, grd_c, x_up, y_up) + if label_param == 'cloud_fraction': + grd_c = np.where(np.isnan(grd_c), 0, grd_c) + else: + grd_c = normalize(grd_c, label_param, mean_std_dct) data = np.stack([grd_a, grd_b, grd_c], axis=2) data = np.expand_dims(data, axis=0)