From ae1b06f47ef6417ef67ef057fd970cc4135b7432 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 8 Feb 2023 14:25:57 -0600
Subject: [PATCH] snapshot...
---
modules/deeplearning/srcnn_cld_frac.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/srcnn_cld_frac.py b/modules/deeplearning/srcnn_cld_frac.py
index 26309e59..d6d4658d 100644
--- a/modules/deeplearning/srcnn_cld_frac.py
+++ b/modules/deeplearning/srcnn_cld_frac.py
@@ -799,7 +799,8 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
nn = SRCNN()
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.save(out_file,
+ [np.squeeze(labels), preds.argmax(axis=3), preds[:, :, :, 0], preds[:, :, :, 1], preds[:, :, :, 2]])
def run_evaluate_static(in_file, out_file, ckpt_dir):
--
GitLab