From 5462374f64c1ed6776173556defb3c79bc8a2a95 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 13 Apr 2022 20:34:40 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/icing_fcn.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index f603c5d6..cceb93ad 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -401,6 +401,8 @@ class IcingIntensityFCN: nda = np.zeros([nd_idxs.size]) nda[:] = self.flight_level nda = tf.one_hot(nda, 5).numpy() + nda = np.expand_dims(nda, axis=0) + nda = np.expand_dims(nda, axis=0) return data, nda -- GitLab