目录
一、简介
二、face SDK
1、环境需要
2、功能实现
2.1人脸库搭建
2.2人脸识别
3、测试结果
4、总结
本文根据京东开源的人脸识别工具包FaceX-Zoo,实现了人脸识别功能。对代码中的faceSDK部分做了修改,并搭建了自己的人脸库,识别结果准确度较高,后期可以打包SDK放到移动端使用。
项目代码地址放在文末...
一、简介
FaceX-Zoo是一个专为人脸识别而生的开源库,是目前准确率最高的开源人脸识别项目。FaceX-Zoo是基于PyTorch实现的。它提供了一个训练模块,其中有各种supervisory head和骨干网络,以实现最先进的人脸识别; 同时也提供了一个标准化的评估模块,该模块只需编辑一个简单的配置就可以评估大多数流行基准中的模型。此外,还提供了一个简单的功能齐全的face SDK,里面包含了大量的预训练模型,用于验证和初步应用训练过的模型。除了人脸识别,还针对实际应用开发了特定功能(如人脸戴口罩等)。
仓库地址:GitHub - JDAI-CV/FaceX-Zoo: A PyTorch Toolbox for Face Recognition
FaceX-zoo实现人脸识别包含三个步骤,第一步人脸检测,第二步人脸对齐,第三步人脸检测。
- 人脸检测(face detection):使用的目前准确度最好的retina网络,经过实测比常见的yolo,ssd和MTCNN等准确很多
- 人脸对齐(face alignment):人脸对齐主要是人脸关键点的检测,使用的是人脸关键点的检测模型是PFLD