devbox
我家小花儿
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

[深度学习入门案例2]基于卷积神经网络与Keras构建人脸识别模型_用深度学习模型进行人脸识别训练,调整(修改)模型,测试,得到结果;跟至少另外2种以

作者：我家小花儿 | 2024-06-10 07:30:05

赞

踩

用深度学习模型进行人脸识别训练,调整(修改)模型,测试,得到结果;跟至少另外2种以

文章目录

一、工具与环境

二、深度学习环境的搭建

三、基于卷积神经网络人脸识别模型的构建与测试

1.核心代码

第一步：采集自己和他人的人脸特征数据，分别对应数据标签0和1

 第二步：训练识别人脸特征的模型，并将模型保存为.h5格式的文件

 第三步：读取.h5文件格式的人脸识别模型，对摄像头录入的人脸图像进行识别

2.识别的基本原理

3.运行与测试

第一步：采集本人与他人的人脸特征数据（注意需要打开电脑摄像头，否则会报错！）

第二步：进行10轮次的人脸数据模型的训练

第三步：人脸识别与比对测试

四、分析与小结

参考文章

一、工具与环境

  • Pycharm 2022.1.4
  • conda version : 4.5.4
  • python version : 3.6.5.final.0
  • platform : win-64

二、深度学习环境的搭建

详情见我写的这篇文章的第二部分，这里不再赘述

[深度学习入门案例1]基于Keras的手写数字图像识别icon-default.png?t=N4P3https://blog.csdn.net/qq_52487066/article/details/131048466?spm=1001.2014.3001.5501环境搭建好之后，还需要额外下载下面的工具包，依次在Anaconda Prompt 中执行以下命令

  1. conda install py-opencv -i https://pypi.tuna.tsinghua.edu.cn/simple
  2. pip install dlib==19.7.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
  3. pip install face_recognition -i https://pypi.tuna.tsinghua.edu.cn/simple
  4. pip install Pillow -i https://pypi.tuna.tsinghua.edu.cn/simple

工具包下载导入完成，下面进行项目的构建与运行

三、基于卷积神经网络人脸识别模型的构建与测试

1.核心代码

第一步：采集自己和他人的人脸特征数据，分别对应数据标签0和1

get_my_face.py 

  1. import cv2
  2. import dlib
  3. import os
  4. import random
  5.  
  6. output_dir = './my_crop_faces'
  7. size = 160
  8. if not os.path.exists(output_dir):
  9.     os.makedirs(output_dir)
  10.  
  11. # 改变图片的亮度与对比度
  12. def relight(img, light=1, bias=0):
  13.     w = img.shape[1]
  14.     h = img.shape[0]
  15.     #image = []
  16.     for i in range(0,w):
  17.         for j in range(0,h):
  18.             for c in range(3):
  19.                 tmp = int(img[j,i,c]*light + bias)
  20.                 if tmp > 255:
  21.                     tmp = 255
  22.                 elif tmp < 0:
  23.                     tmp = 0
  24.                 img[j,i,c] = tmp
  25.     return img
  26.  
  27. #使用dlib自带的frontal_face_detector作为我们的特征提取器
  28. detector = dlib.get_frontal_face_detector()
  29. # 打开摄像头 参数为输入流，可以为摄像头或视频文件
  30. camera = cv2.VideoCapture(0)
  31.  
  32. index = 0
  33. while True:
  34.     if (index <= 1000):
  35.         print('Being processed picture %s' % index)
  36.         # 从摄像头读取照片
  37.         success, img = camera.read()
  38.         # 转为灰度图片
  39.         gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
  40.         # 使用detector进行人脸检测
  41.         dets = detector(gray_img, 1)
  42.  
  43.         for i, d in enumerate(dets):
  44.             x1 = d.top() if d.top() > 0 else 0
  45.             y1 = d.bottom() if d.bottom() > 0 else 0
  46.             x2 = d.left() if d.left() > 0 else 0
  47.             y2 = d.right() if d.right() > 0 else 0
  48.  
  49.             face = img[x1:y1,x2:y2]
  50.             # 调整图片的对比度与亮度， 对比度与亮度值都取随机数，这样能增加样本的多样性
  51.             face = relight(face, random.uniform(0.5, 1.5), random.randint(-50, 50))
  52.             face = cv2.resize(face, (size,size))
  53.             cv2.imshow('image', face)
  54.             cv2.imwrite(output_dir+'/'+str(index)+'.jpg', face)
  55.             index += 1
  56.         key = cv2.waitKey(30) & 0xff
  57.         if key == 27:
  58.             break
  59.     else:
  60.         print('Finished!')
  61.         break

 第二步：训练识别人脸特征的模型，并将模型保存为.h5格式的文件

model_train.py

  1. #-*- coding: utf-8 -*-
  2. from keras.models import Sequential
  3. from keras.layers import Dense, Dropout, Activation, Flatten
  4. from keras.layers import Convolution2D, MaxPooling2D
  5. from keras.optimizers import SGD
  6. from keras.utils import np_utils
  7. import os
  8. import cv2
  9. import numpy as np
  10. from sklearn.model_selection import train_test_split
  11. import random
  12.  
  13. def get_files(input_dir):
  14.     file_list = []
  15.     for (path, dirnames, filenames) in os.walk(input_dir):
  16.         # print(path) #输出对应顶层文件夹
  17.         # print(dirnames)#在当前文件夹下的文件夹
  18.         # print(filenames)#在当前文件夹下的文件夹
  19.         for filename in filenames:
  20.             if filename.endswith('.jpg') or filename.endswith('.bmp'):
  21.                 # print(filename)
  22.                 full_path = os.path.join(path, filename)
  23.                 # print(full_path)
  24.                 file_list.append(full_path)
  25.     return file_list
  26.  
  27. #设置hujianhua文件夹的对应标签为0
  28. def getPaddingSize(img):
  29.     h, w, _ = img.shape
  30.     top, bottom, left, right = (0,0,0,0)
  31.     longest = max(h, w)
  32.  
  33.     if w < longest:
  34.         tmp = longest - w
  35.         # //表示整除符号
  36.         left = tmp // 2
  37.         right = tmp - left
  38.     elif h < longest:
  39.         tmp = longest - h
  40.         top = tmp // 2
  41.         bottom = tmp - top
  42.     else:
  43.         pass
  44.     return top, bottom, left, right
  45.  
  46. def read_img_label(file_list, label):
  47.     size = 64
  48.     imgs = []
  49.     labs = []
  50.     #01
  51.     num = 0
  52.     for filename in file_list:
  53.         # print(filename)
  54.         img = cv2.imread(filename)
  55.         # print(img.shape)
  56.         top, bottom, left, right = getPaddingSize(img)
  57.         # 将图片放大， 扩充图片边缘部分
  58.         img = cv2.copyMakeBorder(img, top, bottom, left, right, cv2.BORDER_CONSTANT, value=[0, 0, 0])
  59.         img = cv2.resize(img, (size, size))
  60.         imgs.append(img)
  61.         labs.append(label)
  62.         num = num + 1
  63.  
  64.     # print(len(imgs))
  65.     # print(len(labs))
  66.     return imgs, labs
  67.  
  68.  
  69. def read_dataset():
  70.     input_dir = "./data_collection/my_crop_faces"
  71.  
  72.  
  73.     all_imgs_list = []
  74.     all_label_list = []
  75.     my_file_list = get_files(input_dir)
  76.     # 0->[0,1] 1->[1,0]
  77.     label = 0 #[0, 1]
  78.     my_imgs_list, my_labs_list = read_img_label(my_file_list, label)
  79.  
  80.     input_dir = "./data_collection/others_img_crop"
  81.  
  82.     others_file_list = get_files(input_dir)
  83.     label = 1 #[1, 0] #->0
  84.     others_imgs_list, others_labs_list = read_img_label(others_file_list, label)
  85.  
  86.     for img in my_imgs_list:
  87.         all_imgs_list.append(img)
  88.     for img in others_imgs_list:
  89.         all_imgs_list.append(img)
  90.  
  91.     for label in my_labs_list:
  92.         all_label_list.append(label)
  93.     for label in others_labs_list:
  94.         all_label_list.append(label)
  95.  
  96.     imgs_array = np.array(all_imgs_list)
  97.     # print(imgs_array.shape)
  98.  
  99.     labs_array = np.array(all_label_list)
  100.     # print(labs_array.shape)
  101.  
  102.     return imgs_array,labs_array
  103.  
  104.  
  105. #加载数据集并按照交叉验证的原则划分数据集并进行相关预处理工作
  106. def load_data(img_rows = 64, img_cols = 64,
  107.          img_channels = 3, nb_classes = 2):
  108.     #加载数据集到内存
  109.     images, labels = read_dataset()
  110.     print(images.shape)
  111.     print(labels.shape)
  112.  
  113.     train_images, valid_images, train_labels, valid_labels = train_test_split(images, labels, test_size = 0.3, random_state = random.randint(0, 100))
  114.     _, test_images, _, test_labels = train_test_split(images, labels, test_size = 0.5, random_state = random.randint(0, 100))
  115.  
  116.     train_images = train_images.reshape(train_images.shape[0], img_rows, img_cols, img_channels)
  117.     valid_images = valid_images.reshape(valid_images.shape[0], img_rows, img_cols, img_channels)
  118.     test_images = test_images.reshape(test_images.shape[0], img_rows, img_cols, img_channels)
  119.     input_shape = (img_rows, img_cols, img_channels)
  120.  
  121.     #输出训练集、验证集、测试集的数量
  122.     print(train_images.shape[0], 'train samples')
  123.     print(valid_images.shape[0], 'valid samples')
  124.     print(test_images.shape[0], 'test samples')
  125.  
  126.     #我们的模型使用categorical_crossentropy作为损失函数，因此需要根据类别数量nb_classes将
  127.     #类别标签进行one-hot编码使其向量化，在这里我们的类别只有两种，经过转化后标签数据变为二维
  128.     train_labels = np_utils.to_categorical(train_labels, nb_classes)
  129.     valid_labels = np_utils.to_categorical(valid_labels, nb_classes)
  130.     test_labels = np_utils.to_categorical(test_labels, nb_classes)
  131.     print(train_labels.shape)
  132.     print(valid_labels.shape)
  133.     print(test_labels.shape)
  134.     #像素数据浮点化以便归一化
  135.     train_images = train_images.astype('float32')
  136.     valid_images = valid_images.astype('float32')
  137.     test_images = test_images.astype('float32')
  138.  
  139.     #将其归一化,图像的各像素值归一化到0~1区间
  140.     train_images /= 255
  141.     valid_images /= 255
  142.     test_images /= 255
  143.  
  144.     return train_images, train_labels, valid_images, valid_labels, test_images, test_labels
  145.  
  146. #建立模型
  147. def build_model(nb_classes = 2):
  148.     #构建一个空的网络模型，它是一个线性堆叠模型，各神经网络层会被顺序添加，专业名称为序贯模型或线性堆叠模型
  149.     model = Sequential()
  150.  
  151.     #以下代码将顺序添加CNN网络需要的各层，一个add就是一个网络层
  152.     model.add(Convolution2D(32, 3, 3, border_mode='same',
  153.                                  input_shape =  (64, 64, 3)))    #1 2维卷积层
  154.     model.add(Activation('relu'))                                  #2 激活函数层
  155.  
  156.     model.add(Convolution2D(32, 3, 3))                             #3 2维卷积层
  157.     model.add(Activation('relu'))                                  #4 激活函数层
  158.  
  159.     model.add(MaxPooling2D(pool_size=(2, 2)))                      #5 池化层
  160.     model.add(Dropout(0.25))                                       #6 Dropout层
  161.  
  162.     model.add(Convolution2D(64, 3, 3, border_mode='same'))         #7  2维卷积层
  163.     model.add(Activation('relu'))                                  #8  激活函数层
  164.  
  165.     model.add(Convolution2D(64, 3, 3))                             #9  2维卷积层
  166.     model.add(Activation('relu'))                                  #10 激活函数层
  167.  
  168.     model.add(MaxPooling2D(pool_size=(2, 2)))                      #11 池化层
  169.     model.add(Dropout(0.25))                                       #12 Dropout层
  170.  
  171.     model.add(Flatten())                                           #13 Flatten层
  172.     model.add(Dense(512))                                          #14 Dense层,又被称作全连接层
  173.     model.add(Activation('relu'))                                  #15 激活函数层
  174.     model.add(Dropout(0.5))                                        #16 Dropout层
  175.     model.add(Dense(nb_classes))                                   #17 Dense层
  176.     model.add(Activation('softmax'))                               #18 分类层，输出最终结果
  177.     #输出模型概况
  178.     print(model.summary())
  179.     return model
  180.  
  181. model = build_model()
  182.  
  183. sgd = SGD(lr=0.01, decay=1e-6,
  184.           momentum=0.9, nesterov=True)  # 采用SGD+momentum的优化器进行训练，首先生成一个优化器对象
  185. model.compile(loss='categorical_crossentropy',
  186.                    optimizer=sgd,
  187.                    metrics=['accuracy'])  # 完成实际的模型配置工作
  188.  
  189. train_images, train_labels, valid_images, valid_labels, test_images, test_labels = load_data()
  190.  
  191. batch_size = 20
  192. nb_epoch = 10
  193. train_history = model.fit(train_images,
  194.                train_labels,
  195.                batch_size=batch_size,
  196.                nb_epoch=nb_epoch,
  197.                validation_data=(valid_images, valid_labels),
  198.                shuffle=True)
  199.  
  200. scores = model.evaluate(test_images, test_labels)
  201. print('accuracy=', scores[1])
  202. prediction = model.predict_classes(test_images)
  203. # print(prediction)
  204. model.save('./me.face.model.h5')

 第三步：读取.h5文件格式的人脸识别模型，对摄像头录入的人脸图像进行识别

video_predict.py

  1. import cv2
  2. import dlib
  3. from keras.models import load_model
  4. import sys
  5.  
  6. size = 64
  7.  
  8. # 使用dlib自带的frontal_face_detector作为我们的特征提取器
  9. detector = dlib.get_frontal_face_detector()
  10.  
  11. cam = cv2.VideoCapture(0)
  12.  
  13. model = load_model('./me.face.model.h5')
  14.  
  15. while True:
  16.     _, img = cam.read()
  17.     gray_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
  18.     dets = detector(gray_image, 1)
  19.     for i, d in enumerate(dets):
  20.         x1 = d.top() if d.top() > 0 else 0
  21.         y1 = d.bottom() if d.bottom() > 0 else 0
  22.         x2 = d.left() if d.left() > 0 else 0
  23.         y2 = d.right() if d.right() > 0 else 0
  24.         face = img[x1:y1, x2:y2]
  25.         # 调整图片的尺寸
  26.         face = cv2.resize(face, (size, size))
  27.         shape_img = (face.reshape(1, size, size, 3)).astype('float32') / 255
  28.  
  29.         prediction = model.predict_classes(shape_img)
  30.         print(prediction[0])
  31.         name = "unknown"
  32.         if prediction[0] == 0:
  33.             print("识别出本人")
  34.             name = "Aricl."
  35.         else:
  36.             print("不是本人")
  37.             name = "unknown"
  38.         cv2.rectangle(img, (x2, x1), (y2, y1), (255, 0, 0), 3)
  39.         font = cv2.FONT_HERSHEY_SIMPLEX
  40.         cv2.putText(img, name, (x2, x1), font, 0.8, (255, 255, 255), 1)
  41.  
  42.     cv2.imshow('image', img)
  43.     key = cv2.waitKey(30) & 0xff
  44.     if key == 27:
  45.         sys.exit(0)

2.识别的基本原理

        核心之一是使用OpenCV（Open source Computer Vision Library），即开放源代码计算机视觉库进行图像的处理。

        它是一套关于计算机视觉的开放源代码的API函数库。这也就意味着，(1)不管是科学研究，还是商业应用，都可以利用它来作开发;(2)所有API函数的源代码都是公开的，你可以看到其内部实现的程序步骤；(3)你可以修改OpenCV的源代码，编译生成你需要的特定API函数。但是，作为一个库，它所提供的，仅仅是一些常用、经典、大众化算法的API。

        一个典型的计算机视觉算法，一般包含以下步骤：

  1. 数据获取
  2. 预处理
  3. 特征提取
  4. 特征选择
  5. 分类器设计与训练
  6. 分类判别

而OpenCV对这六个部分，都分别提供了API以供开发者调用。

        核心之二是使用Keras搭建卷积神经网络。

        Keras对人工智能来说，是一款比较好的入门框架。它是一个高级的Python神经网络框架，已经被添加到TensorFlow中，成为其默认的框架，为TensorFlow提供更高级的API。

        如果将TensorFlow比喻为编程界的Java或者C++，那么Keras就是编程界的Python，它作为TensorFlow的高层封装，可以与TensorFlow联合使用，用它可以快速搭建模型。

        并且Keras是TensorFlow官方支持的。当机器上有可用的GPU时，代码会自动调用GPU进行并行计算，功能十分强大！

3.运行与测试

第一步：采集本人与他人的人脸特征数据（注意需要打开电脑摄像头，否则会报错！）

 

第二步：进行10轮次的人脸数据模型的训练

第三步：人脸识别与比对测试

上图是本人（博主Aricl.），下图是我室友（未知人脸），可见人脸识别的准确率还是不错的！

四、分析与小结

        本次通过构建基于卷积神经网络与Keras的人脸识别模型，实现的基本原理是：引用Python中的OpenCV库，调用其中的人脸分类器，然后调用电脑的摄像头循环读取单帧数据，进行实时人脸检测，并框出人脸，标注信息。

        首先我们需要进行人脸数据的采集，总共采集了1000张本人与他人的人脸特征数据，采集完毕后，进行数据模型的训练，共训练了10轮次，每轮次六千多，将训练好的模型保存为.h5格式的文件。然后调用摄像头读取每一帧的人脸照片，对图片进行灰度化处理，检测出人脸并框出对应的区域，读取.h5人脸识别模型文件进行人脸识别与比对，并在上方标注出是本人或者未知等信息。

        另外，采集时对被采集人和环境有一定要求，光线应该明亮使得摄像头能看清楚人脸，并且人脸应该正对着摄像头，这样采集得到的人脸特征数据才较为准确，识别的效果就越好。

        本次仅仅是人脸识别的一次初步入门，我对此也产生了浓厚的学习兴趣，但博主正在备战考研，只能初步了解，后续有机会再继续学习与研究该领域相关技术，加油！！

参考文章

Python+Keras+opencv实现人脸识别icon-default.png?t=N4P3https://blog.csdn.net/gf19960103/article/details/91038858?ops_request_misc=&request_id=&biz_id=102&utm_term=%E4%BD%BF%E7%94%A8Keras%E6%9E%84%E5%BB%BA%E4%BA%BA%E8%84%B8%E8%AF%86%E5%88%AB%E6%A8%A1%E5%9E%8B&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-3-91038858.nonecase&spm=1018.2226.3001.4187人脸检测及识别python实现系列（5）——利用keras库训练人脸识别模型icon-default.png?t=N4P3https://blog.csdn.net/weixin_44491431/article/details/113839543?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522168601107216800185824691%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=168601107216800185824691&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduend~default-1-113839543-null-null.142^v88^control_2,239^v2^insert_chatgpt&utm_term=%E4%BD%BF%E7%94%A8Keras%E6%9E%84%E5%BB%BA%E4%BA%BA%E8%84%B8%E8%AF%86%E5%88%AB%E6%A8%A1%E5%9E%8B&spm=1018.2226.3001.4187手把手教你使用Keras进行人脸检测和识别icon-default.png?t=N4P3https://blog.csdn.net/m0_55479420/article/details/115268470?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522168601107216800185824691%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=168601107216800185824691&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduend~default-2-115268470-null-null.142^v88^control_2,239^v2^insert_chatgpt&utm_term=%E4%BD%BF%E7%94%A8Keras%E6%9E%84%E5%BB%BA%E4%BA%BA%E8%84%B8%E8%AF%86%E5%88%AB%E6%A8%A1%E5%9E%8B&spm=1018.2226.3001.4187

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/697738
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号