From 86083ac4b10baf19b76f6ce14ce1c4f76e701cd4 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 27 Apr 2021 10:03:39 -0500
Subject: [PATCH] initial commit
---
modules/deeplearning/icing_cnn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 9bef8299..f9ef3d6a 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -199,7 +199,7 @@ class IcingIntensityNN:
data.append(nda)
data = np.stack(data)
data = data.astype(np.float32)
- data = np.transpose(data, axes=(1, 0))
+ data = np.transpose(data, axes=(1, 2, 3, 0))
label = self.h5f['icing_intensity'][nd_idxs]
label = label.astype(np.int32)
--
GitLab