snapshot...

modules/deeplearning/cloud_opd_srcnn_viirs.py

@@ -723,6 +723,8 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     refl = refl[nn.slc_y, nn.slc_x]
     refl = np.expand_dims(refl, axis=0)
 
+    cld_opd_hres = cld_opd_hres[nn.slc_y, nn.slc_x]
+
     bt = np.where(np.isnan(bt), 0, bt)
     bt = bt[nn.slc_y_m, nn.slc_x_m]
     bt = np.expand_dims(bt, axis=0)
@@ -765,19 +767,19 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     # cld_opd_sres = descale(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)
-
-    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, :, :]
-
-    # cld_opd_sres_out = cld_opd_sres[0, :, :, 0]
-    # refl_out = refl[0, :, :]
-    # cld_opd_out = cld_opd[0, :, :]
-    # cld_opd_hres = cld_opd_hres
-    # print(cld_opd_sres_out.shape, refl_out.shape, cld_opd_out.shape, cld_opd_hres.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)
+    #
+    # 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, :, :]
+
+    cld_opd_sres_out = cld_opd_sres[0, :, :, 0]
+    refl_out = refl[0, :, :]
+    cld_opd_out = cld_opd[0, :, :]
+    cld_opd_hres = cld_opd_hres
+    print(cld_opd_sres_out.shape, refl_out.shape, cld_opd_out.shape, cld_opd_hres.shape)
 
     refl_out = denormalize(refl_out, 'refl_0_65um_nom', mean_std_dct)
     cld_opd_out = denormalize(cld_opd_out, label_param, mean_std_dct)