snapshot...

modules/deeplearning/icing_cnn.py

@@ -1011,6 +1011,8 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path):
 
 def run_evaluate_static(h5f, ckpt_dir_s_path, prob_thresh=0.5, satellite='GOES16', domain='FD'):
     data_dct, ll, cc = make_for_full_domain_predict(h5f, name_list=train_params, domain=domain)
+    num_elems = len(cc)
+    num_lines = len(ll)
 
     ckpt_dir_s = os.listdir(ckpt_dir_s_path)
 
@@ -1050,8 +1052,8 @@ def run_evaluate_static(h5f, ckpt_dir_s_path, prob_thresh=0.5, satellite='GOES16
     cc = cc.flatten()
     ll = ll.flatten()
     lon_s, lat_s = nav.lc_to_earth(cc, ll)
-    lons_2d = lon_s.reshape((len(ll), len(cc)))
-    lats_2d = lat_s.reshape((len(ll), len(cc)))
+    lons_2d = lon_s.reshape((num_lines, num_elems))
+    lats_2d = lat_s.reshape((num_lines, num_elems))
 
     ice_mask = preds == 1
     ice_cc = cc[ice_mask]