This repository contains tools to setup and train the facial recognition module for the MagicMirror.
The scripts in this directory are based on scripts from pi-facerec-box and should be used on a computer (with a webcam). You need OpenCV installed on your computer. It works on a RaspberryPi.
- Make sure you have all dependencies (see below) installed.
- Run
python capture.py
. - Decide whether you want to capture images from your web cam or convert existing
.jpg
images. - Enter the name of the person you are about to capture. Images will be stored in a folder named after the captured person in
training_data/
. - Follow screen instructions.
- Make sure you have all dependencies (see below) installed.
- Make sure you have captured all your images.
- Run
python train.py
. The script will automatically scan the directorytraining_data/
for your images. - Wait. You will end up with a
training.xml
file in the current directory. - Copy down the
['name', 'name2','name3']
part because you will later need it for setting up your mirror's face recognition and to test your face recognition model.
With this tool you can test if your facerecognition model is working.
- Make sure you have all dependencies (see below) installed.
- Make sure your
training.xml
from runningtrain.py
is in this directory - specify the face recognition algorithm in the environment with
export FACE_ALGORITHM=1
- specify your user labels in the environment with
export FACE_USERS=Alice,Bob,Casey,Doug
- Run
python facerecognition.py
.
To install OpenCV run:
sudo apt-get install libopencv-dev python-opencv
If you are using virtual environments you will need to need to copy the opencv python modules into your virutal environment path. That will look something like this:
cp /usr/lib/python2.7/dist-packages/cv* ~/.virtualenvs/MY_VIRTUAL_ENV/lib/python2.7/site-packages/
Where python2.7
will be the name of your python version where
opencv was installed and MY_VIRTUAL_ENV
is the name of your
virtual environment.
Install the required python packages.
pip install -r requirements.txt
Currently this is just the future
module for making the scripts python 2 and 3 cross compatible.
###pi-facerec-box The MIT License (MIT)
Copyright (c) 2018 Antoine Dombrecht