Hugging Face Blog·· 2022-02-11AI 评分45
用 🤗 Transformers 微调 ViT 进行图像分类
Fine-Tune ViT for Image Classification with 🤗 Transformers
AI 导读
这是一篇 Hugging Face 教程,讲解如何用 🤗 datasets 加载图像分类数据集(示例为包含健康与病态豆叶的 beans 数据集,共 3 个类别),再用 🤗 transformers 微调预训练的 Vision Transformer(ViT)模型。
来源:Hugging Face Blog · huggingface.co