Skip to content

Sritama401/Garbage_Classification

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🗑️ 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

About

AICTE INTERNSHIP PROJECT

Resources

Code of conduct

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 100.0%