def image_transformation_val(self):
return composite.Composite(self.dim, [
translation.InputCenterToOrigin(self.dim),
scale.Fit(self.dim, self.image_size),
scale.Fixed(self.dim, self.additional_scale),
translation.OriginToOutputCenter(self.dim, self.image_size)
])
def spatial_transformation(self):
"""
The spatial image transformation without random augmentation.
:return: The transformation.
"""
return composite.Composite(self.dim, [
translation.InputCenterToOrigin(self.dim),
scale.Fit(self.dim, self.image_size),
translation.OriginToOutputCenter(self.dim, self.image_size)
])
def spatial_transformation_volumetric(self, image):
"""
The spatial image transformation without random augmentation.
:return: The transformation.
"""
dim = 3
return composite.Composite(dim,
[translation.InputCenterToOrigin(dim),
scale.Fit(dim, self.image_size + [self.num_frames]),
translation.OriginToOutputCenter(dim, self.image_size + [self.num_frames])],
name='image_transformation',
kwparents={'image': image})
def spatial_transformation(self, image):
"""
The spatial image transformation without random augmentation.
:return: The transformation.
"""
return composite.Composite(self.dim,
[translation.InputCenterToOrigin(self.dim),
scale.Fit(self.dim, self.image_size),
scale.Fixed(self.dim, self.additional_scale),
translation.OriginToOutputCenter(self.dim, self.image_size)],
name='image',
kwparents={'image': image})
def spatial_transformation_augmented(self):
"""
The spatial image transformation with random augmentation.
:return: The transformation.
"""
return composite.Composite(self.dim, [
translation.InputCenterToOrigin(self.dim),
scale.Fit(self.dim, self.image_size, self.image_spacing),
translation.Random(self.dim, [20, 20]),
rotation.Random(self.dim, [0.35]),
scale.RandomUniform(self.dim, 0.2),
scale.Random(self.dim, [0.1, 0.1]),
translation.OriginToOutputCenter(self.dim, self.image_size,
self.image_spacing),
deformation.Output(self.dim, [8, 8], 15, self.image_size,
self.image_spacing)
])
