AI Trash Sorter

A basic waste classifier built in Python using Convolutional Neural Networks (CNN).

Table of Contents

1. About The Project
   • Built With
2. Getting Started
   • Prerequisites
   • Installation
3. Usage
4. Examples
5. Acknowledgments

About The Project

Basic Trash Sorter (Waste Classifier) created in Python using AI, convolutional neural networks, and machine learning to self-train and classify different types of waste following the Kaggle dataset: Garbage Classification.

It supports 12 waste categories for efficient classification.

(back to top)

Built With

• 
• 
• 

(back to top)

Getting Started

To get a local copy up and running, follow these simple steps.

Prerequisites

Ensure you have Python installed. Then, install the necessary dependencies:

• pip

  pip install -r requeriments.txt

Installation

1. Clone the repo

   git clone https://github.com/jerichd4c/IA-trash-sorter.git

2. Install the necessary packages

   pip install -r requeriments.txt

3. Ensure the model waste_classifier_model.h5 is in the root directory of the project.

(back to top)

Usage

Running the Application

To start the user interface with Streamlit:

streamlit run src/waste_app.py

Training or Recompiling the Model

If you want to retrain the model:

python src/full_model.py

Note: The process may take time depending on the hardware and dataset size.

(back to top)

Examples

Waste type: clothes Statistics:

Waste type: cardboard Statistics:

Waste type: battery Statistics:

(back to top)

Acknowledgments

• Kaggle Garbage Classification Dataset
• Streamlit Documentation
• TensorFlow Guide

(back to top)