This is an Android app that uses machine learning to classify images of waste into six categories: cardboard, glass, metal, paper, plastic, and trash. It was built using Flutter, a popular framework for building cross-platform mobile apps.
- The dataset used can be downloaded from here.
- Learn more about the ML model trained by accessing the kaggle notebook here.
- Take a photo or select an image from your gallery
- Classify the image using a machine learning model trained on waste images
- See the classification result (cardboard, glass, metal, paper, plastic, or trash)
- [TODO] Get more information about the waste category and how to properly dispose of it.
- Flutter 2.0+
- TensorFlow Lite for machine learning
- InceptionV2 model
To install the app, follow these steps:
- Clone the repository: git clone https://github.com/kshitij-ra/Waste-Classification-App.git
- Open the project in Android Studio or Visual Studio Code
- Connect your Android device or start an emulator
- Run the app
Contributions to this project are welcome! If you would like to contribute, please fork the repository and submit a pull request with your changes.
This project is licensed under the MIT License - see the LICENSE.md file for details.