CommencerCommencez gratuitement

Classification d’images

Dans cet exercice, vous allez utiliser le même jeu de données flickr que précédemment, qui contient 30 000 images avec leurs légendes. Cette fois, vous allez classer l’image à l’aide d’un modèle préentraîné.

Photo de 2 personnes, l’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

<cours>Modèles multimodaux avec Hugging Face</cours>
Voir le cours

Instructions de l’exercice

  • Chargez le pipeline image-classification avec le modèle préentraîné google/mobilenet_v2_1.0_224.
  • Utilisez le pipeline pour prédire la classe de l’image d’exemple.
  • Affichez la première label (probabilité la plus élevée).

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# 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