diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index 87c63fa09d0c5811862397ed0ebe5593cda52371..96fd4fd93aeaa20083f7e9bafe1a331a7d957380 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1032,7 +1032,7 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA ckpt_dir = ckpt_dir_s_path + ckpt if not os.path.isdir(ckpt_dir): continue - nn = IcingIntensityNN(day_night=day_night, use_flight_altitude=use_flight_altitude) + nn = IcingIntensityFCN(day_night=day_night, use_flight_altitude=use_flight_altitude) nn.run_restore(filename_l1b, filename_l2, ckpt_dir) cm_s.append(tf.math.confusion_matrix(nn.test_labels.flatten(), nn.test_preds.flatten())) prob_s.append(nn.test_probs.flatten()) @@ -1068,7 +1068,7 @@ def run_evaluate_static_avg(data_dct, ll, cc, ckpt_dir_s_path, day_night='DAY', ckpt_dir = ckpt_dir_s_path + ckpt if not os.path.isdir(ckpt_dir): continue - nn = IcingIntensityNN(day_night=day_night, use_flight_altitude=use_flight_altitude) + nn = IcingIntensityFCN(day_night=day_night, use_flight_altitude=use_flight_altitude) nn.flight_level = flight_level nn.setup_eval_pipeline(data_dct, num_lines * num_elems) nn.build_model() @@ -1115,7 +1115,7 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l probs_dct = {flvl: None for flvl in flight_levels} preds_dct = {flvl: None for flvl in flight_levels} - nn = IcingIntensityNN(day_night=day_night, l1b_or_l2=l1b_or_l2, use_flight_altitude=use_flight_altitude) + nn = IcingIntensityFCN(day_night=day_night, l1b_or_l2=l1b_or_l2, use_flight_altitude=use_flight_altitude) nn.num_data_samples = num_tiles nn.build_model() nn.build_training() @@ -1143,5 +1143,5 @@ def run_evaluate_static(data_dct, num_tiles, ckpt_dir_s_path, day_night='DAY', l if __name__ == "__main__": - nn = IcingIntensityNN() + nn = IcingIntensityFCN() nn.run('matchup_filename')