| 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 | |