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.
