File size: 470 Bytes
efef84f 7cc5a74 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | ---
language: en
---
## Description
This model is a ViT trained to classify waste images into 6 categories:
- Organic
- Carton
- Glass
- General
- Plastics
- Dangerous.
The repository related to this model is: https://github.com/hectorLop/Waste-Detector
Also, the code related to this model can be found here https://github.com/hectorLop/Waste-Detector/blob/main/waste_detector/classifier/sagemaker/model.py
### Requirements
- Works with RGB images of size 224x224
|