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).
- π 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
| Layer | Technology |
|---|---|
| Frontend | React, Tailwind CSS |
| Backend | Node.js, Express |
| ML API | Flask, TensorFlow |
| ML Model | MobileNetV2 (Transfer Learning) |
- User uploads an image from the UI.
- Image is sent to Express backend, stored temporarily.
- Backend sends the image to the Flask API.
- Flask model predicts the class of waste.
- Response (class + confidence) is sent back to the frontend.
git clone https://github.com/SamarMst/Waste-Classifier.git
cd Waste-Classifiercd frontend
npm install
npm startCopy
Edit
cd ../backend
npm install
npm startCopy
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.pyArchitecture: MobileNetV2
Input Size: 128x128x3
Trained on: TrashNet dataset: https://www.kaggle.com/datasets/feyyazkefe/trashnet/data