V

Vit Bach Demo

由 tcvrishank 开发
基于google/vit-base-patch16-224微调的视觉Transformer模型,适用于图像分类任务
下载量 16
发布时间 : 3/23/2023
模型介绍
内容详情
替代品

模型简介

该模型是基于ViT架构的图像分类模型,在图像文件夹数据集上进行了微调,可用于通用图像识别任务

模型特点

基于ViT架构
使用Vision Transformer架构,具有强大的图像特征提取能力
小规模微调
在基础模型上进行轻量级微调,适合特定图像分类任务
高效训练
使用混合精度训练和线性学习率调度器,训练效率高

模型能力

图像分类
特征提取

使用案例

计算机视觉
通用图像分类
对输入的图像进行分类识别