Face recognition with webrtc, websockets and opencv

This is a webrtc, websockets and opencv experiment developed during a athega hackday.

How does it work?

Frames are captured from the web camera via webrtc and sent to the server over websockets. On the server the frames are processed with opencv and a json response is sent back to the client.

Sample json response:

  {
    "face": {
      "distance": 428.53381034802453,
      "coords": {
        "y": "39",
        "x": "121",
        "height": "137",
        "width": "137"
      },
      "name": "mike"
    }
  }

Everything except distance is pretty self explanatory.

• name is the predicted name of the person in front of the camera.

• coords is where the face is found in the image.

• distance is a measurement on how accurate the prediction is, lower is better.

If we can't get a reliable prediction (10 consecutive frames that contains a face and with a distance lower than 1000) we switch over to training mode. In training mode we capture 10 images and send them together with a name back to the server for retraining. After the training has been completed we switch back to recognition mode and hopefully we'll get a more accurate result.

Running it

Make sure the dependencies are met.

• SQLite
• OpenCV with python bindings (I'm using the trunk version)
• Tornado
• PIL
• Peewee
• scikit-learn (for running the crossvalidation)

How to use

1. chmod +x ./db/create_db.sql
2. Create the database by issuing the following in the data folder cd data && sqlite3 images.db < ../db/create_db.sql
3. Download the AT&T face database and extract it to data/images before the server is started. This is needed to build the initial prediction model

cd data
wget http://www.cl.cam.ac.uk/Research/DTG/attarchive/pub/data/att_faces.tar.Z
tar zxvf att_faces.tar.Z
mv orl_faces images

4. Download haarcascde_frontalface_alt.xml in the data folder

cd data
wget https://raw.githubusercontent.com/opencv/opencv/master/data/haarcascades/haarcascade_frontalface_alt.xml

5. Check chmod +x *.sh
6. Run ./run-docker-compose.sh
7. (First time) Run ./install-python-libraries.sh
8. Run ./run-server.sh
9. Browse http://localhost:8888