diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index 5d8887121fd7dfaaa4fad8d342bafe28138c4448..e3ca1611941c1a90ebdde2634a55f84be2fd1680 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -717,7 +717,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): refl = np.where(np.isnan(refl), 0, bt) refl = refl[slc_y, slc_x] refl = np.expand_dims(refl, axis=0) - refl = upsample_static(refl, x_2, y_2, t, s) + refl = upsample_static(refl, x_2, y_2, t, s, None, None) print(refl.shape) refl = normalize(refl, 'refl_0_65um_nom', mean_std_dct) print('REFL done') @@ -725,7 +725,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): bt = np.where(np.isnan(bt), 0, bt) bt = bt[nn.slc_y, nn.slc_x] bt = np.expand_dims(bt, axis=0) - bt = upsample_static(bt, x_2, y_2, t, s) + bt = upsample_static(bt, x_2, y_2, t, s, None, None) bt = normalize(bt, 'temp_11_0um_nom', mean_std_dct) print('BT done') @@ -743,7 +743,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): cld_opd = np.where(np.isnan(cld_opd), 0, cld_opd) cld_opd = cld_opd[nn.slc_y, nn.slc_x] cld_opd = np.expand_dims(cld_opd, axis=0) - cld_opd = upsample_static(cld_opd, x_2, y_2, t, s) + cld_opd = upsample_static(cld_opd, x_2, y_2, t, s, None, None) cld_opd = normalize(cld_opd, label_param, mean_std_dct) print('OPD done')