1. Học hỏi
  2. /
  3. Khoa Học
  4. /
  5. Mô hình đa phương thức với Hugging Face

Connected

Bài tập

Tạo mô tả ảnh bằng Pipeline

Trong bài tập này, bạn sẽ tiếp tục dùng dataset flickr với 30.000 ảnh và các chú thích (caption) đi kèm. Lần này, bạn sẽ tạo chú thích cho ảnh dưới đây bằng pipeline thay vì dùng các auto class.

Photo of a man standing on a ladder cleaning a window

Dataset (dataset) đã được nạp với cấu trúc sau:

Dataset({
    features: ['image', 'caption', 'sentids', 'split', 'img_id', 'filename'],
    num_rows: 10
})

Module pipeline (pipeline) đã được nạp.

Hướng dẫn

100 XP
  • Tải pipeline image-to-text với mô hình pretrained Salesforce/blip-image-captioning-base.
  • Dùng pipeline để tạo chú thích cho ảnh ở chỉ mục 3.