本项目提供基于TensorFlow框架的人脸识别代码实现,涵盖模型训练、特征提取及面部检测等功能,适用于研究与开发场景。
该代码使用Tensorflow r1.7在Ubuntu 14.04系统上通过Python 2.7和Python 3.5进行了测试,并包含有测试用例。模型采用固定图像标准化处理,训练过程中采用了中科院自动化所的WebFace数据集,此面部检测后得到的数据集中共有453,453张图片及超过10,575个身份信息。在正式开始训练前对数据进行过滤可以带来性能上的提升,具体操作将在后续说明中提供。最佳模型已经在VGGFace2数据集上进行了训练,该数据集包含约3.3M张面部图像和约9000种类别。提供了几个预训练的模型,并且需要注意的是输入到模型中的图片需要使用固定图像标准化进行处理(例如,在运行validate_on_lfw.py时可以使用--use_fixed_image_standardization选项)。