Garbage Classifier

A deep learning model for classifying garbage images into 12 categories using ResNet50.

Categories

• Battery
• Biological
• Brown Glass
• Cardboard
• Clothes
• Green Glass
• Metal
• Paper
• Plastic
• Shoes
• Trash
• White Glass

Requirements

Install dependencies:

pip install -r requirements.txt

Usage

Training

python main.py

Inference

python inference.py

Model Files

• resnet50_phase1.pth - Phase 1 trained model
• resnet50_finetuned.pth - Fine-tuned model

Dataset

Expected dataset structure:

garbage_classification/
├── battery/
├── biological/
├── brown-glass/
├── cardboard/
├── clothes/
├── green-glass/
├── metal/
├── paper/
├── plastic/
├── shoes/
├── trash/
└── white-glass/