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

Downloads last month: -; Downloads are not tracked for this model. How to track

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.