diff --git a/modules/icing/util.py b/modules/icing/util.py index 6d0071609074eae21e765b47307d2a9d17d6dabd..1a26241f12831c48ce5d7722d9758d9c7f402a8c 100644 --- a/modules/icing/util.py +++ b/modules/icing/util.py @@ -12,6 +12,7 @@ from util.setup import model_path_day, model_path_night from aeolus.datasource import CLAVRx, CLAVRx_VIIRS, GOESL1B, CLAVRx_H08 import h5py import datetime +from scipy.signal import medfilt2d def get_training_parameters(day_night='DAY', l1b_andor_l2='both', use_dnb=False): @@ -535,6 +536,7 @@ def run_icing_predict_image(clvrx_dir='/Users/tomrink/data/clavrx/RadC/', output prob_s.append(probs) prob_s = np.stack(prob_s, axis=-1) max_prob = np.max(prob_s, axis=2) + max_prob = medfilt2d(max_prob) max_prob = np.where(max_prob < prob_thresh, np.nan, max_prob) make_icing_image(h5f, max_prob, None, None, clvrx_str_time, satellite, domain,