From 15845612f373d46fdb3dc3ca462bc05a939058d0 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 2 May 2023 14:48:11 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_fraction_fcn_viirs.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/cloud_fraction_fcn_viirs.py b/modules/deeplearning/cloud_fraction_fcn_viirs.py
index 76941cf5..b45c13d2 100644
--- a/modules/deeplearning/cloud_fraction_fcn_viirs.py
+++ b/modules/deeplearning/cloud_fraction_fcn_viirs.py
@@ -760,7 +760,9 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
labels, preds, inputs = nn.run_restore(directory, ckpt_dir)
if out_file is not None:
np.save(out_file,
- [np.squeeze(labels), preds.argmax(axis=3), inputs[:, 1:65, 1:65, 4]])
+ [np.squeeze(labels), preds.argmax(axis=3),
+ inputs[:, 1:65, 1:65, 0], inputs[:, 1:65, 1:65, 1],
+ inputs[:, 1:65, 1:65, 2], inputs[:, 1:65, 1:65, 3], inputs[:, 1:65, 1:65, 4]])
def run_evaluate_static(in_file, out_file, ckpt_dir):
--
GitLab