ПочатиПочніть безкоштовно

Класифікація зображень

У цій вправі ви використаєте той самий набір даних flickr, що й раніше, у ньому 30 000 зображень і підписів до них. Тепер ви класифікуєте зображення за допомогою попередньо натренованої моделі.

Photo of 2 people, 1 is playing the guitar

Зразок зображення (image) і модуль конвеєра (pipeline) уже завантажено.

Ця вправа є частиною курсу

Багатомодальні моделі з Hugging Face

Переглянути курс

Інструкції до вправи

  • Завантажте конвеєр image-classification з попередньо натренованою моделлю google/mobilenet_v2_1.0_224.
  • Використайте конвеєр, щоб передбачити клас зразка зображення.
  • Виведіть першу label (з найвищою ймовірністю).

Інтерактивна практична вправа

Спробуйте виконати цю вправу, доповнивши цей зразок коду.

# 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:", ____)
Редагувати та запускати код