From 8de033d05afe023707bec3ce8ff4c79accc03c0e Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 21 Apr 2023 11:22:12 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_fraction_fcn_viirs.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/cloud_fraction_fcn_viirs.py b/modules/deeplearning/cloud_fraction_fcn_viirs.py
index cd5bf4b1..29722007 100644
--- a/modules/deeplearning/cloud_fraction_fcn_viirs.py
+++ b/modules/deeplearning/cloud_fraction_fcn_viirs.py
@@ -572,7 +572,7 @@ class SRCNN:
self.test_labels.append(labels)
self.test_preds.append(pred.numpy())
- self.test_input.append(inputs[:, :, :, 4])
+ self.test_input.append(inputs)
self.test_loss(t_loss)
self.test_accuracy(labels, pred)
@@ -772,7 +772,7 @@ 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])
+ [np.squeeze(labels), preds.argmax(axis=3), inputs[:, 1:65, 1:65, 4]])
def run_evaluate_static(in_file, out_file, ckpt_dir):
--
GitLab