diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py index 2e7705da32ec7ecb0b5f0ddfdb915ccbb0ff17b2..f04f8116a5ee5c6ff8733024c38036757b987315 100644 --- a/modules/deeplearning/cloud_opd_srcnn_viirs.py +++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py @@ -735,8 +735,12 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): cld_opd_sres = denormalize(cld_opd_sres, label_param, mean_std_dct) _, ylen, xlen, _ = cld_opd_sres.shape cld_opd_sres_out = np.zeros((LEN_Y, LEN_X), dtype=np.float32) + refl_out = np.zeros((LEN_Y, LEN_X), dtype=np.float32) + cld_opd_out = np.zeros((LEN_Y, LEN_X), dtype=np.float32) border = int((KERNEL_SIZE - 1) / 2) cld_opd_sres_out[border:(border+ylen), border:(border+xlen)] = cld_opd_sres[0, :, :, 0] + refl_out[0:(ylen+2*border), 0:(xlen+2*border)] = refl[0, :, :] + cld_opd_out[0:(ylen+2*border), 0:(xlen+2*border)] = cld_opd[0, :, :] if out_file is not None: np.save(out_file, (cld_opd_sres_out, refl, cld_opd))