Trash Classification Model
This repository contains code for training a Convolutional Neural Network (CNN) model to classify images of trash. The model is trained on the TrashNet dataset, which is available through the Hugging Face datasets library.
Project Overview
The goal of this project is to build a machine learning model that can classify images of trash into categories such as plastic, metal, paper, etc. This model could potentially be used in applications to aid in waste management and recycling efforts.
Dataset
The dataset used for this project is TrashNet, a labeled collection of images of different types of trash. I load this dataset using the Hugging Face Datasets library. The dataset includes the following classes:
- Plastic
- Metal
- Paper
- Cardboard
- Glass
- Trash (other miscellaneous items)
Prerequisites
- Python 3.7 or higher
- Git
- Hugging Face Datasets, TensorFlow, and other dependencies listed in
requirements.txt
Github Repository
For the detailed description about the training process and models' performace, you can visit this Github Repository
Usage
- Clone the repository
git clone https://github.com/ardrkha/trash-classification.git
cd trash-classification
- Run the notebook
trash-classification/
├── notebook/
│ ├── [Notebook.ipynb] (run this notebook for the complete implementation)
├── README.md
└── ...
License
This project is licensed under the MIT License - see the LICENSE file for details.
Contact
If you have any questions, feel free to reach out to us at [email protected].
Developed by Hardatama Rakha Ugraha - 2024