From 1ad24bade0e2c3005812940c828737ccf9073cd2 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 13 Oct 2021 20:57:47 -0500
Subject: [PATCH] use meshgrid
---
modules/deeplearning/icing_cnn.py | 13 +++++--------
1 file changed, 5 insertions(+), 8 deletions(-)
diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 85f933fe..8d24ffdf 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1006,15 +1006,12 @@ def run_evaluate_static(h5f, ckpt_dir_s_path, prob_thresh=0.5, satellite='GOES16
else:
preds = np.argmax(probs, axis=1)
- ll, cc = np.meshgrid(ll, cc, indexing='ij')
- #cc = np.array(cc)
- #ll = np.array(ll)
- cc = cc.flatten()
- ll = ll.flatten()
+ ll_grd, cc_grd = np.meshgrid(ll, cc, indexing='ij')
+ cc_grd = cc_grd.flatten()
+ ll_grd = ll_grd.flatten()
ice_mask = preds == 1
- print(cc.shape, ll.shape, ice_mask.shape)
- ice_cc = cc[ice_mask]
- ice_ll = ll[ice_mask]
+ ice_cc = cc_grd[ice_mask]
+ ice_ll = ll_grd[ice_mask]
nav = get_navigation(satellite, domain)
ice_lons = []
--
GitLab