diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index b54f236625759d36fb8ec762edab9acd138e3a02..9f6a45e7ef525a3c21ff0a2cd0c36c0a662ccafe 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -327,8 +327,9 @@ class SRCNN: # ----------------------------------------------------- # ----------------------------------------------------- label = input_label[:, label_idx_i, :, :] - label = get_cldy_frac_opd(cld_prob, label) label = label[:, y_64, x_64] + cld_prob = cld_prob[:, y_64, x_64] + label = get_cldy_frac_opd(cld_prob, label) label = scale(label, label_param, mean_std_dct) label = np.where(np.isnan(label), 0, label)