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<B: Backend>(self, img: Tensor<B, 4>) -> Tensor<B, 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 composed<I: IntoIterator<Item = Self>>(transforms: I) -> Self

Makes a 2d transformation composed of other transformations

pub fn identity() -> Self

Makes an identity transform (x = Ax)

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

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) -> Self

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) -> Self

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) -> Self

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.