From dcfa56acc0a2d73d5c4ce64a181dd75cc551ed1a Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 28 Dec 2022 11:45:54 -0600 Subject: [PATCH] snapshot... --- modules/deeplearning/srcnn_l1b_l2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index ba885c7f..45d7f6be 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 -- GitLab