Skip to content
Snippets Groups Projects
Commit 4b387cdc authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 7ea49592
Branches
No related tags found
No related merge requests found
......@@ -670,10 +670,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 = ESPCN()
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__":
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment