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-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:", ____)