diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 64701b4752d86b402d12d4b5f46688595a40a6e5..3fd0a7e54e8154a5eda3e51f24c3a351720a2515 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -710,9 +710,11 @@ class SRCNN: return self.do_evaluate(data, ckpt_dir) -def run_restore_static(directory, ckpt_dir): +def run_restore_static(directory, ckpt_dir, out_file=None): nn = SRCNN() - return nn.run_restore(directory, ckpt_dir) + labels_denorm, preds_denorm = nn.run_restore(directory, ckpt_dir) + if out_file is not None: + np.save(out_file, [labels_denorm, preds_denorm]) def run_evaluate_static(in_file, out_file, ckpt_dir):