M

My Trash Classification

Developed by tribber93
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
AIbase
Empowering the Future, Your AI Solution Knowledge Base
Š 2025AIbase