hlopez
/

ViT_waste_classifier

Model card Files Files and versions Community

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 isn't deployed by any Inference Provider. 🙋 Ask for provider support