Phân loại ảnh
Trong bài tập này, bạn sẽ dùng lại bộ dữ liệu flickr như trước, gồm 30.000 ảnh và chú thích đi kèm. Giờ bạn sẽ phân loại ảnh bằng một mô hình pretrained.
Ảnh mẫu (image) và module pipeline (pipeline) đã được nạp sẵn.
Bài tập này là một phần của khóa học
Mô hình đa phương thức với Hugging Face
Hướng dẫn bài tập
- Tải pipeline
image-classificationvới mô hình pretrainedgoogle/mobilenet_v2_1.0_224. - Dùng pipeline để dự đoán lớp của ảnh mẫu.
- In ra
labelđầu tiên (có xác suất cao nhất).
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# 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:", ____)