MIC-DKFZ / batchgenerators

A framework for data augmentation for 2D and 3D image classification and segmentation
Apache License 2.0
1.09k stars 221 forks source link

A problem about SpatialTransform_2 #96

Closed TshellT closed 2 years ago

TshellT commented 2 years ago

I use SpatialTransform_2 to augment 3D image, but it has a error as follow: AssertionError: If you provide a list/tuple as center crop make sure it has the same dimension as your data (2d/3d) The settings of SpatialTransform_2 are as follows:

SpatialTransform_2(
            (128, 128, 128), [i // 2 for i in (128, 128, 128)],
            do_elastic_deform=True, deformation_scale=(0, 0.25),
            do_rotation=True,
            angle_x=(- 15 / 360. * 2 * np.pi, 15 / 360. * 2 * np.pi),
            angle_y=(- 15 / 360. * 2 * np.pi, 15 / 360. * 2 * np.pi),
            angle_z=(- 15 / 360. * 2 * np.pi, 15 / 360. * 2 * np.pi),
            do_scale=True, scale=(0.75, 1.25),
            border_mode_data='constant', border_cval_data=0,
            border_mode_seg='constant', border_cval_seg=0,
            order_seg=1, order_data=3,
            random_crop=True,
            p_el_per_sample=0.1, p_rot_per_sample=0.1, p_scale_per_sample=0.1
        )

How can I go about solving this problem, can you give me some advice? Thanks for your reply!