diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index ba885c7f1517b314e83d3f23e32cd674567946ac..45d7f6bedf2aaf040b6c9a25f6ed62b44ac5785a 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, hr_grd_c]) + np.save(out_file, (out_sr[0,:,:, 0], hr_grd_c)) else: return out_sr, hr_grd_c