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

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

Fine-Tune ViT for Image Classification with 🤗 Transformers

AI 导读

本文介绍如何使用 🤗 Transformers 微调 Vision Transformer (ViT) 进行图像分类。通过加载 beans 数据集,利用 ViTImageProcessor 将图像转换为像素值张量,并对预训练的 google/vit-base-patch16-224-in21k 模型进行微调。

来源:Hugging Face Blog · huggingface.co