From af0342e1a076d26e579d1326447ca31cdeb9b5ba Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 25 Jan 2023 11:58:53 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/srcnn_cld_frac.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/srcnn_cld_frac.py b/modules/deeplearning/srcnn_cld_frac.py
index 83fea149..fd24a93e 100644
--- a/modules/deeplearning/srcnn_cld_frac.py
+++ b/modules/deeplearning/srcnn_cld_frac.py
@@ -762,8 +762,9 @@ class SRCNN:
preds = np.concatenate(self.test_preds)
print(labels.shape, preds.shape)
- labels_denorm = denormalize(labels, label_param, mean_std_dct)
- preds_denorm = denormalize(preds, label_param, mean_std_dct)
+ if label_param != 'cloud_probability':
+ labels_denorm = denormalize(labels, label_param, mean_std_dct)
+ preds_denorm = denormalize(preds, label_param, mean_std_dct)
return labels_denorm, preds_denorm
@@ -815,7 +816,7 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
nn = SRCNN()
labels_denorm, preds_denorm = nn.run_restore(directory, ckpt_dir)
if out_file is not None:
- np.save(out_file, [labels_denorm, preds_denorm])
+ np.save(out_file, [np.squeeze(labels_denorm), preds_denorm.argmax(axis=3)])
def run_evaluate_static(in_file, out_file, ckpt_dir):
--
GitLab