Trait ImageTransform

pub trait ImageTransform {
    type Input;
    type Output;

    // Required method
    fn map(&self, x: &Self::Input, device: &Device) -> Result<Self::Output>;
}

A transform over an image. The input may vary but the output is always a Tensor.

Required Associated Types

type Input

type Output

Required Methods

fn map(&self, x: &Self::Input, device: &Device) -> Result<Self::Output>

Implementations on Foreign Types

impl<T: ImageTransform<Input = E, Output = E>, E: Clone> ImageTransform for Option<T>

type Input = <T as ImageTransform>::Input

type Output = <T as ImageTransform>::Output

fn map(&self, x: &T::Input, dev: &Device) -> Result<T::Output>

Implementors

impl ImageTransform for InterpolateResize

type Input = Tensor

type Output = <InterpolateResize as ImageTransform>::Input

impl ImageTransform for Normalize

type Input = Tensor

type Output = <Normalize as ImageTransform>::Input

impl ImageTransform for Rescale

type Input = Tensor

type Output = <Rescale as ImageTransform>::Input

impl ImageTransform for ToTensor

type Input = DynamicImage

type Output = Tensor

impl ImageTransform for ToTensorNoNorm

type Input = DynamicImage

type Output = Tensor