From b1c2ea6c8741493ed7d18c857e8750355f680976 Mon Sep 17 00:00:00 2001 From: "Kovacs, Balint" Date: Wed, 13 Dec 2023 16:26:46 +0100 Subject: [PATCH] Default values update --- .../nnUNetTrainer/nnUNetTrainer_Misalign.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer_Misalign.py b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer_Misalign.py index daf06b3..7e14310 100644 --- a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer_Misalign.py +++ b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer_Misalign.py @@ -98,16 +98,16 @@ def get_training_transforms(patch_size: Union[np.ndarray, Tuple[int]], tr_transforms.append(MisalignTransform(im_channels_2_misalign=[0, ], label_channels_2_misalign=[0, ], do_squeeze=True, - sq_x=[1.0, 1.0], sq_y=[1.5, 1.5], sq_z=[1.0, 1.0], - p_sq_per_sample=1.0, + sq_x=[1.0, 1.0], sq_y=[0.9, 1.1], sq_z=[1.0, 1.0], + p_sq_per_sample=0.1, do_rotation=False, - angle_x=(- 0 / 360. * 2 * np.pi, 0 / 360. * 2 * np.pi), - angle_y=(- 0 / 360. * 2 * np.pi, 0 / 360. * 2 * np.pi), - angle_z=(45 / 360. * 2 * np.pi, 45 / 360. * 2 * np.pi), - p_rot_per_sample=1.0, + angle_x=(- 0 / 360. * 2 * np.pi, 0 / 360. * 2 * np.pi), + angle_y=(- 0 / 360. * 2 * np.pi, 0 / 360. * 2 * np.pi), + angle_z=(-15 / 360. * 2 * np.pi, 15 / 360. * 2 * np.pi), + p_rot_per_sample=0.1, do_transl=True, - tr_x=[100, 100], tr_y=[0, 0], tr_z=[0, 0], - p_transl_per_sample=1.0, + tr_x=[-32, 32], tr_y=[-32, 32], tr_z=[-2, 2], + p_transl_per_sample=0.1, border_mode_data='constant', border_cval_data=0, border_mode_seg='constant', border_cval_seg=0, order_data=3, order_seg=0))