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 em lote não é suportada (
batch
deve ser 1). - apenas entradas RGB são suportadas (
channels
devem ter 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 recursos 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) seja suportado.
Classes aninhadas
aula | ImageSearcher.ImageSearcherOptions | Opções para configurar um ImageSearcher. |
Métodos Públicos
Métodos herdados
Métodos Públicos
public static ImageSearcher createFromBufferAndOptions (opções ByteBuffer modelBuffer, ImageSearcher.ImageSearcherOptions )
Cria uma instância 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 | |
índiceFd |
public static ImageSearcher createFromFileAndOptions (contexto de contexto, String modelPath, opções ImageSearcher.ImageSearcherOptions )
Cria uma instância ImageSearcher
de ImageSearcher.ImageSearcherOptions
.
Parâmetros
contexto | |
---|---|
caminho do modelo | 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 ImageSearcher
.
Parâmetros
arquivomodelo | 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 |
lista pública <NeighborNeighbor> pesquisa (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
oferece suporte às seguintes opções:
- Região de interesse (ROI) (por meio de
ImageProcessingOptions.Builder.setRoi(Rect)
). O padrão é a imagem inteira. - rotação de imagem (por meio de
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 |
---|
lista pública < Neighbor mais próximo > pesquisa (imagem MlImage )
Executa a extração de incorporação no MlImage
fornecido, seguida 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 |
---|
lista pública <NearestNeighbor> pesquisa (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
oferece suporte às seguintes opções:
- Região de interesse (ROI) (por meio de
ImageProcessingOptions.Builder.setRoi(Rect)
). O padrão é a imagem inteira. - rotação de imagem (por meio de
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 |
---|
lista pública < Neighbor mais próximo > pesquisa (imagem TensorImage )
Executa a extração de incorporação no TensorImage
fornecido, seguida 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 |
---|