Skip to content
Snippets Groups Projects
Commit 31994c19 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent c45e991c
Branches
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
modules/deeplearning/icing_fcn.py
+ 2
− 2
View file @ 31994c19
...@@ -1240,11 +1240,11 @@ def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_le ...@@ -1240,11 +1240,11 @@ def run_evaluate_static_2(model, data_dct, num_tiles, prob_thresh=0.5, flight_le
return preds_dct, probs_dct return preds_dct, probs_dct
def load_model(model_path, day_night='NIGHT', l1b_andor_l2='BOTH', satellite='GOES16', use_flight_altitude=False): def load_model(model_path, day_night='NIGHT', l1b_andor_l2='BOTH', use_flight_altitude=False):
ckpt_dir_s = os.listdir(model_path) ckpt_dir_s = os.listdir(model_path)
ckpt_dir = model_path + ckpt_dir_s[0] ckpt_dir = model_path + ckpt_dir_s[0]
model = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_andor_l2, satellite=satellite, use_flight_altitude=use_flight_altitude) model = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_andor_l2, use_flight_altitude=use_flight_altitude)
model.build_model() model.build_model()
model.build_training() model.build_training()
model.build_evaluation() model.build_evaluation()
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment