Skip to content

🌱 A smart waste detection app that classifies trash types (plastic, metal, glass, paper) using a machine learning model. Built with React, Express, and Flask. The frontend πŸ‘‡

Notifications You must be signed in to change notification settings

SamarMst/Waste-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

15 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸ—‘οΈ Waste Classifier App

A smart waste classification app that identifies trash types (plastic, metal, glass, etc.) using a machine learning model. Built with React (frontend), Express (backend), and Flask (ML API with TensorFlow).


🌟 Features

  • πŸ” Upload an image of waste and classify it instantly
  • 🧠 Powered by a trained MobileNetV2 TensorFlow model
  • ♻️ Detects categories like: plastic, metal, glass, and paper
  • πŸ’‘ Modern UI built with React and Tailwind CSS
  • πŸ”„ Full-stack communication using RESTful APIs

πŸ› οΈ Tech Stack

Layer Technology
Frontend React, Tailwind CSS
Backend Node.js, Express
ML API Flask, TensorFlow
ML Model MobileNetV2 (Transfer Learning)

πŸ§ͺ How it Works

  1. User uploads an image from the UI.
  2. Image is sent to Express backend, stored temporarily.
  3. Backend sends the image to the Flask API.
  4. Flask model predicts the class of waste.
  5. Response (class + confidence) is sent back to the frontend.

πŸš€ Installation

1. Clone the repo

git clone https://github.com/SamarMst/Waste-Classifier.git
cd Waste-Classifier

2. Frontend (React)

cd frontend
npm install
npm start

3. Backend (Express)

Copy
Edit
cd ../backend
npm install
npm start

4. Flask API (Python)

Copy
Edit
cd ../model
python -m venv venv
venv\Scripts\activate   # or source venv/bin/activate on Linux/Mac
pip install -r requirements.txt
python app.py

🧠 Model Info

Architecture: MobileNetV2

Input Size: 128x128x3

Trained on: TrashNet dataset: https://www.kaggle.com/datasets/feyyazkefe/trashnet/data

About

🌱 A smart waste detection app that classifies trash types (plastic, metal, glass, paper) using a machine learning model. Built with React, Express, and Flask. The frontend πŸ‘‡

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published