ImageSearcher

classe finale publique ImageSearcher

Effectue une recherche de similarité sur les images.

L’API attend un modèle TFLite avec des métadonnées de modèle TFLite facultatives, mais fortement recommandées. .

  • Tenseur d'image d'entrée ( kTfLiteUInt8 / kTfLiteFloat32 )
    • entrée d'image de taille [batch x height x width x channels] .
    • L'inférence par lots n'est pas prise en charge ( batch doit être 1).
    • seules les entrées RVB sont prises en charge ( channels doivent être au nombre de 3).
    • si le type est kTfLiteFloat32 , les NormalizationOptions doivent être attachées aux métadonnées pour la normalisation des entrées.
  • Tenseur de sortie ( kTfLiteUInt8 / kTfLiteFloat32 )
    • N composants correspondant aux N dimensions du vecteur de caractéristiques renvoyé pour cette couche de sortie.
    • Soit 2 ou 4 dimensions, soit [1 x N] ou [1 x 1 x 1 x N] .

TODO(b/180502532) : ajouter un pointeur vers un exemple de modèle.

TODO(b/222671076) : ajoutez des méthodes de création d'usine sans options, telles que `createFromFile`, une fois que le format de fichier unique (fichier d'index emballé dans le modèle) est pris en charge.

Classes imbriquées

classe ImageSearcher.ImageSearcherOptions Options de configuration d'un ImageSearcher.

Méthodes publiques

ImageSearcher statique
createFromBufferAndOptions (options ByteBuffer modelBuffer, ImageSearcher.ImageSearcherOptions )
Crée une instance ImageSearcher avec un tampon de modèle et ImageSearcher.ImageSearcherOptions .
ImageSearcher statique
createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, options ImageSearcher.ImageSearcherOptions , int indexFd)
ImageSearcher statique
createFromFileAndOptions (contexte contextuel, chaîne modelPath, options ImageSearcher.ImageSearcherOptions )
Crée une instance ImageSearcher à partir de ImageSearcher.ImageSearcherOptions .
ImageSearcher statique
createFromFileAndOptions (options de fichier modelFile, ImageSearcher.ImageSearcherOptions )
Crée une instance ImageSearcher .
Liste < Voisin le plus proche >
recherche (image TensorImage , options ImageProcessingOptions )
Effectue une extraction d'intégration sur le TensorImage fourni avec ImageProcessingOptions , suivie d'une recherche du voisin le plus proche dans l'index.
Liste < Voisin le plus proche >
recherche (image MlImage )
Effectue une extraction d'intégration sur le MlImage fourni, suivie d'une recherche du voisin le plus proche dans l'index.
Liste < Voisin le plus proche >
recherche (image MlImage , options ImageProcessingOptions )
Effectue une extraction d'intégration sur le MlImage fourni avec ImageProcessingOptions , suivie d'une recherche du voisin le plus proche dans l'index.
Liste < Voisin le plus proche >
recherche (image TensorImage )
Effectue une extraction d'intégration sur le TensorImage fourni, suivie d'une recherche du voisin le plus proche dans l'index.

Méthodes héritées

Méthodes publiques

public static ImageSearcher createFromBufferAndOptions (options ByteBuffer modelBuffer, ImageSearcher.ImageSearcherOptions )

Crée une instance ImageSearcher avec un tampon de modèle et ImageSearcher.ImageSearcherOptions .

Paramètres
modèleBuffer un ByteBuffer direct ou un MappedByteBuffer du modèle de recherche
choix
Jetés
Exception d'argument illégal si le tampon de modèle n'est pas un ByteBuffer direct ou un MappedByteBuffer
IOException si une erreur d'E/S se produit lors du chargement du fichier d'index
IllegalStateException s'il y a une erreur interne
Exception d'exécution s'il y a une erreur autrement non spécifiée

public static ImageSearcher createFromBufferAndOptionsImpl ( ByteBuffer modelBuffer, options ImageSearcher.ImageSearcherOptions , int indexFd)

Paramètres
modèleBuffer
choix
indexFd

public static ImageSearcher createFromFileAndOptions (contexte contextuel, chaîne modelPath, options ImageSearcher.ImageSearcherOptions )

Crée une instance ImageSearcher à partir de ImageSearcher.ImageSearcherOptions .

Paramètres
contexte
modèleChemin chemin du modèle de recherche avec métadonnées dans les ressources
choix
Jetés
IOException si une erreur d'E/S se produit lors du chargement du modèle tflite ou du fichier d'index
Exception d'argument illégal si un argument n'est pas valide
IllegalStateException s'il y a une erreur interne
Exception d'exécution s'il y a une erreur autrement non spécifiée

public static ImageSearcher createFromFileAndOptions (options de fichier modelFile, ImageSearcher.ImageSearcherOptions )

Crée une instance ImageSearcher .

Paramètres
fichier modèle l'instance File du modèle de recherche
choix
Jetés
IOException si une erreur d'E/S se produit lors du chargement du modèle tflite ou du fichier d'index
Exception d'argument illégal si un argument n'est pas valide
IllegalStateException s'il y a une erreur interne
Exception d'exécution s'il y a une erreur autrement non spécifiée

liste publique < NearestNeighbor > recherche (image TensorImage , options ImageProcessingOptions )

Effectue une extraction d'intégration sur le TensorImage fourni avec ImageProcessingOptions , suivie d'une recherche du voisin le plus proche dans l'index.

ImageSearcher prend en charge les options suivantes :

ImageSearcher prend en charge les types d'espace colorimétrique TensorImage suivants :

Paramètres
image un objet UINT8 TensorImage qui représente une image RVB ou YUV
choix
Jetés
Exception d'argument illégal si le type d'espace colorimétrique de l'image n'est pas pris en charge

liste publique < NearestNeighbour > recherche (image MlImage )

Effectue une extraction d'intégration sur le MlImage fourni, suivie d'une recherche du voisin le plus proche dans l'index.

Paramètres
image un objet MlImage qui représente une image
Jetés
Exception d'argument illégal si le type ou le format de stockage de l'image n'est pas pris en charge

liste publique < NearestNeighbor > recherche (image MlImage , options ImageProcessingOptions )

Effectue une extraction d'intégration sur le MlImage fourni avec ImageProcessingOptions , suivie d'une recherche du voisin le plus proche dans l'index.

ImageSearcher prend en charge les options suivantes :

Paramètres
image un objet MlImage qui représente une image
choix configure les options, y compris le retour sur investissement et la rotation
Jetés
Exception d'argument illégal si le type ou le format de stockage de l'image n'est pas pris en charge

liste publique < NearestNeighbour > recherche (image TensorImage )

Effectue une extraction d'intégration sur le TensorImage fourni, suivie d'une recherche du voisin le plus proche dans l'index.

ImageSearcher prend en charge les types d'espace colorimétrique TensorImage suivants :

Paramètres
image un objet UINT8 TensorImage qui représente une image RVB ou YUV
Jetés
Exception d'argument illégal si le type d'espace colorimétrique de l'image n'est pas pris en charge