跳到正文
原文
Hugging Face Blog·· 2022-02-11AI 评分28

Hugging Face 演示如何使用 Transformers 微调 ViT 进行图像分类

Fine-Tune ViT for Image Classification with 🤗 Transformers

AI 导读

Hugging Face 展示利用 🤗 datasets 加载 beans 数据集,并通过 🤗 transformers 微调预训练模型 google/vit-base-patch16-224-in21k。该流程使用 ViTImageProcessor 将图像转换为张量输入,实现 Vision Transformer (ViT) 在图像分类任务上的应用。

来源:Hugging Face Blog · huggingface.co