From af279cd93a5ad1e84cde6de12487043c0a00fdfe Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 13 Oct 2021 16:26:06 -0500
Subject: [PATCH] use meshgrid
---
modules/deeplearning/icing_cnn.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index f3431d1d..5c2cc19f 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1006,8 +1006,9 @@ def run_evaluate_static(h5f, ckpt_dir_s_path, prob_thresh=0.5, satellite='GOES16
else:
preds = np.argmax(probs, axis=1)
- cc = np.array(cc)
- ll = np.array(ll)
+ ll, cc = np.meshgrid(ll, cc, indexing='ij')
+ #cc = np.array(cc)
+ #ll = np.array(ll)
ice_mask = preds == 1
print(cc.shape, ll.shape, ice_mask.shape)
ice_cc = cc[ice_mask]
--
GitLab