Edit model card

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]

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
Downloads last month
29
Safetensors
Model size
85.8M params
Tensor type
F32
ยท
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Space using watersplash/waste-classification 1