Default values update

MIC-DKFZ · Dec 13, 2023 · b1c2ea6 · b1c2ea6
1 parent 78c74f8
commit b1c2ea6
Showing 1 changed file with 8 additions and 8 deletions.
diff --git 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))