From 1115f3159772fab59d69df8cd6e7c0e7494272f4 Mon Sep 17 00:00:00 2001 From: LennyN95 Date: Fri, 12 Jan 2024 11:08:08 +0100 Subject: [PATCH] fix missing default value for nnunet model configurable in ProstateRunner module --- models/nnunet_prostate_zonal_task05/utils/ProstateRunner.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/models/nnunet_prostate_zonal_task05/utils/ProstateRunner.py b/models/nnunet_prostate_zonal_task05/utils/ProstateRunner.py index e450d490..27437821 100644 --- a/models/nnunet_prostate_zonal_task05/utils/ProstateRunner.py +++ b/models/nnunet_prostate_zonal_task05/utils/ProstateRunner.py @@ -2,7 +2,7 @@ from mhubio.core import Module, Instance, InstanceData, IO @IO.Config('use_tta', bool, False, the='flag to enable test time augmentation') -@IO.Config('nnunet_model', str, None, the='nnunet model name (2d, 3d_lowres, 3d_fullres, 3d_cascade_fullres)') +@IO.Config('nnunet_model', str, '3d_fullres', the='nnunet model name (2d, 3d_lowres, 3d_fullres, 3d_cascade_fullres)') class ProstateRunner(Module): use_tta: bool @@ -45,5 +45,7 @@ def task(self, instance: Instance, T2: InstanceData, ADC: InstanceData, P: Insta if not self.use_tta: bash_command += ["--disable_tta"] + print(bash_command) + # run command self.subprocess(bash_command, text=True) \ No newline at end of file