trw.transforms.transforms_random_deformation
¶
Module Contents¶
Classes¶
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)¶