face_gui
face_gui copied to clipboard
基于人脸检测的证件照自动生成(Automatic identification photos based on face detection)
face_gui
项目介绍
该项目能够从给定的正面照片中,自动识别脸部区域,并对图像前景进行提取分割,然后替换背景,并按照证件照的规格进行规范化处理。一共由5个模块组成:
1. 头部局部照识别与截取模块设计
利用Vahid Kazemi 和 Josephine Sullivan提出的基于gradient boosting的回归树算法检测面部的68个关键点位置. 论文:One Millisecond Face Alignment with an Ensemble of Regression Trees
2. 图像前景分割
利用opencv中的grabcut方法对人脸周围区域进行分割,将人头和衣服等前景分离出来. 论文:GrabCut” — Interactive Foreground Extraction using Iterated Graph Cuts
3. 证件照规范化
按照规格进行图片处理:分辨率:361×381,分辨率96dpi,位深度24,大小30k左右.
4. 背景替换
根据图片的背景颜色特征进行替换背景(蓝-红-白),如蓝背景变为红背景:将BGR图像转为HSV图像,蓝颜色H通道在78和110之间,然后转换通道将这些像素替换为(0,0,255)即可。
5. 界面设计
利用pyqt5进行界面设计.
项目配置
环境平台
python3
模块安装
pip install opencv-python
pip install pyqt5
pip install dlib
文件下载
shape_predictor_68_face_landmarks.dat.bz2下载后放入data目录中
脚本运行
python head.py