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