View Source TFLiteElixir.ImageClassification (tflite_elixir v0.3.7)

Experimental image classification module.

Summary

Functions

child_spec(init_arg)

Returns a specification to start this module under a supervisor.

predict(pid, input_path, opts \\ [])

set_label(pid, label_file)

set_label_from_associated_file(pid, associated_filename)

start(model, opts \\ [])

Functions

child_spec(init_arg)

Returns a specification to start this module under a supervisor.

See Supervisor.

predict(pid, input_path, opts \\ [])

@spec predict(
  pid(),
  binary()
  | %StbImage{data: term(), shape: term(), type: term()}
  | %Nx.Tensor{
      data: term(),
      names: term(),
      shape: term(),
      type: term(),
      vectorized_axes: term()
    },
  Keyword.t()
) :: term()

set_label(pid, label_file)

@spec set_label(pid(), String.t() | [String.t()]) :: :ok

set_label_from_associated_file(pid, associated_filename)

@spec set_label_from_associated_file(pid(), String.t()) :: :ok | {:error, String.t()}

start(model, opts \\ [])

@spec start(any(), any()) :: :ignore | {:error, any()} | {:ok, pid()}