View Source ExVision.StyleTransfer.UdnieFast (Ex Vision v0.4.0)

Elixir.ExVision.StyleTransfer.UdnieFast is a custom style transfer model optimised for devices with low computational capabilities and CPU inference.

Summary

Types

output_t()

A type consisting of output tesnor (stylized image tensor) from style transfer models of shape {400, 300, 3}.

t()

An instance of the Elixir.ExVision.StyleTransfer.UdnieFast

Functions

batched_run(name \\ __MODULE__, input)

Submits the input for inference to the process running the Nx.Serving for this model.

load(options \\ [])

Creates the model instance

load!(opts \\ [])

Same as load/1, but raises and error on failure.

run(model, input)

Immediatelly applies the model to the given input, in the scope of the current process.

Types

output_t()

@type output_t() :: Nx.Tensor.t()

A type consisting of output tesnor (stylized image tensor) from style transfer models of shape {400, 300, 3}.

t()

@opaque t()

An instance of the Elixir.ExVision.StyleTransfer.UdnieFast

Functions

batched_run(name \\ MODULE, input)

@spec batched_run(atom(), ExVision.Model.input_t()) :: output_t() | [output_t()]

Submits the input for inference to the process running the Nx.Serving for this model.

load(options \\ [])

@spec load([ExVision.Model.Definition.Ortex.load_option_t()]) ::
  {:ok, t()} | {:error, reason :: atom()}

Creates the model instance

load!(opts \\ [])

@spec load!(keyword()) :: t()

Same as load/1, but raises and error on failure.

run(model, input)

@spec run(t(), ExVision.Model.input_t()) :: output_t() | [output_t()]

Immediatelly applies the model to the given input, in the scope of the current process.