tbmoon / facenet

FaceNet for face recognition using pytorch

Facenet for face verification using pytorch

• Pytorch implementation of the paper: "FaceNet: A Unified Embedding for Face Recognition and Clustering".
• Training of network is done using triplet loss.
• This work is modified in some functionality from the original work by Taebong Moon and then retrained for the purpose of completing my BS degree. The full report can be found at this folder: full-and-paper-report
• To use the pretrained model please refer to this repo: https://github.com/khrlimam/res-facenet
• If you wish to try the demo app please clone this repo and follow the installation instruction: https://github.com/khrlimam/demo-facenet

How to train/validate model

• Download vggface2 (for training) and lfw (for validation) datasets.

  • https://www.robots.ox.ac.uk/%7Evgg/data/vgg_face2/
  • http://vis-www.cs.umass.edu/lfw/lfw.tgz

• Align face image files by following David Sandberg's instruction (part of "Face alignment").

  • https://github.com/davidsandberg/facenet/wiki/Classifier-training-of-inception-resnet-v1

• Write list file of face images by running "datasets/write_csv_for_making_dataset.py"

  • python write_csv_for_making_dataset.py --root-dir=/path/to/dataset/dir --final-file=dataset.csv
  • datasets/write_csv_for_making_dataset.py is multiprocess version of previous .ipynb. This way generating csv dataset is much faster.

• Train

  • Again, one need to modify paths in accordance with location of image dataset.
  • Also feel free to change some parameters.

Results

• Accuracy on VGGFace2 and LFW datasets

• Triplet loss on VGGFace2 and LFW datasets

• ROC curve on LFW datasets for validation

• True counts on each threshold

• Test the model on 30 pair of images with threshold 1.5

References

• https://github.com/liorshk/facenet_pytorch
• https://github.com/davidsandberg/facenet
• https://arxiv.org/abs/1503.03832
• https://arxiv.org/abs/1703.07737