diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 45d7f6bedf2aaf040b6c9a25f6ed62b44ac5785a..00b5f3ff1b31c0cc0dfc9c96efd5f0dc474db115 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -798,7 +798,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir): nn = SRCNN() out_sr = nn.run_evaluate(data, ckpt_dir) if out_file is not None: - np.save(out_file, (out_sr[0,:,:, 0], hr_grd_c)) + np.save(out_file, (out_sr[0, :, :, 0], hr_grd_c)) else: return out_sr, hr_grd_c