From 503e2726c8cc26659fc0ce02d4c250102aaba826 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 6 Feb 2023 11:34:58 -0600 Subject: [PATCH] snapshot... --- modules/deeplearning/srcnn_cld_frac.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/deeplearning/srcnn_cld_frac.py b/modules/deeplearning/srcnn_cld_frac.py index a62455e3..57b9f948 100644 --- a/modules/deeplearning/srcnn_cld_frac.py +++ b/modules/deeplearning/srcnn_cld_frac.py @@ -790,9 +790,9 @@ class SRCNN: def run_restore_static(directory, ckpt_dir, out_file=None): nn = SRCNN() - labels_denorm, preds_denorm = nn.run_restore(directory, ckpt_dir) + labels, preds = nn.run_restore(directory, ckpt_dir) if out_file is not None: - np.save(out_file, [np.squeeze(labels_denorm), preds_denorm.argmax(axis=3)]) + np.save(out_file, [np.squeeze(labels), preds.argmax(axis=3)]) def run_evaluate_static(in_file, out_file, ckpt_dir): -- GitLab