superuichang / face-recognition_facenet_pytorch

Face-Recognition_FaceNet_PyTorch

This integrated pytorch based system is a tutorial system for those who are interseted in Computer Vision especially in face recognition. Face recognition method is using FaceNet.

Some parts of this system are copy from other Github. Sites are in the references below, appreciate their contribution.

FaceNet Models are from Openface

---

Environment

On my iMac, the version of necessary environments are:

Environment	Version
Python	>=3.4
PyTorch	0.3.0
Torchvision	0.2.0
Opencv	3.1.0
Dlib	19.9.0

---

Guide

Register.py is for people registion

Warning: Please remember to press 'ESC' to log out after a 10s record video has been recorded.

Recognizer.py is for people recogntion

Warning: If your device supports CPU only. Please remember to modify model loding in Line 53

Environment	Method
CPU	model.load_state_dict(torch.load('path',map_location=lambda storage, loc: storage))
GPU	model.load_state_dict(torch.load('path'))

Warning: Remember do not delete the folder './User/people_ori/Unknown'

---

References

Name	Github
Convert Torch model to PyTorch Model	OpenFacePytorch
MTCNN PyTorch	mtcnn-pytorch

---

Author	Leo（北习）
E-mail	[email protected]