From 8895a0a0702fced2d8a8bf3d3810664dbc0d1d87 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Fri, 7 Jul 2023 11:42:01 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/cloud_opd_srcnn_abi_v2.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py index b304d561..68f3342a 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py @@ -740,7 +740,6 @@ class SRCNN: print(refl.shape) refl = refl[1912:3512, 1912:3512] LEN_Y, LEN_X = refl.shape - LEN_Y_in, LEN_X_in = LEN_Y, LEN_X print(refl.shape) bt = get_grid_values_all(h5f, 'temp_11_0um_nom') bt = bt[1912:3512, 1912:3512] @@ -762,7 +761,7 @@ class SRCNN: LEN_X = 2 * (LEN_X - 8) t0 = time.time() - cld_opd_sres = self.run_inference_(bt, refl, refl_sub_lo, refl_sub_hi, refl_sub_std, cld_opd, LEN_Y, LEN_X) + cld_opd_sres, LEN_Y_in, LEN_X_in = self.run_inference_(bt, refl, refl_sub_lo, refl_sub_hi, refl_sub_std, cld_opd, LEN_Y, LEN_X) t1 = time.time() print('inference time: ', (t1 - t0)) print(cld_opd_sres.shape) @@ -841,7 +840,7 @@ class SRCNN: cld_opd_sres = self.do_inference(data) cld_opd_sres = denormalize(cld_opd_sres, label_param, mean_std_dct) - return cld_opd_sres + return cld_opd_sres, bt_us.shape[0,] def run_restore_static(directory, ckpt_dir, out_file=None): -- GitLab