Struct Transform2D 

pub struct Transform2D { /* private fields */ }

2D point transformation

Useful for resampling: rotating, scaling, translating, etc image tensors

Implementations

impl Transform2D

pub fn transform(self, img: Tensor<4>) -> Tensor<4>

Transforms an image

• img - Images tensor with shape (batch_size, channels, height, width)

Returns

A tensor with the same as the input

pub fn with_padding_mode( self, padding_mode: GridSamplePaddingMode, ) -> Transform2D

Set the padding mode for the transformed images.

Default

GridSamplePaddingMode::Border

pub fn composed<I>(transforms: I) -> Transform2D

where I: IntoIterator<Item = Transform2D>,

Makes a 2d transformation composed of other transformations

pub fn identity() -> Transform2D

Makes an identity transform (x = Ax)

pub fn rotation(theta: f32, cx: f32, cy: f32) -> Transform2D

Makes a Transform2D for rotating a tensor

• theta - In radians, the rotation
• cx - Center of rotation, x
• cy - Center of rotation, y

pub fn scale(sx: f32, sy: f32, cx: f32, cy: f32) -> Transform2D

Makes a Transform2D for scaling an image tensor

• sx - Scale factor in the x direction
• sy - Scale factor in the y direction
• cx - Center of scaling, x
• cy - Center of scaling, y

pub fn translation(tx: f32, ty: f32) -> Transform2D

Makes a Transform2D for translating an image tensor

• tx - Translation in the x direction
• ty - Translation in the y direction

pub fn shear(shx: f32, shy: f32, cx: f32, cy: f32) -> Transform2D

Applies a general shear transformation around the image center, combining both X and Y shear.

Arguments

• shx - Shear factor along the X-axis.
• shy - Shear factor along the Y-axis.
• cx, cy - Coordinates of the image center.

Returns

• Self with a combined shear transform matrix.