Skip to content
Snippets Groups Projects
Commit 02fb2cf7 authored by tomrink's avatar tomrink
Browse files

add some helper methods. work on run_evaluate

parent d65e7c59
No related branches found
No related tags found
No related merge requests found
...@@ -809,6 +809,21 @@ class ESPCN: ...@@ -809,6 +809,21 @@ class ESPCN:
return self.do_evaluate(nda_lr, param, ckpt_dir) 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__": if __name__ == "__main__":
nn = ESPCN() nn = ESPCN()
nn.run('matchup_filename') nn.run('matchup_filename')
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment