From f5604c9b474d1941764d0f3098a5fac72d3c9212 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 23 Mar 2022 12:26:25 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 0e37d3c0..cb5fdaa9 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -221,7 +221,7 @@ class IcingIntensityFCN:
self.X_img = tf.keras.Input(shape=(None, None, n_chans))
self.inputs.append(self.X_img)
- self.inputs.append(tf.keras.Input(shape=(None, 5)))
+ self.inputs.append(tf.keras.Input(shape=(None, None, 5)))
self.flight_level = 0
@@ -359,6 +359,8 @@ class IcingIntensityFCN:
nda[np.logical_and(nda >= 8000, nda < 15000)] = 4
nda = tf.one_hot(nda, 5).numpy()
+ nda = np.expand_dims(nda, axis=1)
+ nda = np.expand_dims(nda, axis=1)
return nda
--
GitLab