ComenzarEmpieza gratis

Clasificación de imágenes

En este ejercicio, utilizarás el mismo conjunto de datos de Flickr que anteriormente, que contiene 30 000 imágenes y sus correspondientes pies de foto. Ahora clasificarás la imagen utilizando un modelo preentrenado.

Foto de dos personas, una de ellas tocando la guitarra.

Se han cargado la imagen de muestra (image) y el módulo de canalización (pipeline).

Este ejercicio forma parte del curso

Modelos multimodales con Hugging Face

Ver curso

Instrucciones del ejercicio

  • Carga el pipeline image-classification con el modelo preentrenado google/mobilenet_v2_1.0_224.
  • Utiliza el proceso para predecir la clase de la imagen de muestra.
  • Imprime el primero (el de mayor probabilidad) label.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

# 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 y ejecutar código