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.
To set up the AI part, follow these steps:
-
Create and activate a virtual environment (optional but recommended):
python3 -m venv venv source venv/bin/activate # On Windows: venv\Scripts\activate
-
Install the required dependencies using pip:
pip install -r requirements.txt
-
Run the
predict.pyscript to classify waste:python predict.py path/to/your/image.jpg
-
Once you're done, deactivate the virtual environment:
deactivate
That's it! You have now successfully installed and used the waste detection system.