ComeçarComece de graça

Classificação de imagens

Neste exercício, você vai usar o mesmo conjunto de dados do Flickr que usou antes, que tem 30.000 imagens e legendas associadas. Agora você vai classificar a imagem usando um modelo pré-treinado.

Foto de duas pessoas, uma delas tocando violão

A imagem de exemplo (image) e o módulo do pipeline (pipeline) foram carregados.

Este exercício faz parte do curso

Modelos multimodais com Hugging Face

Ver curso

Instruções do exercício

  • Carregue o pipeline image-classification com o modelo pré-treinado google/mobilenet_v2_1.0_224.
  • Use o pipeline para prever a classe da imagem de amostra.
  • Imprima o primeiro (com maior probabilidade) label.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

# 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:", ____)
Editar e executar o código