README.zh-CN.md

Language: 🇨🇳 🇺🇸

«facenet»复现了论文FaceNet: A Unified Embedding for Face Recognition and Clustering

• 解析： FaceNet: A Unified Embedding for Face Recognition and Clustering

基于similarity实现，增加了如下功能：

1. 通过配置文件模块设置训练参数；
2. 支持多GPU训练以及混合精度训练；
3. 添加多种预处理函数以及训练配置。

关于FMNIST/CIFAR/LFW的训练结果，请参见详细

内容列表

• 内容列表
• 背景
• 安装
• 用法
• 主要维护人员
• 致谢
• 参与贡献方式
• 许可证

背景

facenet是一篇非常优秀的人脸识别论文，它创新性的提出了一种新的训练范式 - 三元组损失（triplet loss）训练。三元组损失的核心思想在于缩小同类人脸之间的欧式距离的同时尽可能的的扩大不同类之间的距离，每次训练都需要采集训练图像（锚点）以及同类正样本和不同类负样本。

similarity提供了一个非常棒的facenet训练工程。每轮训练开始之前先通过采样器对数据进行分配，保证每批次数据中包含P个类别以及每个类别拥有K个训练样本；完成前向计算后对同批次数据进行两两计算，采集符合semi-hard定义（负样本并没有比正样本接近于锚点，但是其距离仍位于边界范围内）的正负样本对参与损失函数计算。

上述工程的不足之处在于它并不支持多GPU训练、混合精度训练以及没有很好的模块化设计导致扩展性不强，不能够直接作用于实际应用。 为了更好的训练facenet，本仓库基于similarity进行了增强操作，提供了更友好的训练实现。

安装

$ pip install -r requirements.txt

用法

1. 获取数据文件。查看how to get data
2. 训练
   1. 单GPU训练

      $ CUDA_VISIBLE_DEVICES=0 python tools/train.py -cfg=configs/lfw/r18_lfw_224_e2_adam_g1.yaml
      

   2. 多GPU训练

      $ CUDA_VISIBLE_DEVICES=0,1,2,3 python tools/train.py -cfg=configs/lfw/r18_lfw_224_e2_adam_g4.yaml
      

主要维护人员

• zhujian - Initial work - zjykzj

致谢

• pytorch/vision

@article{2015,
   title={FaceNet: A unified embedding for face recognition and clustering},
   url={http://dx.doi.org/10.1109/CVPR.2015.7298682},
   DOI={10.1109/cvpr.2015.7298682},
   journal={2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
   publisher={IEEE},
   author={Schroff, Florian and Kalenichenko, Dmitry and Philbin, James},
   year={2015},
   month={Jun}
}

参与贡献方式

欢迎任何人的参与！打开issue或提交合并请求。

注意:

• GIT提交，请遵守Conventional Commits规范
• 语义版本化，请遵守Semantic Versioning 2.0.0规范
• README编写，请遵守standard-readme规范

许可证

Apache License 2.0 © 2021 zjykzj