Executa classificação em imagens.
A API espera um modelo TFLite com opcional, mas altamente recomendável, TFLite modelo de metadados. .
A API oferece suporte a modelos com um tensor de entrada de imagem e um tensor de saída de classificação. Para ser mais específico, aqui estão os requisitos.
- Imagem de entrada tensor (
kTfLiteUInt8/kTfLiteFloat32)- de entrada de imagem de tamanho
[batch x height x width x channels]. - inferência lote não está suportado (
batché requerido para ser 1). - única entradas RGB são suportados (
channelsé necessário ser 3). - se o tipo é
kTfLiteFloat32, NormalizationOptions são obrigados a ser ligado ao metadados para normalização de entrada.
- de entrada de imagem de tamanho
kTfLiteUInt8 / kTfLiteFloat32 )- com
Nclasses de 2 ou 4 dimensões, tais como[1 x N]ou[1 x 1 x 1 x N] - o arquivo de etiqueta deve ser compactado nos metadados. Veja o exemplo de criação de metadados para um classificador imagem . Se nenhum arquivo de etiqueta for compactado, ele usará o índice como etiqueta no resultado.
Um exemplo de tal modelo pode ser encontrado na TensorFlow Hub. .
Classes aninhadas
| classe | ImageClassifier.ImageClassifierOptions | Opções para configurar um ImageClassifier. | |
Métodos Públicos
Métodos herdados
Métodos Públicos
pública Lista < Classificações > classificar ( TensorImage imagem)
Realiza classificação real no fornecidos TensorImage .
ImageClassifier suporta os seguintes TensorImage tipos de espaço de cores:
Parâmetros
| imagem | um UINT8 TensorImage objecto que representa uma imagem RGB ou YUV |
|---|
Lança
| Exceção de argumento ilegal | se o tipo de espaço de cor da imagem não for compatível |
|---|
pública Lista < Classificações > classificar ( TensorImage imagem, ImageProcessingOptions opções)
Realiza classificação real no fornecidos TensorImage com ImageProcessingOptions .
ImageClassifier suporta as seguintes opções:
- Região de interesse (ROI) (através
ImageProcessingOptions.Builder.setRoi(Rect)). O padrão é a imagem inteira. - rotação da imagem (através
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)). O padrão éImageProcessingOptions.Orientation.TOP_LEFT.
ImageClassifier suporta os seguintes TensorImage tipos de espaço de cores:
Parâmetros
| imagem | um UINT8 TensorImage objecto que representa uma imagem RGB ou YUV |
|---|---|
| opções |
Lança
| Exceção de argumento ilegal | se o tipo de espaço de cor da imagem não for compatível |
|---|
pública Lista < Classificações > classificar ( MlImage imagem)
Executa a classificação real no fornecidos MlImage .
Parâmetros
| imagem | um MlImage objecto que representa uma imagem |
|---|
Lança
| Exceção de argumento ilegal | se o tipo de armazenamento ou formato da imagem não for compatível |
|---|
pública Lista < Classificações > classificar ( MlImage imagem, ImageProcessingOptions opções)
Executa a classificação real no fornecidos MlImage com ImageProcessingOptions .
ImageClassifier suporta as seguintes opções:
- Região de interesse (ROI) (através
ImageProcessingOptions.Builder.setRoi(Rect)). O padrão é a imagem inteira. - rotação da imagem (através
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)). O padrão éImageProcessingOptions.Orientation.TOP_LEFT.MlImage.getRotation()não é eficaz.
Parâmetros
| imagem | um MlImage objecto que representa uma imagem |
|---|---|
| opções | configura opções incluindo ROI e rotação |
Lança
| Exceção de argumento ilegal | se o tipo de armazenamento ou formato da imagem não for compatível |
|---|
public static ImageClassifier createFromBuffer ( ByteBuffer modelBuffer)
Cria um ImageClassifier exemplo com um tampão de modelo e os predefinidos ImageClassifier.ImageClassifierOptions .
Parâmetros
| modelBuffer | uma direta ByteBuffer ou um MappedByteBuffer do modelo de classificação |
|---|
Lança
| Exceção de argumento ilegal | Se o tampão de modelo não é um directa ByteBuffer ou um MappedByteBuffer |
|---|---|
| IllegalStateException | se houver um erro interno |
| Exceção de tempo de execução | se houver um erro não especificado de outra forma |
public static ImageClassifier createFromBufferAndOptions ( ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions opções)
Cria um ImageClassifier exemplo com um tampão e modelo ImageClassifier.ImageClassifierOptions .
Parâmetros
| modelBuffer | uma direta ByteBuffer ou um MappedByteBuffer do modelo de classificação |
|---|---|
| opções |
Lança
| Exceção de argumento ilegal | Se o tampão de modelo não é um directa ByteBuffer ou um MappedByteBuffer |
|---|---|
| IllegalStateException | se houver um erro interno |
| Exceção de tempo de execução | se houver um erro não especificado de outra forma |
public static ImageClassifier CreateFromFile (contexto Contexto, Cordas modelPath)
Cria um ImageClassifier instância do padrão ImageClassifier.ImageClassifierOptions .
Parâmetros
| contexto | |
|---|---|
| modelPath | caminho do modelo de classificação com metadados nos ativos |
Lança
| IOException | se ocorrer um erro de I / O ao carregar o modelo tflite |
|---|---|
| Exceção de argumento ilegal | se um argumento é inválido |
| IllegalStateException | se houver um erro interno |
| Exceção de tempo de execução | se houver um erro não especificado de outra forma |
public static ImageClassifier CreateFromFile ( Arquivo modelFile)
Cria um ImageClassifier instância do padrão ImageClassifier.ImageClassifierOptions .
Parâmetros
| modelFile | a classificação modelo File instância |
|---|
Lança
| IOException | se ocorrer um erro de I / O ao carregar o modelo tflite |
|---|---|
| Exceção de argumento ilegal | se um argumento é inválido |
| IllegalStateException | se houver um erro interno |
| Exceção de tempo de execução | se houver um erro não especificado de outra forma |
public static ImageClassifier createFromFileAndOptions (contexto Contexto, Cordas modelPath, ImageClassifier.ImageClassifierOptions opções)
Cria um ImageClassifier exemplo de ImageClassifier.ImageClassifierOptions .
Parâmetros
| contexto | |
|---|---|
| modelPath | caminho do modelo de classificação com metadados nos ativos |
| opções |
Lança
| IOException | se ocorrer um erro de I / O ao carregar o modelo tflite |
|---|---|
| Exceção de argumento ilegal | se um argumento é inválido |
| IllegalStateException | se houver um erro interno |
| Exceção de tempo de execução | se houver um erro não especificado de outra forma |
public static ImageClassifier createFromFileAndOptions ( Arquivo modelFile, ImageClassifier.ImageClassifierOptions opções)
Cria um ImageClassifier exemplo.
Parâmetros
| modelFile | a classificação modelo File instância |
|---|---|
| opções |
Lança
| IOException | se ocorrer um erro de I / O ao carregar o modelo tflite |
|---|---|
| Exceção de argumento ilegal | se um argumento é inválido |
| IllegalStateException | se houver um erro interno |
| Exceção de tempo de execução | se houver um erro não especificado de outra forma |