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é.

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
Instructions
- Chargez le pipeline d'
image-classifications 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:", ____)