From ef144182e124e8a48c4ac0eab7b423b71709eed6 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 15 Jun 2023 11:28:55 -0500
Subject: [PATCH] snapshot...

---
 modules/deeplearning/cloud_opd_srcnn_abi.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py
index 962bc9db..c2723ff7 100644
--- a/modules/deeplearning/cloud_opd_srcnn_abi.py
+++ b/modules/deeplearning/cloud_opd_srcnn_abi.py
@@ -208,7 +208,7 @@ class SRCNN:
         self.test_label_files = None
 
         # self.n_chans = len(data_params_half) + len(data_params_full) + 1
-        self.n_chans = 3
+        self.n_chans = 6
 
         self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
 
@@ -267,6 +267,18 @@ class SRCNN:
             tmp = normalize(tmp, param, mean_std_dct)
             data_norm.append(tmp)
 
+        for param in sub_fields:
+            idx = params.index(param)
+            tmp = input_data[:, idx, :, :]
+            tmp = np.where(np.isnan(tmp), 0, tmp)
+            tmp = tmp[:, self.slc_y_m, self.slc_x_m]
+            tmp = self.upsample(tmp)
+            if param != 'refl_substddev_ch01':
+                tmp = normalize(tmp, 'refl_0_65um_nom', mean_std_dct)
+            else:
+                tmp = np.where(np.isnan(tmp), 0, tmp)
+            data_norm.append(tmp)
+
         # for param in sub_fields:
         #     idx = params.index(param)
         #     tmp = input_data[:, idx, :, :]
-- 
GitLab