Skip to content
Snippets Groups Projects
Commit 1ad24bad authored by tomrink's avatar tomrink
Browse files

use meshgrid

parent 569c3f17
Branches
No related tags found
No related merge requests found
...@@ -1006,15 +1006,12 @@ def run_evaluate_static(h5f, ckpt_dir_s_path, prob_thresh=0.5, satellite='GOES16 ...@@ -1006,15 +1006,12 @@ def run_evaluate_static(h5f, ckpt_dir_s_path, prob_thresh=0.5, satellite='GOES16
else: else:
preds = np.argmax(probs, axis=1) preds = np.argmax(probs, axis=1)
ll, cc = np.meshgrid(ll, cc, indexing='ij') ll_grd, cc_grd = np.meshgrid(ll, cc, indexing='ij')
#cc = np.array(cc) cc_grd = cc_grd.flatten()
#ll = np.array(ll) ll_grd = ll_grd.flatten()
cc = cc.flatten()
ll = ll.flatten()
ice_mask = preds == 1 ice_mask = preds == 1
print(cc.shape, ll.shape, ice_mask.shape) ice_cc = cc_grd[ice_mask]
ice_cc = cc[ice_mask] ice_ll = ll_grd[ice_mask]
ice_ll = ll[ice_mask]
nav = get_navigation(satellite, domain) nav = get_navigation(satellite, domain)
ice_lons = [] ice_lons = []
... ...
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment