caomy7 / facenet-pytorch-vggface2 Goto Github PK
View Code? Open in Web Editor NEWThis project forked from tamerthamoqa/facenet-pytorch-glint360k
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.