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