From cbf5a56d6551f1fa426d07dad0aeafe7a97f88d5 Mon Sep 17 00:00:00 2001 From: georgeyiasemis Date: Tue, 13 Aug 2024 10:46:57 +0200 Subject: [PATCH] Return registered image if registration --- direct/nn/vsharp/vsharp_engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/direct/nn/vsharp/vsharp_engine.py b/direct/nn/vsharp/vsharp_engine.py index 12ea3b49..a6a24fba 100644 --- a/direct/nn/vsharp/vsharp_engine.py +++ b/direct/nn/vsharp/vsharp_engine.py @@ -161,7 +161,7 @@ def _do_iteration( output_image = output_images[-1] return DoIterationOutput( - output_image=output_image, + output_image=(output_image, registered_image) if "registration_model" in self.models else output_image, sensitivity_map=data["sensitivity_map"], sampling_mask=data["sampling_mask"], data_dict={**loss_dict},