Realiza clasificación de imágenes.
La API espera un modelo TFLite con metadatos del modelo TFLite opcionales, pero muy recomendados. .
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, aquí están los requisitos.
- Tensor de imagen de entrada (
kTfLiteUInt8
/kTfLiteFloat32
)- entrada de imagen de tamaño
[batch x height x width x channels]
. - No se admite la inferencia por lotes (se requiere que
batch
sea 1). - Solo se admiten entradas RGB (se requieren 3
channels
). - Si el tipo es
kTfLiteFloat32
, se deben adjuntar NormalizationOptions a los metadatos para la normalización de la entrada.
- entrada de imagen de tamaño
kTfLiteUInt8
/ kTfLiteFloat32
)- con
N
clases de 2 o 4 dimensiones, como[1 x N]
o[1 x 1 x 1 x N]
- Es necesario empaquetar el archivo de etiqueta en los metadatos. Vea el ejemplo de creación de metadatos para un clasificador de imágenes . Si no se empaquetan archivos de etiquetas, utilizará el índice como etiqueta en el resultado.
Se puede encontrar un ejemplo de dicho modelo en TensorFlow Hub. .
Clases anidadas
clase | ImageClassifier.ImageClassifierOptions | Opciones para configurar un ImageClassifier. |
Métodos públicos
Métodos heredados
Métodos públicos
Lista pública <Clasificaciones> clasificar (imagen TensorImage )
Realiza una clasificación real en la TensorImage
proporcionada.
ImageClassifier
admite los siguientes tipos de espacio de color TensorImage
:
Parámetros
imagen | un objeto UINT8 TensorImage 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 |
---|
Lista pública <Clasificaciones> clasificar (imagen TensorImage , opciones ImageProcessingOptions )
Realiza una clasificación real en el TensorImage
proporcionado con ImageProcessingOptions
.
ImageClassifier
admite las siguientes opciones:
- Región de interés (ROI) (a través de
ImageProcessingOptions.Builder.setRoi(Rect)
). El valor predeterminado es toda la imagen. - rotación de imagen (a través de
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). El valor predeterminado esImageProcessingOptions.Orientation.TOP_LEFT
.
ImageClassifier
admite los siguientes tipos de espacio de color TensorImage
:
Parámetros
imagen | un objeto UINT8 TensorImage 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 |
---|
Lista pública <Clasificaciones> clasificar (imagen MlImage )
Realiza una clasificación real en el MlImage
proporcionado.
Parámetros
imagen | un objeto MlImage que representa una imagen |
---|
Lanza
Argumento de excepción ilegal | si el tipo de almacenamiento o formato de la imagen no es compatible |
---|
Lista pública <Clasificaciones> clasificar (imagen MlImage , opciones ImageProcessingOptions )
Realiza una clasificación real en el MlImage
proporcionado con ImageProcessingOptions
.
ImageClassifier
admite las siguientes opciones:
- Región de interés (ROI) (a través de
ImageProcessingOptions.Builder.setRoi(Rect)
). El valor predeterminado es toda la imagen. - rotación de imagen (a través de
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). El valor predeterminado esImageProcessingOptions.Orientation.TOP_LEFT
.MlImage.getRotation()
no es efectivo.
Parámetros
imagen | un objeto MlImage 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 formato de la imagen no es compatible |
---|
ImageClassifier estático público createFromBuffer ( ByteBuffer modelBuffer)
Crea una instancia ImageClassifier
con un búfer de modelo y el ImageClassifier.ImageClassifierOptions
predeterminado.
Parámetros
modeloBuffer | un ByteBuffer directo o un MappedByteBuffer del modelo de clasificación |
---|
Lanza
Argumento de excepción ilegal | si el búfer del modelo no es un ByteBuffer directo o un MappedByteBuffer |
---|---|
Excepción de estado ilegal | si hay un error interno |
Excepción en tiempo de ejecución | si hay un error no especificado |
ImageClassifier estático público createFromBufferAndOptions (opciones ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions )
Crea una instancia ImageClassifier
con un búfer de modelo e ImageClassifier.ImageClassifierOptions
.
Parámetros
modeloBuffer | un ByteBuffer directo o un MappedByteBuffer del modelo de clasificación |
---|---|
opciones |
Lanza
Argumento de excepción ilegal | si el búfer del modelo no es un ByteBuffer directo o un MappedByteBuffer |
---|---|
Excepción de estado ilegal | si hay un error interno |
Excepción en tiempo de ejecución | si hay un error no especificado |
ImageClassifier estático público createFromFile (contexto de contexto, ruta de modelo de cadena )
Crea una instancia ImageClassifier
a partir del ImageClassifier.ImageClassifierOptions
predeterminado.
Parámetros
contexto | |
---|---|
modeloRuta | ruta del modelo de clasificación con metadatos en los activos |
Lanza
IOExcepción | si ocurre un error de E/S al cargar el modelo tflite |
---|---|
Argumento de excepción ilegal | si un argumento no es válido |
Excepción de estado ilegal | si hay un error interno |
Excepción en tiempo de ejecución | si hay un error no especificado |
ImageClassifier estático público createFromFile ( archivo modelFile)
Crea una instancia ImageClassifier
a partir del ImageClassifier.ImageClassifierOptions
predeterminado.
Parámetros
modeloArchivo | el modelo de clasificación Instancia File |
---|
Lanza
IOExcepción | si ocurre un error de E/S al cargar el modelo tflite |
---|---|
Argumento de excepción ilegal | si un argumento no es válido |
Excepción de estado ilegal | si hay un error interno |
Excepción en tiempo de ejecución | si hay un error no especificado |
ImageClassifier estático público createFromFileAndOptions (contexto de contexto, ruta de modelo de cadena , opciones ImageClassifier.ImageClassifierOptions )
Crea una instancia ImageClassifier
a partir de ImageClassifier.ImageClassifierOptions
.
Parámetros
contexto | |
---|---|
modeloRuta | ruta del modelo de clasificación con metadatos en los activos |
opciones |
Lanza
IOExcepción | si ocurre un error de E/S al cargar el modelo tflite |
---|---|
Argumento de excepción ilegal | si un argumento no es válido |
Excepción de estado ilegal | si hay un error interno |
Excepción en tiempo de ejecución | si hay un error no especificado |
ImageClassifier estático público createFromFileAndOptions (opciones de archivo modelFile, ImageClassifier.ImageClassifierOptions )
Crea una instancia ImageClassifier
.
Parámetros
modeloArchivo | el modelo de clasificación Instancia File |
---|---|
opciones |
Lanza
IOExcepción | si ocurre un error de E/S al cargar el modelo tflite |
---|---|
Argumento de excepción ilegal | si un argumento no es válido |
Excepción de estado ilegal | si hay un error interno |
Excepción en tiempo de ejecución | si hay un error no especificado |