Skip to content
Snippets Groups Projects
Commit 8895a0a0 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent a2c1db80
Branches
No related tags found
No related merge requests found
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment