TensorFlow est de retour sur Google I/O le 14 mai ! S'inscrire maintenant

Cette page a été traduite par l'API Cloud Translation.

ImageClassifier

classe finale publique ImageClassifier

Effectue une classification sur les images.

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

L'API prend en charge les modèles avec un tenseur d'entrée d'image et un tenseur de sortie de classification. Pour être plus précis, voici les exigences.

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 du score de sortie ( kTfLiteUInt8 / kTfLiteFloat32 )
- avec N classes de 2 ou 4 dimensions, telles que [1 x N] ou [1 x 1 x 1 x N]
- le fichier d'étiquette doit être compressé dans les métadonnées. Voir l' exemple de création de métadonnées pour un classificateur d'images . Si aucun fichier d'étiquette n'est compressé, il utilisera l'index comme étiquette dans le résultat.

Un exemple d'un tel modèle peut être trouvé sur TensorFlow Hub. .

Classes imbriquées

classe ImageClassifier.ImageClassifierOptions Options de configuration d'un ImageClassifier.

Méthodes publiques

Liste <Classifications>	classer (image TensorImage ) Effectue une classification réelle sur le `TensorImage` fourni.
Liste <Classifications>	classer (image TensorImage , options ImageProcessingOptions ) Effectue une classification réelle sur le `TensorImage` fourni avec `ImageProcessingOptions` .
Liste <Classifications>	classer (image MlImage ) Effectue une classification réelle sur le `MlImage` fourni.
Liste <Classifications>	classer (image MlImage , options ImageProcessingOptions ) Effectue une classification réelle sur le `MlImage` fourni avec `ImageProcessingOptions` .
ImageClassifier statique	createFromBuffer ( ByteBuffer modelBuffer) Crée une instance `ImageClassifier` avec un tampon de modèle et le `ImageClassifier.ImageClassifierOptions` par défaut.
ImageClassifier statique	createFromBufferAndOptions (options ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions ) Crée une instance `ImageClassifier` avec un tampon de modèle et `ImageClassifier.ImageClassifierOptions` .
ImageClassifier statique	createFromFile (contexte contextuel, String modelPath) Crée une instance `ImageClassifier` à partir du `ImageClassifier.ImageClassifierOptions` par défaut.
ImageClassifier statique	createFromFile ( Fichier modèleFichier) Crée une instance `ImageClassifier` à partir du `ImageClassifier.ImageClassifierOptions` par défaut.
ImageClassifier statique	createFromFileAndOptions (contexte contextuel, chaîne modelPath, options ImageClassifier.ImageClassifierOptions ) Crée une instance `ImageClassifier` à partir de `ImageClassifier.ImageClassifierOptions` .
ImageClassifier statique	createFromFileAndOptions (options de fichier modelFile, ImageClassifier.ImageClassifierOptions ) Crée une instance `ImageClassifier` .

Méthodes héritées

De la classe org.tensorflow.lite.task.core.BaseTaskApi

vide synchronisé	fermer () Libérez la mémoire allouée par C++ et désenregistrez la bibliothèque du support statique.
long	getNativeHandle ()
booléen	est fermé ()

De la classe java.lang.Object

booléen	est égal à ( Objet arg0)
Classe finale <?>	obtenirClasse ()
int	Code de hachage ()
vide final	notifier ()
vide final	notifierTous ()
Chaîne	àChaîne ()
vide final	attendre (long arg0, int arg1)
vide final	attendez (long arg0)
vide final	attendez ()

Depuis l'interface java.io.Closeable

vide abstrait

fermer ()

Depuis l'interface java.lang.AutoCloseable

vide abstrait

fermer ()

Méthodes publiques

liste publique <Classifications> classifier (image TensorImage )

Effectue une classification réelle sur le TensorImage fourni.

ImageClassifier 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

liste publique <Classifications> classifier (image TensorImage , options ImageProcessingOptions )

Effectue une classification réelle sur le TensorImage fourni avec ImageProcessingOptions .

ImageClassifier prend en charge les options suivantes :

Région d'intérêt (ROI) (via ImageProcessingOptions.Builder.setRoi(Rect) ). Il s'agit par défaut de l'image entière.
rotation de l'image (via ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation) ). La valeur par défaut est ImageProcessingOptions.Orientation.TOP_LEFT .

ImageClassifier 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 <Classifications> classifier (image MlImage )

Effectue une classification réelle sur le MlImage fourni.

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 <Classifications> classifier (image MlImage , options ImageProcessingOptions )

Effectue une classification réelle sur le MlImage fourni avec ImageProcessingOptions .

ImageClassifier prend en charge les options suivantes :

Région d'intérêt (ROI) (via ImageProcessingOptions.Builder.setRoi(Rect) ). Il s'agit par défaut de l'image entière.
rotation de l'image (via ImageProcessingOptions.Builder.setOrientation(ImageProcessingOptions.Orientation) ). La valeur par défaut est ImageProcessingOptions.Orientation.TOP_LEFT . MlImage.getRotation() n'est pas efficace.

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

public static ImageClassifier createFromBuffer ( ByteBuffer modelBuffer)

Crée une instance ImageClassifier avec un tampon de modèle et le ImageClassifier.ImageClassifierOptions par défaut.

Paramètres

modèleBuffer	un `ByteBuffer` direct ou un `MappedByteBuffer` du modèle de classification

Jetés

Exception d'argument illégal	si le tampon de modèle n'est pas un `ByteBuffer` direct ou un `MappedByteBuffer`
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 ImageClassifier createFromBufferAndOptions (options ByteBuffer modelBuffer, ImageClassifier.ImageClassifierOptions )

Crée une instance ImageClassifier avec un tampon de modèle et ImageClassifier.ImageClassifierOptions .

Paramètres

modèleBuffer	un `ByteBuffer` direct ou un `MappedByteBuffer` du modèle de classification
choix

Jetés

Exception d'argument illégal	si le tampon de modèle n'est pas un `ByteBuffer` direct ou un `MappedByteBuffer`
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 ImageClassifier createFromFile (contexte contextuel, String modelPath)

Crée une instance ImageClassifier à partir du ImageClassifier.ImageClassifierOptions par défaut.

Paramètres

contexte
modèleChemin	chemin du modèle de classification avec métadonnées dans les actifs

Jetés

IOException	si une erreur d'E/S se produit lors du chargement du modèle tflite
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 ImageClassifier createFromFile ( Fichier modelFile)

Crée une instance ImageClassifier à partir du ImageClassifier.ImageClassifierOptions par défaut.

Paramètres

fichier modèle	le modèle de classification Instance `File`

Jetés

IOException	si une erreur d'E/S se produit lors du chargement du modèle tflite
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 ImageClassifier createFromFileAndOptions (contexte de contexte, String modelPath, options ImageClassifier.ImageClassifierOptions )

Crée une instance ImageClassifier à partir de ImageClassifier.ImageClassifierOptions .

Paramètres

contexte
modèleChemin	chemin du modèle de classification avec métadonnées dans les actifs
choix

Jetés

IOException	si une erreur d'E/S se produit lors du chargement du modèle tflite
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 ImageClassifier createFromFileAndOptions (options de fichier modelFile, ImageClassifier.ImageClassifierOptions )

Crée une instance ImageClassifier .

Paramètres

fichier modèle	le modèle de classification Instance `File`
choix

Jetés

IOException	si une erreur d'E/S se produit lors du chargement du modèle tflite
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