GitHub repository: vishwas1234567/facenet-pytorch-vggface2
Description: 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.