Executa pesquisa de similaridade em imagens.
A API espera um modelo TFLite com metadados de modelo TFLite opcionais, mas altamente recomendados . .
- Tensor de imagem de entrada (
kTfLiteUInt8
/kTfLiteFloat32
)- entrada de imagem de tamanho
[batch x height x width x channels]
. - a inferência de lote não é suportada (o
batch
deve ser 1). - apenas entradas RGB são suportadas (os
channels
devem ser 3). - se type for
kTfLiteFloat32
, NormalizationOptions deverá ser anexado aos metadados para normalização de entrada.
- entrada de imagem de tamanho
kTfLiteUInt8
/ kTfLiteFloat32
)-
N
componentes correspondentes àsN
dimensões do vetor de feição retornado para esta camada de saída. - 2 ou 4 dimensões, ou seja,
[1 x N]
ou[1 x 1 x 1 x N]
.
TODO(b/180502532): adiciona ponteiro ao modelo de exemplo.
TODO(b/222671076): adicionar métodos de criação de fábrica sem opções, como `createFromFile`, uma vez que o formato de arquivo único (arquivo de índice compactado no modelo) é suportado.
Classes aninhadas
aula | ImageSearcher.ImageSearcherOptions | Opções para configurar um ImageSearcher. |
Métodos públicos
Pesquisador de imagens estático | createFromBufferAndOptions ( ByteBuffer modelBuffer, opções ImageSearcher.ImageSearcherOptions ) Cria uma instância de ImageSearcher com um buffer de modelo e ImageSearcher.ImageSearcherOptions . |
Pesquisador de imagens estático | createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opções ImageSearcher.ImageSearcherOptions , int indexFd) |
Pesquisador de imagens estático | createFromFileAndOptions (contexto de contexto, String modelPath, opções ImageSearcher.ImageSearcherOptions ) Cria uma instância de ImageSearcher de ImageSearcher.ImageSearcherOptions . |
Pesquisador de imagens estático | createFromFileAndOptions ( Opções de arquivo modelFile, ImageSearcher.ImageSearcherOptions ) Cria uma instância de ImageSearcher . |
Lista < Vizinho Mais Próximo > | search ( imagem TensorImage , opções ImageProcessingOptions ) Executa a extração de incorporação no TensorImage fornecido com ImageProcessingOptions , seguido pela pesquisa do vizinho mais próximo no índice. |
Lista < Vizinho Mais Próximo > | |
Lista < Vizinho Mais Próximo > | search (imagem MlImage , opções ImageProcessingOptions ) Executa a extração de incorporação no MlImage fornecido com ImageProcessingOptions , seguido pela pesquisa do vizinho mais próximo no índice. |
Lista < Vizinho Mais Próximo > | pesquisa (imagem TensorImage ) Executa a extração de incorporação no TensorImage fornecido, seguido pela pesquisa do vizinho mais próximo no índice. |
Métodos Herdados
Métodos públicos
public static ImageSearcher createFromBufferAndOptions (opções ByteBuffer modelBuffer, ImageSearcher.ImageSearcherOptions )
Cria uma instância de ImageSearcher
com um buffer de modelo e ImageSearcher.ImageSearcherOptions
.
Parâmetros
modeloBuffer | um ByteBuffer direto ou um MappedByteBuffer do modelo de pesquisa |
---|---|
opções |
Lança
Exceção de argumento ilegal | se o buffer do modelo não for um ByteBuffer direto ou um MappedByteBuffer |
---|---|
IOException | se ocorrer um erro de E/S ao carregar o arquivo de índice |
IllegalStateException | se houver um erro interno |
Exceção de tempo de execução | se houver um erro não especificado |
public static ImageSearcher createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, opções ImageSearcher.ImageSearcherOptions , int indexFd)
Parâmetros
modeloBuffer | |
---|---|
opções | |
indexFd |
public static ImageSearcher createFromFileAndOptions (contexto de contexto, String modelPath, opções ImageSearcher.ImageSearcherOptions )
Cria uma instância de ImageSearcher
de ImageSearcher.ImageSearcherOptions
.
Parâmetros
contexto | |
---|---|
modelPath | caminho do modelo de pesquisa com metadados nos ativos |
opções |
Lança
IOException | se ocorrer um erro de E/S ao carregar o modelo tflite ou o arquivo de índice |
---|---|
Exceção de argumento ilegal | se um argumento for inválido |
IllegalStateException | se houver um erro interno |
Exceção de tempo de execução | se houver um erro não especificado |
public static ImageSearcher createFromFileAndOptions ( Opções de arquivo modelFile, ImageSearcher.ImageSearcherOptions )
Cria uma instância de ImageSearcher
.
Parâmetros
modelFile | a instância File do modelo de pesquisa |
---|---|
opções |
Lança
IOException | se ocorrer um erro de E/S ao carregar o modelo tflite ou o arquivo de índice |
---|---|
Exceção de argumento ilegal | se um argumento for inválido |
IllegalStateException | se houver um erro interno |
Exceção de tempo de execução | se houver um erro não especificado |
public List < NearestNeighbor > search ( imagem TensorImage , opções ImageProcessingOptions )
Executa a extração de incorporação no TensorImage
fornecido com ImageProcessingOptions
, seguido pela pesquisa do vizinho mais próximo no índice.
ImageSearcher
suporta as seguintes opções:
- Região de interesse (ROI) (por meio de
ImageProcessingOptions.Builder.setRoi(Rect)
). O padrão é a imagem inteira. - rotação da imagem (por meio
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). O padrão éImageProcessingOptions.Orientation.TOP_LEFT
.
ImageSearcher
oferece suporte aos seguintes tipos de espaço de cores TensorImage
:
Parâmetros
imagem | um objeto UINT8 TensorImage que representa uma imagem RGB ou YUV |
---|---|
opções |
Lança
Exceção de argumento ilegal | se o tipo de imagem do espaço de cores não for compatível |
---|
Public List < NearestNeighbor > search ( imagem MlImage )
Executa a extração de incorporação no MlImage
fornecido, seguido pela pesquisa do vizinho mais próximo no índice.
Parâmetros
imagem | um objeto MlImage 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 |
---|
public List < NearestNeighbor > search ( imagem MlImage , opções ImageProcessingOptions )
Executa a extração de incorporação no MlImage
fornecido com ImageProcessingOptions
, seguido pela pesquisa do vizinho mais próximo no índice.
ImageSearcher
suporta as seguintes opções:
- Região de interesse (ROI) (por meio de
ImageProcessingOptions.Builder.setRoi(Rect)
). O padrão é a imagem inteira. - rotação da imagem (por meio
ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation)
). O padrão éImageProcessingOptions.Orientation.TOP_LEFT
.MlImage.getRotation()
não é eficaz.
Parâmetros
imagem | um objeto MlImage 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 List < NearestNeighbor > search ( imagem TensorImage )
Executa a extração de incorporação no TensorImage
fornecido, seguido pela pesquisa do vizinho mais próximo no índice.
ImageSearcher
oferece suporte aos seguintes tipos de espaço de cores TensorImage
:
Parâmetros
imagem | um objeto UINT8 TensorImage que representa uma imagem RGB ou YUV |
---|
Lança
Exceção de argumento ilegal | se o tipo de imagem do espaço de cores não for compatível |
---|