CommencerCommencer gratuitement

Classification d'images

Dans cet exercice, vous utiliserez le même ensemble de données Flickr que précédemment, qui contient 30 000 images et leurs légendes associées. Vous allez maintenant classer l'image à l'aide d'un modèle pré-entraîné.

Photo de deux personnes, dont une joue de la guitare

L'image d'exemple (image) et le module de pipeline (pipeline) ont été chargés.

Cet exercice fait partie du cours

Modèles multimodaux avec Hugging Face

Afficher le cours

Instructions

  • Chargez le pipeline d'image-classification s avec le modèle pré-entraîné google/mobilenet_v2_1.0_224.
  • Veuillez utiliser le pipeline pour prédire la classe de l'image échantillon.
  • Imprimez la première (probabilité la plus élevée) label.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Load the image-classification pipeline
pipe = ____

# Use the pipeline to predict the class of the sample image
pred = ____

# Print the first (highest probability) label
print("Predicted class:", ____)
Modifier et exécuter le code