Image Classification
Transformers
TensorBoard
Safetensors
vit
food
fruits
junkfood
Generated from Trainer
Instructions to use gutkia01/vit-food-classification-gutkia01 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use gutkia01/vit-food-classification-gutkia01 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="gutkia01/vit-food-classification-gutkia01") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("gutkia01/vit-food-classification-gutkia01") model = AutoModelForImageClassification.from_pretrained("gutkia01/vit-food-classification-gutkia01", device_map="auto") - Notebooks
- Google Colab
- Kaggle
File size: 1,541 Bytes
e26ca67 3b659d3 e26ca67 966b941 e26ca67 6f23618 fb27214 e26ca67 6f23618 e26ca67 3b659d3 e26ca67 e2b18e1 e26ca67 e2b18e1 e26ca67 e2b18e1 e26ca67 e2b18e1 e26ca67 e2b18e1 e26ca67 e2b18e1 e26ca67 e2b18e1 e26ca67 e2b18e1 e26ca67 bdf8d12 e26ca67 e2b18e1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 | ---
library_name: transformers
license: apache-2.0
base_model: google/vit-base-patch16-224
tags:
- image-classification
- food
- fruits
- junkfood
- generated_from_trainer
datasets:
- imagefolder
model-index:
- name: vit-food-classification-gutkia01
results:
- task:
name: Image Classification
type: image-classification
dataset:
name: imagefolder
type: image
metrics:
- name: Accuracy
type: accuracy
value: 0.9998
---
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->
# vit-food-classification-gutkia01
This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the food-classification dataset.
## Model description
More information needed
## Intended uses & limitations
More information needed
## Training and evaluation data
More information needed
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 5e-05
- train_batch_size: 16
- eval_batch_size: 8
- seed: 42
- optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: linear
- num_epochs: 6
### Training results
### Framework versions
- Transformers 4.52.3
- Pytorch 2.6.0+cu124
- Datasets 3.6.0
- Tokenizers 0.21.1
|