From 4131da9d48237fa17439c5fbcdf19ae4ca477460 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Tue, 22 Mar 2022 19:20:43 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/icing_fcn.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 05e2284d..5b31a587 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -220,7 +220,8 @@ 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, None, 5))) + #self.inputs.append(tf.keras.Input(shape=(None, None, 5))) + self.inputs.append(tf.keras.Input(shape=(5))) self.flight_level = 0 @@ -358,8 +359,6 @@ 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