Bildklassifizierung
In dieser Übung wirst du wieder den gleichen Flickr-Datensatz wie vorher verwenden, der 30.000 Bilder und dazugehörige Bildunterschriften enthält. Jetzt klassifizierst du das Bild mit einem vorab trainierten Modell.
Das Beispielbild (image) und das Pipeline-Modul (pipeline) wurden geladen.
Diese Übung ist Teil des Kurses
Multimodale Modelle mit Hugging Face
Anleitung zur Übung
- Lade die Pipeline „
image-classification“ mit dem vortrainierten Modell „google/mobilenet_v2_1.0_224“. - Benutz die Pipeline, um die Klasse des Beispielbildes vorherzusagen.
- Druck die erste (wahrscheinlichste) „
label“ aus.
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:", ____)