Skip to content

Latest commit

 

History

History
40 lines (30 loc) · 1.12 KB

File metadata and controls

40 lines (30 loc) · 1.12 KB

🗑️ Garbage Classification with Transfer Learning

This project was developed during the Shell-Edunet Skills4Future Internship (June–July 2025). It aims to classify garbage images into six categories using deep learning and transfer learning, achieving up to 98% accuracy.

🧠 Objective

Automatically classify garbage into:

  • Cardboard
  • Glass
  • Metal
  • Paper
  • Plastic
  • Trash

🗃️ Dataset

  • Folder: TrashType_Image_Dataset
  • Loaded via ImageDataGenerator with train/validation split
  • Includes preprocessing and augmentation for better generalization

🧪 Model Details

  • Base Models: EfficientNetV2B2 (primary), MobileNetV2 (for comparison)
  • Framework: TensorFlow / Keras
  • Accuracy Achieved: ✅ 98%

📦 Project Structure

Garbage_Classification/ ├── Week1/ ├── Week2/ ├── Dataset/ └── README.md

🚀 Highlights

  • High-accuracy image classification (98%)
  • Transfer learning with EfficientNetV2B2
  • Deployed via Gradio / Streamlit for live predictions

🔗 Repository

🔗 GitHub Link