ÎncepețiÎncepe gratuit

Clasificarea imaginilor

În acest exercițiu, vei folosi același set de date flickr ca anterior, care conține 30.000 de imagini și descrierile asociate. Acum vei clasifica imaginea folosind un model preantrenat.

Fotografie cu 2 persoane, una cântând la chitară

Imagine exemplu (image) și modulul pipeline (pipeline) au fost deja încărcate.

Acest exercițiu face parte din cursul

Modele Multi-Modale cu Hugging Face

Vezi cursul

Instrucțiuni pentru exercițiu

  • Încarcă pipeline-ul image-classification cu modelul preantrenat google/mobilenet_v2_1.0_224.
  • Folosește pipeline-ul pentru a prezice clasa imaginii exemplu.
  • Afișează primul label (cel cu cea mai mare probabilitate).

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# 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:", ____)
Editează și rulează codul