Project Drowsy is a computer vision project to predict driver drowsiness based on two key characteristics - eyes open/closed and yawning/not-yawning.
The model uses mediapipe for face/eye detection and a custom trained CNN model to predict eyes open/closed and yawning/not-yawning. The CNN model was trained using a kaggle dataset containing
- 726 images of eyes closed
- 726 images of eyes open
- 725 images of drivers yawning
- 723 images of drivers not yawning
Try it out F.R.I.D.A for yourself here!
This project was created as part of the Le Wagon Data Science Bootcamp (batch #753) by Patrick Rigg, Will Graham, Kai Majerus and Julien Festou.
The initial setup.
Create virtualenv and install the project:
sudo apt-get install virtualenv python-pip python-dev
deactivate; virtualenv ~/venv ; source ~/venv/bin/activate ;\
pip install pip -U; pip install -r requirements.txt
Unittest test:
make clean install test
Check for project_drowsy in gitlab.com/{group}. If your project is not set please add it:
- Create a new project on
gitlab.com/{group}/project_drowsy
- Then populate it:
## e.g. if group is "{group}" and project_name is "project_drowsy"
git remote add origin [email protected]:{group}/project_drowsy.git
git push -u origin master
git push -u origin --tags
Functionnal test with a script:
cd
mkdir tmp
cd tmp
project_drowsy-run
Go to https://github.com/{group}/project_drowsy
to see the project, manage issues,
setup you ssh public key, ...
Create a python3 virtualenv and activate it:
sudo apt-get install virtualenv python-pip python-dev
deactivate; virtualenv -ppython3 ~/venv ; source ~/venv/bin/activate
Clone the project and install it:
git clone [email protected]:{group}/project_drowsy.git
cd project_drowsy
pip install -r requirements.txt
make clean install test # install and test
Functionnal test with a script:
cd
mkdir tmp
cd tmp
project_drowsy-run