| language: en | |
| ## Description | |
| This model is a ViT trained to detect waste from images. | |
| The repository related to this model is: https://github.com/hectorLop/Waste-Detector | |
| This model was created using [Icevision](https://github.com/airctic/icevision), and all the code related to the training can be found here https://github.com/hectorLop/Waste-Detector/tree/main/waste_detector/object_detection/sagemaker | |
| ### Requirements | |
| - It is an EffientDet D1 model | |
| - Works with RBG images of size 512x512 |