diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 3ce7f08aea6e30e7f6ab02d1eca35ab7658c7031..0b51fc4baeaaaecb9d8eed6076eadc3d08481cea 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1201,9 +1201,10 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l return preds_dct, probs_dct -def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_levels=[0, 1, 2, 3, 4], use_flight_altitude=False): +def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_levels=[0, 1, 2, 3, 4], + use_flight_altitude=False, use_max_cth_level=False): - if not use_flight_altitude: + if not use_flight_altitude or use_max_cth_level: flight_levels = [0] probs_dct = {flvl: None for flvl in flight_levels}