/FaceRecognition

Webcam face recognition using tensorflow and opencv

Primary LanguagePythonMIT LicenseMIT

FaceRecognition

Webcam face recognition using tensorflow and opencv. The application tries to find faces in the webcam image and match them against images in an id folder using deep neural networks.

Dependencies

  • OpenCv
  • Tensorflow
  • Scikit-learn
  • easygui

Inspiration

Models, training code and inspriation can be found in the facenet repository. Multi-task Cascaded Convolutional Networks are used for facial and landmark detection while an Inception Resnet is used for ID classification. A direct link to the pretrained Inception Resnet model can be found here.

How to

Get the model from facenet and setup your id folder. The id folder should contain subfolders, each containing at least one image of one person. The subfolders should be named after the person in the folder since this name is used as output when a match is found.

E.g. id folder named ids containing subfolders Adam and Eve, each containing images of the respective person.

├── ids
│   ├── Adam
│   │   ├── Adam0.png
│   │   ├── Adam1.png
│   ├── Eve
│   │   ├── Eve0.png

Download and unpack the model to a folder and run python3 main.py ./folder/model.pb ./ids/ to start the program. Make sure to replace ./folder/model.pb with the path to the downloaded model.

Visualization hotkeys:

  • l - toggle facial landmarks
  • b - toggle bounding box
  • i - toggle id
  • f - toggle frames per second
  • s - save image face detections to id folder

alt text