From aebe68f24753a2073d3fa15a3a32de66ce06bcc2 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 12 Dec 2022 11:47:18 -0600
Subject: [PATCH] snapshot..
---
modules/deeplearning/srcnn_l1b_l2.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 64701b47..3fd0a7e5 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -710,9 +710,11 @@ class SRCNN:
return self.do_evaluate(data, ckpt_dir)
-def run_restore_static(directory, ckpt_dir):
+def run_restore_static(directory, ckpt_dir, out_file=None):
nn = SRCNN()
- return nn.run_restore(directory, ckpt_dir)
+ labels_denorm, preds_denorm = nn.run_restore(directory, ckpt_dir)
+ if out_file is not None:
+ np.save(out_file, [labels_denorm, preds_denorm])
def run_evaluate_static(in_file, out_file, ckpt_dir):
--
GitLab