Skip to content

Repository files navigation

♻️ AI-Powered Smart Waste Segregation System

An AI-powered web application that automatically detects and classifies waste materials using YOLO object detection, helping users dispose of waste responsibly and promote sustainable waste management.

Python React Flask YOLO SQLite License


📖 Overview

The AI-Powered Smart Waste Segregation System is a full-stack web application that uses YOLO object detection to identify waste materials from uploaded images or captured photos. The system classifies waste into different categories and provides disposal recommendations, making waste management more efficient and environmentally friendly.

The application also includes secure user authentication, report generation, email notifications, analytics, and an admin dashboard for monitoring system usage.


✨ Features

  • 🔍 AI-powered waste detection using YOLO
  • 📷 Upload image or capture using camera
  • ♻️ Waste classification into multiple categories
  • 💡 Disposal recommendations
  • 👤 User Registration & Login (JWT Authentication)
  • 🔒 Forgot Password with OTP
  • 📧 Email notifications using Gmail SMTP
  • 📄 PDF Detection Reports
  • 📊 Admin Dashboard & Analytics
  • 📈 Detection History
  • 🌙 Dark / Light Theme
  • 📱 Fully Responsive UI

🧠 Waste Categories

  • Biodegradable
  • Cardboard
  • Glass
  • Metal
  • Paper
  • Plastic

🏗 Tech Stack

Frontend

  • React
  • Vite
  • Tailwind CSS
  • Axios
  • React Router
  • Lucide Icons

Backend

  • Flask
  • Python
  • SQLite
  • JWT Authentication
  • SMTP Email Integration

AI / Machine Learning

  • Ultralytics YOLO
  • OpenCV
  • NumPy

📂 Project Structure

AI-Powered-Smart-Waste-Segregation-System/
│
├── frontend/                 # React Frontend
│   ├── src/
│   ├── public/
│   └── package.json
│
├── services/                 # Backend Services
├── templates/
├── app.py
├── config.py
├── predict.py
├── train.py
├── evaluate.py
├── requirements.txt
├── .gitignore
└── README.md

⚙️ Installation

Clone Repository

git clone https://github.com/ashishkumar817/AI-Powered-Smart-Waste-Segregation-System.git

cd AI-Powered-Smart-Waste-Segregation-System

Backend Setup

Create Virtual Environment

python -m venv venv

Activate

Windows

venv\Scripts\activate

Install Dependencies

pip install -r requirements.txt

Run Flask Server

python app.py

Frontend Setup

cd frontend
npm install
npm run dev

🤖 Model

This project uses Ultralytics YOLO for object detection.

The model detects waste objects and classifies them into six categories.

Note: The trained model weights are not included in this repository if they exceed GitHub's file size limits.


🔮 Future Improvements

  • Live Camera Detection
  • Video Detection
  • Mobile Application
  • Cloud Deployment
  • Multi-language Support
  • Smart Bin Integration
  • Real-time Analytics
  • More Waste Categories

🤝 Contributing

Contributions are welcome.

  1. Fork the repository
  2. Create your feature branch
git checkout -b feature-name
  1. Commit your changes
git commit -m "Add new feature"
  1. Push
git push origin feature-name
  1. Open a Pull Request

📄 License

This project is licensed under the MIT License.


👨‍💻 Author

Ashish Kumar


⭐ Support

If you found this project useful, consider giving it a ⭐ on GitHub.

It helps others discover the project and supports future development.

About

AI-powered waste segregation system using YOLO8, React, Flask, and SQLite for intelligent waste classification and sustainable disposal.

Topics

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages