/back-end

Food quality evaluation system using the OAK-D to assist food suppliers and consumers on appropriate food and waste management.

Primary LanguageJupyter NotebookMIT LicenseMIT

The African Eagles

Table of Contents

About the team

We are a team of 4 members:

Front End Developer Team Lead AI Engineer AI Engineer
Abderrahim SOUBAI-ELIDRISI Fatima-Ezzahra Sara EL-ATEIF Ikechukwu Nigel Ogbuchi

About our project "TRIA"

Purpose :
Leveraging Computer Vision Techniques for Image-based Food Quality Evaluation.

Solution :
Food quality evaluation system using the OAK-D to assist food suppliers and consumers on appropriate food and waste management.

How to run

Requirements

  • Python 3.8
  • OpenVino
  • Depthai library
  • OAK-D device from OpenCV
  • Python IDE (used PyCharm)

For all needed libraries please install from requirements.txt by typing in command line : pip install requirements.txt

Step 1: Train the ML model

Navigate to the model training folder and run one of the notebooks provided in Colab. This will generate a yaml and bin file that you need to turn into blob by using the provided OpenVino platform as described in the notebooks.

Step 2: Run project

Create a virtual environment for your project and install necessary libraries. Then place one of the py files available in oakd files folder according to the model you trained. Finally, link the OAK-D device to your computer and run the py file.

Step 3: Predict

This project was trained to predict rotten and fresh fruits, so you can use some real world fruits to keep track of their quality over time.

Step 4: Generate summary data

At the end of the day, you can run the post_processing.py file to generate summary data to keep track of how many fruits were rotten or fresh.

TRIA USER GUIDE

Watch the demo

TRIA Demo

License

See the LICENSE file for license rights and limitations (MIT).