From 1544cfe1f125ae5b7abcd1311f75afb2cf452fab Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 28 Apr 2023 10:28:48 -0500
Subject: [PATCH] snapshot...

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

diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py
index 9db4b165..913b195f 100644
--- a/modules/deeplearning/cloud_opd_srcnn_viirs.py
+++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py
@@ -694,6 +694,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     bt = get_grid_values_all(h5f, 'orig/temp_11_0um')
     bt = np.where(np.isnan(bt), 0, bt)
     bt = bt[nn.slc_y_m, nn.slc_x_m]
+    bt = np.expand_dims(bt, axis=0)
     bt = nn.upsample(bt)
     bt = normalize(bt, 'temp_11_0um_nom', mean_std_dct)
 
@@ -710,7 +711,8 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
 
     cld_opd = get_grid_values_all(h5f, 'orig/'+label_param)
     cld_opd = np.where(np.isnan(cld_opd), 0, cld_opd)
-    cld_opd = cld_opd[:, nn.slc_y_2, nn.slc_x_2]
+    cld_opd = cld_opd[nn.slc_y_2, nn.slc_x_2]
+    cld_opd = np.expand_dims(cld_opd, axis=0)
     cld_opd = nn.upsample(cld_opd)
     cld_opd = normalize(cld_opd, label_param, mean_std_dct)
 
-- 
GitLab