NovatecConsulting / FaceRecognition-in-ARKit

Detects faces using the Vision-API and runs the extracted face through a CoreML-model to identiy the specific persons.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FaceRecognition in ARKit

This is a simple showcase project, that detects faces using the Vision-API and runs the extracted face through a CoreML-model to identiy the specific persons.

image of scene with face recognition

Requirements

  • Xcode 9
  • iPhone 6s or newer
  • Machine-Learning model

Machine-Learning model

To create your own machine-learning model, you can read our blog post "How we created our Face-Recognition model"

The short version is:

  • We trained a model in the AWS using Nvidia DIGITS
  • Took a couple of hundred pictures of each person, and extracted the faces
  • Also added an "unknown" category with differnent faces.
  • Used a pretrained model fine-tuned for face-recognition.

Acknowledgements

About

Detects faces using the Vision-API and runs the extracted face through a CoreML-model to identiy the specific persons.

License:MIT License


Languages

Language:Swift 98.1%Language:Ruby 1.9%