基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第1张图片

微信搜索“AI大道理”,选择“置顶”公众号

重磅干货,深入讲解AI大道理

——————

本设计研究人脸识别技术,基于卷积神经网络构建了一套人脸在线检测识别系统,系统将由以下几个部分构成:制作人脸数据集、CNN神经网络模型训练、人脸检测、人脸识别。经过实验,确定该系统可对本人的人脸进行快速并准确的检测与识别。

关键词: 神经网络; 图像处理; 人脸检测;人脸识别;TensorFlow;模型训练

一、设计目标

1.完成卷积神经网络人脸在线识别系统算法设计;

2.完成卷积神经网络人脸在线识别系统模型训练;

3.检测并且识别出人脸,人脸识别误差率<2%

二、制作人脸数据集

2.1、制作我的人脸数据集

人脸检测出人脸位置,返回坐标、尺寸把脸用数组切片的方法截取下来,把截取的小图片保存下来作为数据集。

本系统获取本人的人脸数据集10000张,使用的是dlib来识别人脸,虽然速度比OpenCV识别慢,但是识别效果更好。

其中,人脸大小:64*64。

2.2、主要步骤

(1)加载 dlib机器学习的开源库

(2)图像预处理 cvtColor(灰度化)。

(3)使用dlib自带的frontal_face_detector进行特征提取。

(4)使用特征提取器进行人脸检测。

(5)人脸保存。

2.3、制作结果

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第2张图片

71a94b494291a6f9bbc1b2382184b639.png

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第3张图片

2.5 、制作其他人脸数据集

本系统使用人脸数据集下载:http://vis-www.cs.umass.edu/lfw/lfw.tgz。

先将下载的图片集放在img_source目录下,用dlib来批量识别图片中的人脸部分,并保存到指定目录 faces_other。

人脸大小为64*64。

2.6、主要步骤

(1)读入人脸数据集源文件

(2)加载 dlib机器学习的开源库。

(3)用dlib来批量识别图片中的人脸部分。

(4)人脸保存。

2.7、制作结果

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第4张图片

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第5张图片

三、 神经网络模型训练

3.1、卷积神经网络内部分析

第一、二层(卷积层1、池化层1),

输入图片64*64*3,输出图片32*32*32

第三、四层(卷积层2、池化层2),

输入图片32*32*32,输出图片16*16*64;

第五、六层(卷积层3、池化层3),

输入图片16*16*64,输出图片8*8*64;

第七层(全连接层),

输入图片8*8*64,reshape到1*4096,输出1*512;

第八层(输出层),

输入1*512,输出1*2。

输出神经网络输出的1*2,与标签y_ [0,1]、[1,0]对比得出损失,损失函数为交叉熵,优化器采用Adan优化器,计算模型准确率。

cross_entropy= tf.reduce_mean

(tf.nn.softmax_cross_entropy_with_logits(logits=out, labels=y_))

optimizer =tf.train.AdamOptimizer

(learning_rate).minimize(cross_entropy)

accuracy =tf.reduce_mean

(tf.cast(tf.equal(tf.argmax(out,1),tf.argmax(y_,1)), tf.float32))

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第6张图片

3.2、输入层

人脸检测出人脸位置,返回坐标、尺寸把脸用数组切片的方法截取下来,把截取的小图片送入神经网络进行训练,输入的图片64*64*3。

3.3、卷积层

本网路共三层卷积层,卷积核大小为(3,3),卷积步长为[1,1,1,1],即一步,padding=“SAME”。卷积层提取特征,增加通道数,图片大小不变。

3.4、池化层

本文提出的池化层采用最大值采样,采样大小为2*2,即把输入的特征图分割成不重叠的2*2大小的矩形,对每个矩形取最大值,所以输出特征图的长和宽均是输入特征图的一半。

3.5、全连接层

为了增强网络的非线性能力,同时限制网络规模的大小,网络在特征提取层提取特征后,接入一个全连接层,该层的每一个神经元与前一层的所有神经元互相连接,将图片的卷积输出压扁成一个一维向量,输出1*512。

3.6、输出层

本系统共分两类,一类我的人脸(yes),另一类其他人脸(no),从全连接层输入1*512,输出层输出1*2。为求loss做准备。

3.7、主要步骤

(1)读入制作好的人脸数据集,将图片数据与标签转化成数组。

(2)划分测试集:训练集=20:1,做归一化处理。

(3)传入卷积神经网络。

(4)训练。

3.8、训练结果

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第7张图片

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第8张图片

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第9张图片

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第10张图片

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第11张图片

四、利用模型进行人脸识别

4.1 、主要步骤

(1)打开摄像头获取图片进行灰度化。

(2)人脸检测。

(3)导入训练好的神经网络模型。

(4)人脸识别。

4.2、识别结果

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第12张图片

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第13张图片

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第14张图片

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第15张图片

五、总结

本次设计主要研究了基于卷积神经网络的人脸在线识别系统。设计了基于机器视觉的人脸图像采集系统,自己制作了训练需要的人脸数据集。设计并实现了基于卷积神经网络的人脸识别系统,完成了基于卷积神经网络的人脸在线识别系统算法设计、神经网络模型训练,实现了对人脸的在线检测与识别。检测出本人人脸,则在人脸上显示”yes, my face”,若不是我的人脸,则在人脸上显示”no, other face ”。

六 、代码

见个人GitHub:

https://github.com/hhhvvvddd/CNN_faces_recognition

个人博客园:

https://www.cnblogs.com/fpzs/p/10602995.html

——————

浅谈则止,深入理解AI大道理

扫描下方“AI大道理”,选择“关注”公众号

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第16张图片

欢迎加入!

基于cnn的人脸识别_基于卷积神经网络(CNN)的人脸在线识别系统_第17张图片

你可能感兴趣的:(基于cnn的人脸识别)