watersplash/waste-classification

Model Card for Model ID

An Image Classifier model fine-tuned on ViT. This model can classify garbage images.

Model Details

Model Description

• Finetuned from model : ViT

Model Sources [optional]

• Repository: https://github.com/KomaliValluru/waste-classification

Uses

• Target classes: Battery, Biological, Brown-grass, Cardboard, Clothes, Green-Glass, Metal, Paper, Plastic, Shoes, Trash, White-Glass

Training Details

Training Data

https://www.kaggle.com/datasets/mostafaabla/garbage-classification

Metrics

Accuracy

Results

Accuracy: 98%

Summary

• Hours used: 1 hour 30 minutes
• References: Based on the model yangy50/garbage-classification