From c828cd65f8e73253b4258496feb7e7125bd72b58 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 18 Dec 2022 15:20:34 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/icing_fcn.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 00900f84..f8fd9b68 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/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,
--
GitLab