snapshot...

modules/deeplearning/icing_fcn.py

@@ -1108,7 +1108,7 @@ class IcingIntensityFCN:
 
 
 def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DAY', l1b_or_l2='both',
-                       satellite='GOES16', use_flight_altitude=False):
+                       satellite='GOES16', use_flight_altitude=False, out_file=None):
     ckpt_dir_s = os.listdir(ckpt_dir_s_path)
     cm_s = []
     prob_s = []
@@ -1134,7 +1134,10 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA
     cm_avg /= num
     prob_avg /= num
 
-    return labels, prob_avg, cm_avg
+    if out_file is not None:
+        np.save(out_file, [labels, prob_avg, cm_avg])
+    else:
+        return labels, prob_avg, cm_avg
 
 
 def run_evaluate_static_avg(data_dct, ll, cc, ckpt_dir_s_path, day_night='DAY', flight_level=4,