diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index 440d959f46a7aad5dddecd6827f92fc45595d4e3..247c4bc2f515c850f21f339dd8f1fe539777b61d 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -702,7 +702,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): nn = SRCNN(LEN_Y=2*LEN_Y-16, LEN_X=2*LEN_X-16) refl = np.where(np.isnan(refl), 0, bt) - #refl = refl[nn.slc_y_m, nn.slc_x_m] + refl = refl[nn.slc_y, nn.slc_x] refl = np.expand_dims(refl, axis=0) refl = nn.upsample(refl) print(refl.shape) @@ -710,7 +710,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): print('REFL done') bt = np.where(np.isnan(bt), 0, bt) - #bt = bt[nn.slc_y_m, nn.slc_x_m] + bt = bt[nn.slc_y, nn.slc_x] bt = np.expand_dims(bt, axis=0) bt = nn.upsample(bt) bt = normalize(bt, 'temp_11_0um_nom', mean_std_dct) @@ -728,7 +728,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): # refl_avg = np.squeeze(refl_avg) cld_opd = np.where(np.isnan(cld_opd), 0, cld_opd) - #cld_opd = cld_opd[nn.slc_y_m, nn.slc_x_m] + cld_opd = cld_opd[nn.slc_y, nn.slc_x] cld_opd = np.expand_dims(cld_opd, axis=0) cld_opd = nn.upsample(cld_opd) cld_opd = normalize(cld_opd, label_param, mean_std_dct)