diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index a648ef010af3d180c187a663ddaa85e9d8d7f94e..5c6392b6cf3fa92a538891d8cd5bd2c6e7dcb7d7 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -703,7 +703,6 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): cld_opd = get_grid_values_all(h5f, 'cld_opd_dcomp_1') ylen, xlen = cld_opd.shape - cld_opd = cld_opd[::2, ::2] cld_opd_hres = cld_opd.copy() refl_sub_lo = get_grid_values_all(h5f, 'refl_0_65um_nom_min_sub') @@ -734,19 +733,19 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): bt_us = normalize(bt_us, 'temp_11_0um_nom', mean_std_dct) print('BT done') + refl_sub_lo = refl_sub_lo[slc_y, slc_x] refl_sub_lo = np.expand_dims(refl_sub_lo, axis=0) refl_sub_lo = upsample_nearest(refl_sub_lo) - refl_sub_lo = refl_sub_lo[:, slc_y, slc_x] refl_sub_lo = normalize(refl_sub_lo, 'refl_0_65um_nom', mean_std_dct) + refl_sub_hi = refl_sub_hi[slc_y, slc_x] refl_sub_hi = np.expand_dims(refl_sub_hi, axis=0) refl_sub_hi = upsample_nearest(refl_sub_hi) - refl_sub_hi = refl_sub_hi[:, slc_y, slc_x] refl_sub_hi = normalize(refl_sub_hi, 'refl_0_65um_nom', mean_std_dct) + refl_sub_std = refl_sub_std[slc_y, slc_x] refl_sub_std = np.expand_dims(refl_sub_std, axis=0) refl_sub_std = upsample_nearest(refl_sub_std) - refl_sub_std = refl_sub_std[:, slc_y, slc_x] cld_opd = np.where(np.isnan(cld_opd), 0, cld_opd) cld_opd = cld_opd[slc_y, slc_x]