From 1e35d43961c8787c0cbcfd5224f7481ea467b376 Mon Sep 17 00:00:00 2001 From: jakob Date: Tue, 17 Dec 2024 12:58:25 +0100 Subject: [PATCH] add inference test - add comment --- nnunetv2/tests/integration_tests/run_nnunet_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nnunetv2/tests/integration_tests/run_nnunet_inference.py b/nnunetv2/tests/integration_tests/run_nnunet_inference.py index 116d0490d..3d72d226a 100755 --- a/nnunetv2/tests/integration_tests/run_nnunet_inference.py +++ b/nnunetv2/tests/integration_tests/run_nnunet_inference.py @@ -34,7 +34,7 @@ def run_tests_and_exit_on_failure(): img_gt = nib.load(f"nnunetv2/tests/example_data/example_ct_sm_T300_output.nii.gz").get_fdata() img_pred = nib.load(f"nnunetv2/tests/github_actions_output/example_ct_sm.nii.gz").get_fdata() dice = dice_score(img_gt, img_pred) - images_equal = dice > 0.99 + images_equal = dice > 0.99 # allow for a small difference in the segmentation, otherwise the test will fail often assert images_equal, f"The nnunet segmentation is not correct (dice: {dice:.5f})." # Clean up