diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py index 8c662ff33cc37b029daf032d2544c5bc8f915ff7..f590c83882a2b12a305c715fa2cf9d29e9f0698b 100644 --- a/modules/deeplearning/cloud_opd_srcnn_viirs.py +++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py @@ -732,9 +732,10 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): cld_opd_sres = nn.run_evaluate(data, 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) border = int((KERNEL_SIZE - 1) / 2) - cld_opd_sres_out[border:LEN_Y - border, border:LEN_X - border] = cld_opd_sres[0, :, :, 0] + cld_opd_sres_out[border:(border+ylen), border:(border+xlen)] = cld_opd_sres[0, :, :, 0] if out_file is not None: np.save(out_file, (cld_opd_sres_out, bt, refl))