self.bspline_transformation = None
def randomise(self, image_dict, spatial_rank=3):
images = image_dict.values()
equal_shapes = np.all([images[0].shape == image.shape for image in images])
if spatial_rank == 3 and equal_shapes:
self._randomise_bspline_transformation_3d(images[0].shape)
else:
After Change
if not self.do_augmentation:
pass
images = list(image_dict.values())
equal_shapes = np.all([images[0].shape == image.shape for image in images])
if spatial_rank == 3 and equal_shapes:
self._randomise_bspline_transformation_3d(images[0].shape)
else: