From a2c1db80e445be7c9054b3fda6ee2654c47acc79 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 7 Jul 2023 11:37:11 -0500
Subject: [PATCH] snapshot...

---
 modules/deeplearning/cloud_opd_srcnn_abi_v2.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
index b056f4ef..b304d561 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi_v2.py
@@ -740,6 +740,7 @@ 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]
@@ -766,7 +767,7 @@ class SRCNN:
         print('inference time: ', (t1 - t0))
         print(cld_opd_sres.shape)
 
-        cld_opd_sres_out = np.zeros((LEN_Y, LEN_X), dtype=np.int8)
+        cld_opd_sres_out = np.zeros((LEN_Y_in, LEN_X_in), dtype=np.int8)
         border = int((KERNEL_SIZE - 1) / 2)
         cld_opd_sres_out[border:LEN_Y - border, border:LEN_X - border] = cld_opd_sres[0, :, :, 0]
 
-- 
GitLab