From 31a17806af2973b8ee4373fa52439a528710ab3f Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 15 Dec 2021 13:11:52 -0600
Subject: [PATCH] snapshot...

---
 modules/deeplearning/icing_cnn.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py
index 9e15c938..85038acf 100644
--- a/modules/deeplearning/icing_cnn.py
+++ b/modules/deeplearning/icing_cnn.py
@@ -1051,7 +1051,7 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA
     return labels, prob_avg, cm_avg
 
 
-def run_evaluate_static(data_dct, ll, cc, ckpt_dir_s_path, day_night='DAY', flight_level=4,
+def run_evaluate_static_avg(data_dct, ll, cc, ckpt_dir_s_path, day_night='DAY', flight_level=4,
                         use_flight_altitude=False, prob_thresh=0.5,
                         satellite='GOES16', domain='FD'):
     num_elems = len(cc)
@@ -1109,6 +1109,9 @@ def run_evaluate_static_new(data_dct, num_lines, num_elems, ckpt_dir_s_path, day
     ckpt_dir_s = os.listdir(ckpt_dir_s_path)
     ckpt_dir = ckpt_dir_s_path + ckpt_dir_s[0]
 
+    if not use_flight_altitude:
+        flight_levels = [0]
+
     probs_2d_dct = {flvl: None for flvl in flight_levels}
     preds_2d_dct = {flvl: None for flvl in flight_levels}
 
-- 
GitLab