snapshot...

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