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
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 |
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'
| Name | Github |
|---|---|
| Convert Torch model to PyTorch Model | OpenFacePytorch |
| MTCNN PyTorch | mtcnn-pytorch |
| Author | Leo（北习） |
|---|---|
| [email protected] |