Garbage Classifier

Description The Recycling Assistant is an AI-powered application that helps users properly sort and dispose of waste items. Using computer vision and machine learning, it identifies different types of trash and provides guidance on the correct recycling or disposal method.

Features Real-time camera feed for scanning items

Image upload capability for offline analysis

AI-powered classification of 11 waste categories

User-friendly GUI with clear disposal instructions

Color-coded results for easy interpretation

Installation Clone the repository

Install required dependencies:

text pip install PyQt5 torch torchvision opencv-python pillow Ensure the model file trash_classifier_finetuned_20250405_003634_best.pth is in the root directory

Usage Run the application:

text python recycling_assistant.py Requirements Python 3.7+

PyQt5

PyTorch

torchvision

OpenCV

Pillow

For a pre trianed model, download from here: https://drive.google.com/file/d/1RsYzTVHEJq8FK1OVERfJdd1U6x_Jyifg/view?usp=sharing, otherwise you may train your own model with yoour own datasets.

License MIT License

Contributors Nicholas Tanner Tony Ren

For more information, please refer to the documentation.