https://github.com/tanishq-ctrl/waste-classification
A deep learning-based web application that classifies different types of waste materials using computer vision. The system helps in proper waste segregation by identifying whether an item belongs to categories like cardboard, glass, metal, paper, plastic, or trash.
https://github.com/tanishq-ctrl/waste-classification
computer-vision deep-learning environmental flask image-classification keras mobilenetv2 sustainability tensorflow transfer-learning waste-management waste-management-system web-application
Last synced: 2 months ago
JSON representation
A deep learning-based web application that classifies different types of waste materials using computer vision. The system helps in proper waste segregation by identifying whether an item belongs to categories like cardboard, glass, metal, paper, plastic, or trash.
- Host: GitHub
- URL: https://github.com/tanishq-ctrl/waste-classification
- Owner: tanishq-ctrl
- Created: 2025-01-17T08:05:56.000Z (4 months ago)
- Default Branch: main
- Last Pushed: 2025-01-17T09:57:11.000Z (4 months ago)
- Last Synced: 2025-01-27T06:34:50.287Z (4 months ago)
- Topics: computer-vision, deep-learning, environmental, flask, image-classification, keras, mobilenetv2, sustainability, tensorflow, transfer-learning, waste-management, waste-management-system, web-application
- Language: Jupyter Notebook
- Homepage: https://github.com/tanishq-ctrl/waste-classification
- Size: 6.38 MB
- Stars: 2
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 🗑️ Waste Management Classification System
[](https://www.python.org)
[](https://tensorflow.org/)
[](https://flask.palletsprojects.com/)
[](LICENSE)
---
A deep learning-based web application that classifies different types of waste materials using computer vision. The system helps in proper waste segregation by identifying whether an item belongs to categories like cardboard, glass, metal, paper, plastic, or trash.
🎯 Categories
Cardboard • Glass • Metal • Paper • Plastic • Trash
## ✨ Features
- 🚀 Real-time waste classification using deep learning
- 🌐 Web-based user interface for easy interaction
- 📸 Support for common image formats (PNG, JPG, JPEG)
- ⚡ Instant classification results with visual feedback
## 🛠️ Technology Stack
- **Backend**:
- 
- 
- **Deep Learning**:
- 
- 
- **Frontend**:
- 
- 
## 🧠 Model Architecture
The waste classification model is built using transfer learning with MobileNetV2 as the base model:
1. 🏗️ **Base Model**: Pre-trained MobileNetV2 on ImageNet
2. 🔄 **Fine-tuning**: Last 50 layers unfrozen for training
3. ➕ **Additional Layers**:
```
├── Global Average Pooling
├── Dense Layer (128 units, ReLU)
├── Dropout (0.6)
└── Output Layer (6 units, Softmax)
```
## 📊 Model Training
- 🖼️ **Input Image Size**: 128x128 pixels
- 📦 **Batch Size**: 32
- 🎯 **Training Strategy**:
```
├── Data augmentation (rotation, shift, shear, zoom, flip)
├── Learning rate scheduling with exponential decay
├── L2 regularization
└── Class weight balancing
```
- 📈 **Training Results**:
- Training Accuracy: 
- Validation Accuracy: 
## 📁 Project Structure
```
waste_management/
├── 🌐 app.py # Flask application
├── 🛠️ utils.py # Utility functions
├── 📓 WASTE_MANAGEMENT.ipynb # Model training notebook
├── 📂 static/
│ ├── 🎨 css/
│ │ └── style.css # Custom styling
│ └── 📤 uploads/ # Image upload directory
└── 📂 templates/
├── 🏠 index.html # Home page
└── 📊 result.html # Results page
```
### 📸 Dataset Overview
- **Total Images**: 
- **Image Format**: 
- **Resolution**: 
### 🗂️ Category Distribution
```
dataset-resized/
├── 📦 cardboard/ │ 403 images │ ████████░░░░░░░░░░ │ 15.7%
├── 🔍 glass/ │ 501 images │ ██████████░░░░░░░░ │ 19.5%
├── ⚙️ metal/ │ 410 images │ ████████░░░░░░░░░░ │ 15.9%
├── 📄 paper/ │ 594 images │ ████████████░░░░░░ │ 23.1%
├── 🏷️ plastic/ │ 482 images │ █████████░░░░░░░░░ │ 18.7%
└── 🗑️ trash/ │ 182 images │ ███░░░░░░░░░░░░░░░ │ 7.1%
```
### 💾 Getting the Dataset
🔄 **From Kaggle**:
- Visit [TrashNet Dataset](https://www.kaggle.com/datasets/feyzazkefe/trashnet/data)
- Click 'Download' button
- Extract the downloaded archive
## 🚀 Setup and Installation
1. Clone the repository:
```bash
git clone
cd waste_management
```
2. Install dependencies:
```bash
pip install -r requirements.txt
```
3. Run the application:
```bash
python app.py
```
4. Open your browser and navigate to `http://localhost:5000`
## 📱 Usage
1. 🌐 Access the web interface through your browser
2. 📤 Upload an image of the waste item you want to classify
3. ✨ Click submit to get the classification result
4. 📊 View the predicted category and confidence score
## 🔬 Model Training Process
The model was trained using transfer learning on MobileNetV2:
1. 📥 **Data Preparation**:
```
├── Dataset split: 80% training, 20% validation
├── Image resizing to 128x128 pixels
└── Data augmentation for better generalization
```
2. ⚙️ **Training Configuration**:
```
├── Optimizer: Adam with learning rate scheduling
├── Loss function: Categorical Cross-entropy
├── Metrics: Accuracy
└── Epochs: 50
```
3. 🎯 **Performance Optimization**:
```
├── Dropout for reducing overfitting
├── L2 regularization
└── Class weight balancing
```
### 🏷️ Topics
### 🏷️ Topics
## 🤝 Contributing
Feel free to submit issues, fork the repository, and create pull requests for any improvements.
## 📄 License
This project is licensed under the [MIT License](LICENSE)
---
Made with ❤️ for a cleaner 🌍