V

Vit Base Cats Vs Dogs

由 ismgar01 开发
基于ViT架构的猫狗图像分类模型,在cats_vs_dogs数据集上微调,准确率达99.37%
下载量 22
发布时间 : 3/2/2022
模型介绍
内容详情
替代品

模型简介

该模型是基于Google的ViT-base-patch16-224-in21k预训练模型,在cats_vs_dogs数据集上进行微调得到的图像分类模型,专门用于区分猫和狗的图片。

模型特点

高准确率
在cats_vs_dogs测试集上达到99.37%的分类准确率
基于ViT架构
使用Vision Transformer(ViT)架构,具有强大的图像特征提取能力
轻量级微调
基于预训练模型进行微调,仅需少量训练数据即可获得优异性能

模型能力

图像分类
猫狗识别
视觉特征提取

使用案例

宠物识别
宠物照片分类
自动识别照片中的动物是猫还是狗
分类准确率99.37%
宠物应用集成
可集成到宠物相关应用中提供自动分类功能
教育演示
机器学习教学
作为计算机视觉和迁移学习的教学案例