Aan de slagGa gratis aan de slag

Beeldclassificatie

In deze oefening gebruik je dezelfde Flickr-gegevensset als eerder, met 30.000 afbeeldingen en bijbehorende bijschriften. Nu ga je de afbeelding classificeren met een vooraf getraind model.

Photo of 2 people, 1 is playing the guitar

De voorbeeldafbeelding (image) en de pipeline-module (pipeline) zijn geladen.

Deze oefening maakt deel uit van de cursus

Multi-modale modellen met Hugging Face

Cursus bekijken

Oefeninstructies

  • Laad de image-classification-pipeline met het vooraf getrainde model google/mobilenet_v2_1.0_224.
  • Gebruik de pipeline om de klasse van de voorbeeldafbeelding te voorspellen.
  • Print het eerste (hoogste waarschijnlijkheid) label.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# 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:", ____)
Code bewerken en uitvoeren