snapshot...

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))