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

使用 Hugging Face Transformers 微调 ViT 进行图像分类

Fine-Tune ViT for Image Classification with 🤗 Transformers

AI 导读

教程演示利用 Hugging Face datasets 和 transformers 库微调 Vision Transformer (ViT) 处理图像分类。使用 google/vit-base-patch16-224-in21k 模型及其配套的 ViTImageProcessor,通过加载 beans 数据集并应用标准化预处理,将图像转换为模型可接受的张量输入以完成微调流程。

来源:Hugging Face Blog · huggingface.co