Hugging Face Blog·· 2022-02-11精選AI 評分62
使用 🤗 Transformers 微調 ViT 進行圖像分類
Fine-Tune ViT for Image Classification with 🤗 Transformers
AI 導讀
這篇教程示範使用 🤗 Datasets 與 🤗 Transformers，將預訓練模型 google/vit-base-patch16-224-in21k 微調為豆類葉片圖像分類器。
推薦理由
文章以 beans 葉片分類串起資料載入、ViT 圖像預處理、Trainer 微調與評估，並展示資料即時轉換及訓練參數設定。
來源：Hugging Face Blog · huggingface.co