README.md

AI project

This directory contains scripts and files related to the AI part of the ESP project. The main script, predict.py, utilizes a pre-trained deep learning model to classify waste into various categories such as cardboard, glass, metal, paper, plastic, and trash. The model is provided in the file trash_detection_model.keras and the dataset here.

Installation

To set up the AI part, follow these steps:

1. Create and activate a virtual environment (optional but recommended):

   python3 -m venv venv
   source venv/bin/activate  # On Windows: venv\Scripts\activate

2. Install the required dependencies using pip:

   pip install -r requirements.txt

3. Run the predict.py script to classify waste:

   python predict.py path/to/your/image.jpg

4. Once you're done, deactivate the virtual environment:

   deactivate

That's it! You have now successfully installed and used the waste detection system.