From 3ef82ad769c017dda8d815728528d50d2c9ec5d3 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 1 Sep 2022 10:30:23 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/srcnn.py | 14 ++++++++++----
1 file changed, 10 insertions(+), 4 deletions(-)
diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py
index a30a2c3b..9d1858d2 100644
--- a/modules/deeplearning/srcnn.py
+++ b/modules/deeplearning/srcnn.py
@@ -644,7 +644,6 @@ class SRCNN:
self.restore(ckpt_dir)
def run_evaluate(self, nda_lr, param, ckpt_dir):
- # self.setup_eval_pipeline(filename)
self.num_data_samples = 80000
self.build_model()
self.build_training()
@@ -661,10 +660,17 @@ def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'):
return nda_lr
-def run_evaluate_static(nda_lr, param='temp_11_0um_nom', ckpt_dir='/Users/tomrink/tf_model_sres/run-20220805173619/'):
+def run_evaluate_static(in_file, out_file, param='temp_11_0um_nom', ckpt_dir='/Users/tomrink/tf_model_sres/run-20220805173619/'):
+ nda = np.load(in_file)
+ nda = nda[:, data_idx, 2:133:2, 2:133:2]
+ nda = np.expand_dims(nda, axis=3)
+
nn = SRCNN()
- out_sr = nn.run_evaluate(nda_lr, param, ckpt_dir)
- return out_sr
+ out_sr = nn.run_evaluate(nda, param, ckpt_dir)
+ if out_file is not None:
+ np.save(out_file, out_sr)
+ else:
+ return out_sr
if __name__ == "__main__":
--
GitLab