From bc0d3fa6c25b75728fbd9796380d9b82c5a2583f Mon Sep 17 00:00:00 2001
From: FelixSteinbauer <43598998+FelixSteinbauer@users.noreply.github.com>
Date: Thu, 24 Aug 2023 18:28:25 +0200
Subject: [PATCH] Added parameter to toggle NCC computation

Introduced a parameter (compute_ncc, defaut=True) that allows to turn off the (time-consuming) computation of the 4 NCC metrics.
---
 GANDLF/cli/generate_metrics.py | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/GANDLF/cli/generate_metrics.py b/GANDLF/cli/generate_metrics.py
index aa47e5f69..eaef3f77f 100644
--- a/GANDLF/cli/generate_metrics.py
+++ b/GANDLF/cli/generate_metrics.py
@@ -253,18 +253,20 @@ def __percentile_clip(input_tensor, reference_tensor=None, p_min=0.5, p_max=99.5
             ] = structural_similarity_index(gt_image_infill, output_infill, mask).item()
 
             # ncc metrics
-            overall_stats_dict[current_subject_id]["ncc_mean"] = ncc_mean(
-                gt_image_infill, output_infill
-            )
-            overall_stats_dict[current_subject_id]["ncc_std"] = ncc_std(
-                gt_image_infill, output_infill
-            )
-            overall_stats_dict[current_subject_id]["ncc_max"] = ncc_max(
-                gt_image_infill, output_infill
-            )
-            overall_stats_dict[current_subject_id]["ncc_min"] = ncc_min(
-                gt_image_infill, output_infill
-            )
+            compute_ncc = parameters.get("compute_ncc", True)
+            if compute_ncc:
+                overall_stats_dict[current_subject_id]["ncc_mean"] = ncc_mean(
+                    gt_image_infill, output_infill
+                )
+                overall_stats_dict[current_subject_id]["ncc_std"] = ncc_std(
+                    gt_image_infill, output_infill
+                )
+                overall_stats_dict[current_subject_id]["ncc_max"] = ncc_max(
+                    gt_image_infill, output_infill
+                )
+                overall_stats_dict[current_subject_id]["ncc_min"] = ncc_min(
+                    gt_image_infill, output_infill
+                )
 
             # only voxels that are to be inferred (-> flat array)
             # these are required for mse, psnr, etc.