Downloads · 30 days
12
1% of all-time downloads
fast_food_image_detection
dima806/fast_food_image_detection
fast_food_image_detection is a image classification model from dima806. 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.
Returns fast food type based on an image with about 98% accuracy.
All-time downloads
1.3K
Public
Parameters
85.8M
3.4 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt2.1 GB · 55%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Sep 17, 2023
- Updated
- Oct 19, 2024
- SHA
- ca4f6d51
Base models
Model card
From the Hugging Face model README
Returns fast food type based on an image with about 98% accuracy.
See https://www.kaggle.com/code/dima806/fast-food-image-detection-vit for more details.
Classification report:
precision recall f1-score support
Burger 0.9466 0.9750 0.9606 400
Taco 0.9578 0.9650 0.9614 400
Baked Potato 0.9827 0.9925 0.9876 400
Hot Dog 0.9872 0.9698 0.9784 397
Pizza 0.9875 0.9875 0.9875 400
Sandwich 0.9724 0.9724 0.9724 399
Fries 0.9748 0.9675 0.9711 400
Donut 0.9827 1.0000 0.9913 397
Crispy Chicken 0.9822 0.9650 0.9735 400
Taquito 0.9923 0.9700 0.9810 400
accuracy 0.9765 3993
macro avg 0.9766 0.9765 0.9765 3993
weighted avg 0.9766 0.9765 0.9765 3993
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Sep 17, 2023
- Updated
- Oct 19, 2024