Face recognition using triplet loss, implementing FaceNet with pytorch. A small face dataset which consists of 62 IDs with 20 face images per ID for testing. The algorithm achieves accuracy above 97%.
人脸识别项目，提供一个小型数据集用作验证，使用三元组损失函数提升准确率和泛化能力，对FaceNet进行了一种实现。提供了一个小型的人脸数据集用于测试和验证。总共62个人，每人20张人脸图像。经测试，算法的准确率在97%以上，如果数据集更大一些，效果会更好。

断点模型链接

python3, pytorch, torchvision, cnn_finetune, opencv-python, numpy, tqdm. etc.

dataset.py

utils.py

main.py

Inspired by Google's FaceNet, we design a special loss to improve the model's descriminative power for fine-grained classification task like face recognition. The basic principle is quite plain and simple: if a group face images belong to one person identity, they are definitelt closer to each other than image faces belonging to other person's identity. Its main thought is similar to clusting algorithm.
受谷歌FaceNet的启发，我们设计了一种特殊的损失函数(类似于三元组损失函数)，用来提升细粒度分类任务的性能，比如人脸识别任务。它的基本**简单而朴素，如果一组人脸图像属于同一个人，那么这些图像相比其他人的人脸图像，他们之间肯定会更近似彼此。这个**类似于聚类。

We take advantage of a joint loss of classification and triplet loss to make the training convergence more easily and stable, the model is more able to discriminate between different ID's faces as illustrated above.
本算法在FaceNet的基础上改进了loss函数表达式，采用联合损失函数：CrossEntropy分类损失 + triplet损失，加快训练收敛并使得训练过程更稳定。