From e19a6cc558285736b388bb30eddbb6bb644c49e8 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Tue, 28 Mar 2023 12:02:14 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/cloud_fraction_fcn.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/modules/deeplearning/cloud_fraction_fcn.py b/modules/deeplearning/cloud_fraction_fcn.py index bcd94297..e5433caf 100644 --- a/modules/deeplearning/cloud_fraction_fcn.py +++ b/modules/deeplearning/cloud_fraction_fcn.py @@ -761,7 +761,7 @@ def run_restore_static(directory, ckpt_dir, out_file=None): labels, preds = nn.run_restore(directory, ckpt_dir) if out_file is not None: np.save(out_file, - [np.squeeze(labels), preds.argmax(axis=3), preds[:, :, :, 0], preds[:, :, :, 1], preds[:, :, :, 2]]) + [np.squeeze(labels), preds.argmax(axis=3)]) def run_evaluate_static(in_file, out_file, ckpt_dir): @@ -816,9 +816,6 @@ def analyze_3cat(file): tup = np.load(file, allow_pickle=True) lbls = tup[0] pred = tup[1] - prob_0 = tup[2] - prob_1 = tup[3] - prob_2 = tup[4] lbls = lbls.flatten() pred = pred.flatten() @@ -918,9 +915,6 @@ def analyze_5cat(file='/Users/tomrink/cld_opd_frac.npy'): tup = np.load(file, allow_pickle=True) lbls = tup[0] pred = tup[1] - # prob_0 = tup[2] - # prob_1 = tup[3] - # prob_2 = tup[4] lbls = lbls.flatten() pred = pred.flatten() -- GitLab