Model

clase pública Model

La clase wrapper para un modelo de TFLite y un intérprete de TFLite.

Nota: Un Model solo puede contener 1 modelo de TFLite a la vez y siempre contiene una instancia de intérprete de TFLite para ejecutarlo.

Categorías anidadas

clase Model.Builder Esta clase dejó de estar disponible. Usa Model.createModel(Context, String, Options).  
enum Model.Device El tipo de dispositivo de entorno de ejecución que se usa para ejecutar la clasificación. 
clase Model.Options Opciones para ejecutar el modelo. 

Métodos públicos

void
close()
Model estático
createModel(contexto de contexto, String modelPath, opciones de Model.Options)
Carga un modelo a partir de elementos y, luego, inicializa el intérprete de TFLite con las opciones determinadas.
Model estático
createModel(contexto de contexto, String modelPath)
Carga un modelo a partir de elementos y, luego, inicializa el intérprete de TFLite.
MappedByteBuffer
getData()
Muestra los datos del modelo asignados a la memoria.
Tensor
getInputTensor(int inputIndex)
Obtiene el tensor asociado con el índice de entrada proporcionado.
Tensor
getOutputTensor(int outputIndex)
Obtiene el tensor asociado con el índice de salida proporcionado.
int[]
getOutputTensorShape(int outputIndex)
Muestra la forma de salida.
String
getPath()
Muestra la ruta de acceso del archivo de modelo almacenado en Recursos.
void
run(Object[] entradas, Map<Integer, Object> salidas)
Ejecuta la inferencia de modelo en varias entradas y muestra varias salidas.

Métodos heredados

Métodos públicos

public void close ()

estática pública Model createModel (contexto de contexto, String modelPath, opciones de Model.Options)

Carga un modelo a partir de elementos y, luego, inicializa el intérprete de TFLite con las opciones determinadas.

Parámetros
Contexto Contexto de la app
modelPath La ruta de acceso del archivo de modelo.
Opciones Las opciones para ejecutar el modelo.
Arroja
IOException si se produce alguna excepción cuando se abra el archivo del modelo.
Consulta también

Público estático Model createModel (contexto de contexto, String modelPath)

Carga un modelo a partir de elementos y, luego, inicializa el intérprete de TFLite.

Las opciones predeterminadas son: (1) dispositivo de CPU; (2) subproceso.

Parámetros
Contexto Contexto de la app
modelPath La ruta de acceso del archivo de modelo.
Arroja
IOException si se produce alguna excepción cuando se abra el archivo del modelo.

public MappedByteBuffer getData ()

Muestra los datos del modelo asignados a la memoria.

public Tensor getInputTensor (int inputIndex)

Obtiene el tensor asociado con el índice de entrada proporcionado.

Parámetros
inputIndex
Arroja
IllegalStateException si el intérprete está cerrado.

public Tensor getOutputTensor (int outputIndex)

Obtiene el tensor asociado con el índice de salida proporcionado.

Parámetros
outputIndex
Arroja
IllegalStateException si el intérprete está cerrado.

public int[] getOutputTensorShape (int outputIndex)

Muestra la forma de salida. Es útil si la forma de salida solo se determina cuando se crea el gráfico.

Parámetros
outputIndex
Arroja
IllegalStateException si el intérprete está cerrado.

public String getPath ()

Muestra la ruta de acceso del archivo de modelo almacenado en Recursos.

public void run (Object[] entradas, Map<Integer, Object> output)

Ejecuta la inferencia de modelo en varias entradas y muestra varias salidas.

Parámetros
ocultas un array de datos de entrada. Las entradas deben estar en el mismo orden que las entradas del modelo. Cada entrada puede ser un array o un array multidimensional, o bien una ByteBuffer de tipos primitivos, como int, float, long y byte. ByteBuffer es la forma preferida de pasar datos de entrada grandes, mientras que los tipos de cadenas requieren el uso de la ruta de entrada de array (multidimensional). Cuando se usa ByteBuffer, su contenido no debe cambiarse hasta que se complete la inferencia de modelo.
salidas una asignación de índices de salida a arrays multidimensionales de datos de salida o ByteBuffer de tipos primitivos, como int, float, long y byte. Solo necesita conservar entradas para que las salidas se usen.