From 0c5110febe82fd023bce65306864a34328044c01 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 8 Aug 2022 11:31:01 -0500
Subject: [PATCH] minor
---
modules/deeplearning/espcn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 764af4b1..6ddb3f14 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -817,7 +817,8 @@ class ESPCN:
def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'):
nda = np.load(filename)
- nda = nda[:, param_idx, :, :]
+ #nda = nda[:, param_idx, :, :]
+ nda = nda[:, param_idx, 3:67, 3:67]
nda = np.expand_dims(nda, axis=3)
# nda_lr = tf.image.resize(nda, (36, 36)).numpy()
nda_lr = tf.image.resize(nda, (32, 32)).numpy()
@@ -826,7 +827,6 @@ def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'):
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
--
GitLab