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

snapshot..

parent 390a0b44
Branches
No related tags found
No related merge requests found
......@@ -710,9 +710,11 @@ class SRCNN:
return self.do_evaluate(data, ckpt_dir)
def run_restore_static(directory, ckpt_dir):
def run_restore_static(directory, ckpt_dir, out_file=None):
nn = SRCNN()
return nn.run_restore(directory, ckpt_dir)
labels_denorm, preds_denorm = nn.run_restore(directory, ckpt_dir)
if out_file is not None:
np.save(out_file, [labels_denorm, preds_denorm])
def run_evaluate_static(in_file, out_file, ckpt_dir):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment