diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py index feb90a73c15e5b9257f45fe0a2d6b1babd9147aa..a14019fb9e0ef27bf11d6bcf222e8883222a85f3 100644 --- a/modules/deeplearning/icing_fcn.py +++ b/modules/deeplearning/icing_fcn.py @@ -1116,8 +1116,10 @@ def run_restore_static(filename_l1b, filename_l2, ckpt_dir_s_path, day_night='DA cm_avg /= num prob_avg /= num + print(cm_avg) + if out_file is not None: - np.save(out_file, [labels, prob_avg, cm_avg]) + np.save(out_file, [labels, prob_avg]) else: return labels, prob_avg, cm_avg