diff --git a/modules/deeplearning/icing_cnn.py b/modules/deeplearning/icing_cnn.py index 9e15c938c66c9b18a77c56c6b8f79b81bbf1a4b7..85038acf7b7e9e83eb617ce0ed74d88dfdfa49f0 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}