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