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 is not currently available via any of the supported Inference Providers.

The model cannot be deployed to the HF Inference API: The model has no library tag.