Vit Bach Demo
V
Vit Bach Demo
Developed by tcvrishank
A vision Transformer model fine-tuned based on google/vit-base-patch16-224, suitable for image classification tasks
Downloads 16
Release Time : 3/23/2023
Model Overview
This model is an image classification model based on the ViT architecture, fine-tuned on an image folder dataset, and can be used for general image recognition tasks
Model Features
Based on ViT architecture
Uses Vision Transformer architecture with powerful image feature extraction capabilities
Small-scale fine-tuning
Lightweight fine-tuning on the base model, suitable for specific image classification tasks
Efficient training
Uses mixed-precision training and linear learning rate scheduler for high training efficiency
Model Capabilities
Image classification
Feature extraction
Use Cases
Computer vision
General image classification
Classify and recognize input images
Featured Recommended AI Models