Image classification
In this exercise, you will use the same flickr dataset as previously, which has 30,000 images and associated captions. Now you will classify the image using a pretrained model.
The sample image (image
) and pipeline module (pipeline
) have been loaded.
Diese Übung ist Teil des Kurses
Multi-Modal Models with Hugging Face
Anleitung zur Übung
- Load the
image-classification
pipeline withgoogle/mobilenet_v2_1.0_224
pretrained model. - Use the pipeline to predict the class of the sample image.
- Print the first (highest probability)
label
.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# 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:", ____)