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, 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
Inference Providers
NEW
This model is not currently available via any of the supported third-party Inference Providers, and
HF Inference API was unable to determine this model's library.