Realiza clasificación de imágenes.
La API espera un modelo con TFLite opcional, pero muy recomendable, TFLite modelo de metadatos. .
La API admite modelos con un tensor de entrada de imagen y un tensor de salida de clasificación. Para ser más específicos, estos son los requisitos.
- Tensor de la imagen de entrada (
kTfLiteUInt8/kTfLiteFloat32)- de entrada de imagen de tamaño
[batch x height x width x channels]. - No se admite la inferencia por lotes (
batchse requiere que sea 1). - solamente entradas RGB son compatibles (
channelsse requiere que sea 3). - si el tipo es
kTfLiteFloat32, se requieren NormalizationOptions que se adjunta a los metadatos para la normalización de entrada.
- de entrada de imagen de tamaño
kTfLiteUInt8 / kTfLiteFloat32 )- con
Nclases de 2 o 4 dimensiones, como[1 x N]o[1 x 1 x 1 x N] - el archivo de etiqueta debe estar empaquetado en los metadatos. Ver el ejemplo de la creación de metadatos para un clasificador de la imagen . Si no hay archivos de etiquetas empaquetados, usará el índice como etiqueta en el resultado.
Un ejemplo de tal modelo se puede encontrar en TensorFlow concentradores. .
Clases anidadas
| clase | ImageClassifier.ImageClassifierOptions | Opciones para configurar un ImageClassifier. | |
Métodos públicos
Métodos heredados
Métodos públicos
pública Lista < Clasificaciones > clasificar ( TensorImage imagen)
Realiza la clasificación real en el proporcionado TensorImage .
ImageClassifier soporta los siguientes TensorImage tipos de espacio de color:
Parámetros
| imagen | un UINT8 TensorImage objeto que representa una imagen RGB o YUV |
|---|
Lanza
| Argumento de excepción ilegal | si el tipo de espacio de color de la imagen no es compatible |
|---|
pública Lista < Clasificaciones > clasificar ( TensorImage imagen, ImageProcessingOptions opciones)
Realiza la clasificación real en el proporcionado TensorImage con ImageProcessingOptions .
ImageClassifier soporta las siguientes opciones:
- Región de interés (ROI) (a través de
ImageProcessingOptions.Builder.setRoi(Rect)). De forma predeterminada, la imagen completa. - rotación de imagen (a través de
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)). Por defecto esImageProcessingOptions.Orientation.TOP_LEFT.
ImageClassifier soporta los siguientes TensorImage tipos de espacio de color:
Parámetros
| imagen | un UINT8 TensorImage objeto que representa una imagen RGB o YUV |
|---|---|
| opciones |
Lanza
| Argumento de excepción ilegal | si el tipo de espacio de color de la imagen no es compatible |
|---|
pública Lista < Clasificaciones > clasificar ( MlImage imagen)
Realiza la clasificación real en el proporcionado MlImage .
Parámetros
| imagen | un MlImage objeto que representa una imagen |
|---|
Lanza
| Argumento de excepción ilegal | si el tipo de almacenamiento o el formato de la imagen no es compatible |
|---|
pública Lista < Clasificaciones > clasificar ( MlImage imagen, ImageProcessingOptions opciones)
Realiza la clasificación real en el proporcionado MlImage con ImageProcessingOptions .
ImageClassifier soporta las siguientes opciones:
- Región de interés (ROI) (a través de
ImageProcessingOptions.Builder.setRoi(Rect)). De forma predeterminada, la imagen completa. - rotación de imagen (a través de
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)). Por defecto esImageProcessingOptions.Orientation.TOP_LEFT.MlImage.getRotation()no es eficaz.
Parámetros
| imagen | un MlImage objeto que representa una imagen |
|---|---|
| opciones | configura opciones que incluyen ROI y rotación |
Lanza
| Argumento de excepción ilegal | si el tipo de almacenamiento o el formato de la imagen no es compatible |
|---|
public static ImageClassifier createFromBuffer ( ByteBuffer modelBuffer)
Crea un ImageClassifier ejemplo con un tampón de modelo y el valor predeterminado ImageClassifier.ImageClassifierOptions .
Parámetros
| modelBuffer | una directa ByteBuffer o una MappedByteBuffer del modelo de clasificación |
|---|
Lanza
| Argumento de excepción ilegal | si el búfer no es un modelo directo ByteBuffer o una MappedByteBuffer |
|---|---|
| IllegalStateException | si hay un error interno |
| Excepción en tiempo de ejecución | si hay un error de otro modo no especificado |
public static ImageClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions opciones)
Crea un ImageClassifier ejemplo con un tampón de modelo y ImageClassifier.ImageClassifierOptions .
Parámetros
| modelBuffer | una directa ByteBuffer o una MappedByteBuffer del modelo de clasificación |
|---|---|
| opciones |
Lanza
| Argumento de excepción ilegal | si el búfer no es un modelo directo ByteBuffer o una MappedByteBuffer |
|---|---|
| IllegalStateException | si hay un error interno |
| Excepción en tiempo de ejecución | si hay un error de otro modo no especificado |
public static ImageClassifier CreateFromFile (contexto Contexto, Cadena modelPath)
Crea un ImageClassifier ejemplo, de los predeterminados ImageClassifier.ImageClassifierOptions .
Parámetros
| contexto | |
|---|---|
| modelPath | ruta del modelo de clasificación con metadatos en los activos |
Lanza
| IOException | si se produce un error de E / S al cargar el modelo tflite |
|---|---|
| Argumento de excepción ilegal | si un argumento no es válido |
| IllegalStateException | si hay un error interno |
| Excepción en tiempo de ejecución | si hay un error de otro modo no especificado |
public static ImageClassifier CreateFromFile ( Archivo MODELFILE)
Crea un ImageClassifier ejemplo, de los predeterminados ImageClassifier.ImageClassifierOptions .
Parámetros
| modelFile | el modelo de clasificación File instancia |
|---|
Lanza
| IOException | si se produce un error de E / S al cargar el modelo tflite |
|---|---|
| Argumento de excepción ilegal | si un argumento no es válido |
| IllegalStateException | si hay un error interno |
| Excepción en tiempo de ejecución | si hay un error de otro modo no especificado |
public static ImageClassifier createFromFileAndOptions (contexto Contexto, Cadena modelPath, ImageClassifier.ImageClassifierOptions opciones)
Crea un ImageClassifier ejemplo de ImageClassifier.ImageClassifierOptions .
Parámetros
| contexto | |
|---|---|
| modelPath | ruta del modelo de clasificación con metadatos en los activos |
| opciones |
Lanza
| IOException | si se produce un error de E / S al cargar el modelo tflite |
|---|---|
| Argumento de excepción ilegal | si un argumento no es válido |
| IllegalStateException | si hay un error interno |
| Excepción en tiempo de ejecución | si hay un error de otro modo no especificado |
public static ImageClassifier createFromFileAndOptions ( Archivo MODELFILE, ImageClassifier.ImageClassifierOptions opciones)
Crea un ImageClassifier ejemplo.
Parámetros
| modelFile | el modelo de clasificación File instancia |
|---|---|
| opciones |
Lanza
| IOException | si se produce un error de E / S al cargar el modelo tflite |
|---|---|
| Argumento de excepción ilegal | si un argumento no es válido |
| IllegalStateException | si hay un error interno |
| Excepción en tiempo de ejecución | si hay un error de otro modo no especificado |