Downloads · 30 days
12
2% of all-time downloads
vit-recycling-classification
jacobthebanana/vit-recycling-classification
vit-recycling-classification is a image classification model from jacobthebanana. Use it when you need a label for an image. It is set up for transformers.
Technical details: - Base model: google/vit-base-patch16-224 - Dataset: [Recycling Classification (12 classes)] (https://www.kaggle.com/datasets/mostafaabla/garbage-classification) - Learning rate: 0.0001 - Effective…
All-time downloads
488
Public
Repo size
690 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.msgpack343 MB · 100%
At a glance
- Task
- Image Classification
- Library
- transformers
- Model type
- vit
- Access
- Public
- Created
- Nov 26, 2022
- Updated
- Nov 27, 2022
- SHA
- 260bd61b
Model card
From the Hugging Face model README
Technical details:
- Base model:
google/vit-base-patch16-224 - Dataset: [Recycling Classification (12 classes)] (https://www.kaggle.com/datasets/mostafaabla/garbage-classification)
- Learning rate: 0.0001
- Effective training batch size: 16 (2 items per TPU core over 8 cores.)
Research supported with Cloud TPUs from Google's TPU Research Cloud (TRC)
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- Created
- Nov 26, 2022
- Updated
- Nov 27, 2022