From 02fb2cf7cbef266c7babe5a49cb77588688eff26 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 7 Aug 2022 14:04:57 -0500
Subject: [PATCH] add some helper methods. work on run_evaluate
---
modules/deeplearning/espcn.py | 15 +++++++++++++++
1 file changed, 15 insertions(+)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index b7e561c9..36e52904 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -809,6 +809,21 @@ class ESPCN:
return self.do_evaluate(nda_lr, param, ckpt_dir)
+def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'):
+ nda = np.load(filename)
+ nda = nda[:, param_idx, :, :]
+ nda = np.expand_dims(nda, axis=3)
+ nda_lr = tf.image.resize(nda, (32, 32)).numpy()
+ return nda_lr
+
+
+def run_evaluate_static(nda_lr, param='temp_11_0um_nom', ckpt_dir='/Users/tomrink/tf_model_sres/run-20220805173619/'):
+ nn = ESPCN()
+ # out_sr = nn.run_evaluate(nda_lr[2, :, :, 0], param, ckpt_dir)
+ out_sr = nn.run_evaluate(nda_lr, param, ckpt_dir)
+ return out_sr
+
+
if __name__ == "__main__":
nn = ESPCN()
nn.run('matchup_filename')
--
GitLab