diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 822185c43e9997f91bf4af0b0a83f7c9ae24c618..31b01ba89d4303c4a2df8844106600466991235f 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]