My Trash Classification
An image classification solution implemented using Hugging Face's pre-trained Vision Transformer (ViT) model, capable of classifying images into six types of garbage
Downloads 259
Release Time : 12/22/2024
Model Overview
This model is fine-tuned based on a pre-trained ViT model, specifically designed for garbage image classification tasks. It can identify six categories of garbage: cardboard, glass, metal, paper, plastic, and other waste
Model Features
High Accuracy Classification
Achieves a classification accuracy of 94.26% on the test set
Based on ViT Architecture
Utilizes the Vision Transformer (ViT) model with powerful image feature extraction capabilities
Six-category Garbage Recognition
Can distinguish between cardboard, glass, metal, paper, plastic, and other waste
Model Capabilities
Image Classification
Garbage Recognition
Visual Feature Extraction
Use Cases
Garbage Sorting
Smart Garbage Sorting System
Used in the image recognition module of automated garbage sorting devices
Accurately identifies garbage categories, improving sorting efficiency
Environmental Education App
Helps users identify garbage types in mobile applications
Enhances public awareness of garbage classification
Featured Recommended AI Models
Š 2025AIbase