From 52f9268819b5cca363521fb45e5b912e0dd246b0 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 6 Dec 2021 14:22:14 -0600 Subject: [PATCH] minor --- modules/deeplearning/icing_cnn.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 822185c4..31b01ba8 100644 --- a/modules/deeplearning/icing_cnn.py +++ b/modules/deeplearning/icing_cnn.py @@ -1077,6 +1077,10 @@ def run_evaluate_static(data_dct, ll, cc, ckpt_dir_s_path, flight_level=4, prob_ preds = np.argmax(probs, axis=1) preds_2d = preds.reshape((num_lines, num_elems)) + ll, cc = np.meshgrid(ll, cc, indexing='ij') + cc = cc.flatten() + ll = ll.flatten() + ice_mask = preds == 1 ice_cc = cc[ice_mask] ice_ll = ll[ice_mask] -- GitLab