freefly518 / arcface-keras-flask

🎭 ArcFace Implementation using keras and Flask deploy

Face Recognition using ArcFace

ArcFace Implementation using keras and Flask deploy

Custom Model

I haven't use any pretrained model. So, full customized model with Arc loss. Anyone can easily customize their own model.

Folder/File Structure

+-Faces
|
+--- [Person1]
|     |
|     +--- image1.jpg
|     +--- image2.jpg
      ....
      ....
|
+--- [Person2]
|     |
|     +--- image1.jpg
|     +--- image2.jpg
      ....
      ....
...
...

Model Architecture

[Image]

Usage

• $ python main.py - Start the Flask app in the browser.
• $ python face_extract.py - Extract faces via webcam
• $ python num.py - Make two npy file from extracted faces [Ex. faces.npy, labels.npy]
• $ python train.py - Train the model
• $ python mtcnn_detect.py - Start face recognizer using webcam on OpenCV window

Notebook

Check ArcFace.ipynb for inference understanding.