Downloads · 30 days
48
2% of all-time downloads
vit-indian-food
therealcyberlord/vit-indian-food
vit-indian-food is a image classification model from therealcyberlord. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
This model is a fine-tuned Vision Transformer (ViT) for the task of classifying images of Indian foods. The model was trained on the Indian Foods Dataset from Hugging Face Datasets.
All-time downloads
2.7K
Public
Parameters
85.8M
3.1 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt687 MB · 67%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Apr 7, 2024
- Updated
- Apr 9, 2024
- SHA
- 85ebd167
Model card
From the Hugging Face model README
Indian Food Classification with Vision Transformer (ViT)
Overview
This model is a fine-tuned Vision Transformer (ViT) for the task of classifying images of Indian foods. The model was trained on the Indian Foods Dataset from Hugging Face Datasets.
Dataset
The Indian Foods Dataset contains 4,770 images across 15 different classes of popular Indian dishes. The dataset is split into:
- Training: 3,047 images
- Validation: 762 images
- Testing: 961 images
Model
The base model used is the vision transformer (google/vit-base-patch16-224-in21k). The model was fine-tuned on the Indian Foods Dataset for 10 epochs using the AdamW optimizer with a learning rate of 2e-4.
Evaluation
The model was evaluated on the test set and achieved the following metrics:
- Accuracy: 0.9667
- Precision: 0.9670
- Recall: 0.9667
Usage
You can use this pre-trained model directly from Hugging Face
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Apr 7, 2024
- Updated
- Apr 9, 2024