1. 项目简介
本项目是一个基于pytorch实现的人脸检测平台,包含了了如下四个方面的功能:
- 人脸检测
- 人脸识别
- 人脸关键点检测
- 人脸属性识别
项目中包含了一个简易的UI和一个基于flask实现的后台。
1.1 人脸检测和识别
人脸检测就是使用矩形框去框出人脸在图片中的位置。该模块使用pytorch自带的MTCNN和facenet模型实现。
1.2 人脸关键点检测
该模块用于检测人脸区域的68个关键点,这些关键点的分布如下所示:
在该模块中 我们使用了300W-LP 数据集进行训练。训练代码可以从附录中找到。
1.3 人脸属性识别
在该模块中, 将使用模型去判断人脸的一些属性。在本项目我们仅仅检测男/女属性和张嘴/不张嘴 的属性。 该模型使用了celeba1 数据集进行训练。
2. 效果展示:
打开浏览器,打开摄像头,效果如下:
3. 项目地址
项目地址:face-detection-project
人脸关键点检测训练代码:face-key-point-pytorch
人脸属性检测训练代码:face-property