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

snapshot...

parent a599f19a
Branches
Tags
No related merge requests found
Show whitespace changes
Inline Side-by-side
modules/deeplearning/srcnn_l1b_l2.py
+ 1
− 1
View file @ dcfa56ac
...@@ -798,7 +798,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): ...@@ -798,7 +798,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
nn = SRCNN() nn = SRCNN()
out_sr = nn.run_evaluate(data, ckpt_dir) out_sr = nn.run_evaluate(data, ckpt_dir)
if out_file is not None: if out_file is not None:
np.save(out_file, [out_sr, hr_grd_c]) np.save(out_file, (out_sr[0,:,:, 0], hr_grd_c))
else: else:
return out_sr, hr_grd_c return out_sr, hr_grd_c
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment