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

snapshot...

parent a599f19a
No related branches found
No related tags found
No related merge requests found
Hide 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.
Finish editing this message first!
Please register or to comment