trw.transforms.transforms_random_deformation

Module Contents

Classes

TransformRandomDeformation

Transform an image using a random deformation field.

class trw.transforms.transforms_random_deformation.TransformRandomDeformation(control_points: Union[int, Sequence[int]] = 6, max_displacement: Optional[Union[float, Sequence[float]]] = 0.5, criteria_fn: Optional[trw.transforms.transforms.CriteriaFn] = None, interpolation: typing_extensions.Literal[linear, nearest] = 'linear', padding_mode: typing_extensions.Literal[zeros, border, reflection] = 'zeros', gaussian_filter_sigma: Optional[float] = 1.5, align_corners: bool = False)

Bases: trw.transforms.transforms.TransformBatchWithCriteria

Transform an image using a random deformation field.

Only 2D or 3D supported transformation.

The gradient can be back-propagated through this transform.

_transform(self, features_names, batch)