Downloads · 30 days
9
1% of all-time downloads
food_type_image_detection
ewanlong/food_type_image_detection
food_type_image_detection is a image classification model from ewanlong. Use it when you need a label for an image. It is set up for transformers.
- Model Name: Food Type Image Detection Vision Transformer - Original Model: Vision Transformer (ViT) model pre-trained on ImageNet-21k (14 million images, 21,843 classes) at resolution 224x224. - It was introduced in…
All-time downloads
1.4K
Public
Parameters
85.8M
1 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt687 MB · 50%
At a glance
- Task
- Image Classification
- Library
- transformers
- Model type
- vit
- Access
- Public
- Created
- Apr 6, 2024
- Updated
- Apr 10, 2024
- SHA
- 3eae94d8
Model card
From the Hugging Face model README
Model Card
- Model Name: Food Type Image Detection Vision Transformer
- Original Model: Vision Transformer (ViT) model pre-trained on ImageNet-21k (14 million images, 21,843 classes) at resolution 224x224.
- It was introduced in the paper An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale by Dosovitskiy et al. and first released in this repository.
- Images are presented to the model as a sequence of fixed-size patches (resolution 16x16), which are linearly embedded.
- Does not provide any fine-tuned heads, as these were zero'd by Google researchers.
- Model Type: Image Classification
- Model Architecture: Vision Transformer (ViT)
- Fine-tuning:
- Fine-tuned on Food Image Classification Dataset by using 12 varieties of these 35 varieties
- Optimizer: AdamW
- Epochs: 20
- Model Performance: Achieved an accuracy of 96.23% on all of the kinds of Food Image Classification Dataset
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- Created
- Apr 6, 2024
- Updated
- Apr 10, 2024