From 569c3f17aac32d62065f4c8f65dc073a74bf689f Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 13 Oct 2021 16:51:16 -0500
Subject: [PATCH] use meshgrid
---
modules/deeplearning/icing_cnn.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 5c2cc19f..85f933fe 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1009,6 +1009,8 @@ def run_evaluate_static(h5f, ckpt_dir_s_path, prob_thresh=0.5, satellite='GOES16
ll, cc = np.meshgrid(ll, cc, indexing='ij')
#cc = np.array(cc)
#ll = np.array(ll)
+ cc = cc.flatten()
+ ll = ll.flatten()
ice_mask = preds == 1
print(cc.shape, ll.shape, ice_mask.shape)
ice_cc = cc[ice_mask]
--
GitLab