diff --git a/modules/icing/util.py b/modules/icing/util.py index c92fe002130ab301a65c77925a61c53926808c8c..6d0071609074eae21e765b47307d2a9d17d6dabd 100644 --- a/modules/icing/util.py +++ b/modules/icing/util.py @@ -373,7 +373,8 @@ def run_icing_predict_image(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output extent=[-105, -70, 15, 50], pirep_file='/Users/tomrink/data/pirep/pireps_202109200000_202109232359.csv', icing_dict=None, - obs_lons=None, obs_lats=None, obs_times=None, obs_alt=None, flight_level=None): + obs_lons=None, obs_lats=None, obs_times=None, obs_alt=None, obs_intensity=None, + flight_level=None): if model_type == 'CNN': model_module = icing_cnn elif model_type == 'FCN': @@ -521,6 +522,7 @@ def run_icing_predict_image(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output elif obs_times is not None: keep = np.logical_and(obs_times >= ts_0, obs_times < ts_1) keep = np.where(keep, np.logical_and(obs_alt >= alt_lo, obs_alt < alt_hi), False) + keep = np.where(keep, obs_intensity > 1, False) keep_lons = obs_lons[keep] keep_lats = obs_lats[keep] else: