/facenet-pytorch-vggface2

A PyTorch implementation of Google's FaceNet [1] paper for training a facial recognition model with Triplet Loss and an implementation of the Shenzhen Institutes of Advanced Technology's 'Center Loss' [2] combined with Cross Entropy Loss using the VGGFace2 dataset. A pre-trained model using Triplet Loss is available for download.

Primary LanguagePython

No issues in this repository yet.