File size: 505 Bytes
f9e3f8b
 
 
 
 
 
 
47e6e53
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
---
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