0


【基于深度学习的人脸识别】(Dlib+ResNet残差神经网络)——QT(C++)+Linux

1.基于深度学习的人脸识别模型

1.人脸识别模型

dlib_face_recognition_resnet_model_v1.dat 是基于深度学习的人脸识别模型,是dlib库中的一个重要组件。该模型的原理涉及到深度卷积神经网络(DCNN)和具体的人脸识别算法。

dlib 人脸识别采用了 Resnet 残差神经网络,识别精度高于普通神经网络,同样我们可以到官网去下载训练好的模型 dlib_face_recognition_ resnet_ model_v1. dat,通过net(接口返回 128 维人脸特征,然后再通过目标图像也同样得到128 维人脸特征,将两组特征进行对比即可判断出要识别的对象。
以下是一些详细的解释:

1. 深度卷积神经网络 (DCNN): dlib_face_recognition_resnet_model_v1.dat使用了深度学习中的卷积神经网络。深度卷积神经网络是一种强大的模型,能够自动学习图像中的特征。在人脸识别任务中,这些特征可能包括脸部的形状、眼睛、鼻子、嘴巴等。
2. ResNet 架构:
模型采用了残差网络(ResNet)的架构。ResNet是一种具有跳跃连接(skip connection)的深度神经网络结构,能够更有效地训练深层网络。这对于人脸识别等任务是有益的,因为它允许网络学习残差,即输入与期望输出之间的差异。
3. 人脸识别算法:
该模型在训练过程中学习了一种将输入图像映射到高维特征空间中的算法。这个高维向量就是人脸的描述符,包含了识别人脸的信息。通过比较两个人脸的描述符,可以判断它们之间的相似性。
4. 训练数据:
该模型是在大规模的人脸数据集上进行训练的,以学习能够泛化到不同人脸的特征。这样,模型在实际应用中能够较好地适应各种人脸。

2.模型下载

下载链接(文件名dlib_face_recognition_resnet_model_v1.dat)
下载到Ubuntu中

2.人脸录入模块

1.创建人脸录入模块

  1. 创建新界面ui 在原本人脸特征值识别模块项目上新建一个界面在这里插入图片描述

这里我命名为Face_in
在主界面创建一个按钮实现跳转,并在face_in界面上创建一个返回按钮,接下来实现两界面的跳转

face_in.ui
在这里插入图片描述

mianwindow.ui
在这里插入图片描述

  1. 界面跳转逻辑实现

这里我们使用信号与槽机制实现界面跳转(以下只是简单示例)

  1. //先在mainwindow.h中声明face_in界面和槽函数
  2. Face_in *face_in;
  3. void backfromfacein();
  4. //构造函数中实例化
  5. face_in = new Face_in(); //人脸录入界面
  6. face_in->setGeometry(this->geometry());
  7. face_in->hide();
  8. //人脸录入按钮点击事件
  9. void MainWindow::on_face_in_clicked(){
  10. cap.release(); //关闭摄像头
  11. timer->stop();
  12. this->hide();
  13. face_in->show();}
  14. //槽函数
  15. void MainWindow::backfromfacein(){
  16. cap.open(0);
  17. timer->start(10);
  18. this->show();
  19. face_in->hide();}
  20. //face_in.h中
  21. signals:
  22. void backtohome();
  23. private slots:
  24. void on_back_clicked(); //返回主界面
  25. void on_start_face_clicked(); //开始录入人脸
  26. void on_open_camera_clicked(); //开启摄像头
  27. void updateframe(); //画面流
  28. private
  29. VideoCapture cap;
  30. Mat frame;
  31. QTimer* timer;
  32. //face_in.cpp中
  33. void Face_in::on_back_clicked(){
  34. timer->stop();
  35. cap.release();
  36. emit backtohome();}
  37. //开始采集人脸
  38. void Face_in::on_start_face_clicked(){
  39. //后续人脸录入逻辑
  40. }
  41. //开启摄像头
  42. void Face_in::on_open_camera_clicked(){
  43. cap.open(0);if(!cap.isOpened()){
  44. qDebug()<<"Open camera failed!";return;}
  45. // 创建定时器,每隔一定时间更新图像(timer在构造函数中实例化)
  46. connect(timer, &QTimer::timeout, this, &Face_in::updateframe);
  47. timer->start(30); // 设置定时器间隔,单位毫秒
  48. }
  49. void Face_in::updateframe(){if(cap.isOpened()){
  50. cap >> frame;
  51. // 反转图像
  52. flip(frame, frame, 1);
  53. // 转换到label上显示
  54. QImage img(frame.data, frame.cols, frame.rows, frame.step, QImage::Format_RGB888);
  55. this->ui->label->setPixmap(QPixmap::fromImage(img.rgbSwapped()));
  56. this->ui->label->setFixedSize(frame.cols, frame.rows);}}

如果不了解QT信号与槽机制,可参考
以上实现了两界面的切换,并且可在人脸录入模块显示摄像头视频

2.人脸录入模块逻辑实现

我们利用摄像头对人像进行拍照并提取128D人脸特征值保存到人脸数据文件中,以便进行人脸识别
4.14:优化了人脸录入代码:录入时进行灰度化,均衡化处理,并且在采集时显示采集的照片,询问用户是否保存。并且将人脸录入和人脸识别拆开了(上文的界面跳转没有用了,仅供参考)

  1. 利用dlib获取人脸特征
  1. #ifndef WIDGET_H#define WIDGET_H#include <QWidget>#include <dlib/dnn.h>#include <dlib/image_io.h>#include <dlib/image_processing.h>#include <dlib/opencv.h>#include <dlib/image_processing/frontal_face_detector.h>#include <dlib/image_processing/render_face_detections.h>#include <opencv2/opencv.hpp>#include <opencv2/core.hpp>#include <QDebug>#include <QTimer>#include <QMessageBox>
  2. using namespace cv;
  3. using namespace dlib;
  4. using namespace std;
  5. // ----------------------------------------------------------------------------------------
  6. template <template <int,template<typename>class,int,typename> class block, int N, template<typename>class BN, typename SUBNET>
  7. using residual = add_prev1<block<N,BN,1,tag1<SUBNET>>>;
  8. template <template <int,template<typename>class,int,typename> class block, int N, template<typename>class BN, typename SUBNET>
  9. using residual_down = add_prev2<avg_pool<2,2,2,2,skip1<tag2<block<N,BN,2,tag1<SUBNET>>>>>>;
  10. template <int N, template <typename> class BN, int stride, typename SUBNET>
  11. using block = BN<con<N,3,3,1,1,relu<BN<con<N,3,3,stride,stride,SUBNET>>>>>;
  12. template <int N, typename SUBNET> using ares = relu<residual<block,N,affine,SUBNET>>;
  13. template <int N, typename SUBNET> using ares_down = relu<residual_down<block,N,affine,SUBNET>>;
  14. template <typename SUBNET> using alevel0 = ares_down<256,SUBNET>;
  15. template <typename SUBNET> using alevel1 = ares<256,ares<256,ares_down<256,SUBNET>>>;
  16. template <typename SUBNET> using alevel2 = ares<128,ares<128,ares_down<128,SUBNET>>>;
  17. template <typename SUBNET> using alevel3 = ares<64,ares<64,ares<64,ares_down<64,SUBNET>>>>;
  18. template <typename SUBNET> using alevel4 = ares<32,ares<32,ares<32,SUBNET>>>;
  19. using anet_type = loss_metric<fc_no_bias<128,avg_pool_everything<
  20. alevel0<
  21. alevel1<
  22. alevel2<
  23. alevel3<
  24. alevel4<
  25. max_pool<3,3,2,2,relu<affine<con<32,7,7,2,2,
  26. input_rgb_image_sized<150>>>>>>>>>>>>>;
  27. // ----------------------------------------------------------------------------------------
  28. std::vector<matrix<rgb_pixel>> jitter_image(
  29. const matrix<rgb_pixel>& img
  30. );
  31. // ----------------------------------------------------------------------------------------
  32. QT_BEGIN_NAMESPACE
  33. namespace Ui { class Widget;}
  34. QT_END_NAMESPACE
  35. class Widget : public QWidget
  36. {
  37. Q_OBJECT
  38. public:
  39. Widget(QWidget *parent = nullptr);
  40. ~Widget();
  41. private slots:
  42. void on_start_face_clicked();
  43. void on_open_camera_clicked();
  44. void updateframe(); //画面流
  45. void on_close_camera_clicked();
  46. void updateImageDisplay(const cv::Mat& image);//显示采集的图像
  47. private:
  48. Mat frame;
  49. VideoCapture cap;
  50. QTimer* timer;
  51. frontal_face_detector detector; //dlib的人脸检测器
  52. shape_predictor sp; //预测器
  53. anet_type net; // 创建dlib的人脸识别器
  54. std::vector<dlib::rectangle> faces; //检测到的人脸
  55. QImage currentImage;//存储当前采集的图像
  56. Ui::Widget *ui;};#endif // WIDGET_H
  1. #include "widget.h"#include "ui_widget.h"
  2. Widget::Widget(QWidget *parent): QWidget(parent)
  3. , ui(new Ui::Widget){
  4. ui->setupUi(this);
  5. timer = new QTimer(this);
  6. //dlib的人脸关键点检测器
  7. detector = get_frontal_face_detector();
  8. deserialize("/home/tj/model/shape_predictor_68_face_landmarks.dat")>> sp;
  9. //dlib的人脸识别器
  10. deserialize("/home/tj/model/dlib_face_recognition_resnet_model_v1.dat")>> net;}
  11. Widget::~Widget(){
  12. delete ui;}
  13. //采集人脸按钮
  14. void Widget::on_start_face_clicked(){
  15. timer->stop(); //先暂停视频采集
  16. if(cap.isOpened()){ //确保摄像头打开才进行采集
  17. //人脸录入逻辑实现
  18. QString name = ui->name_input->text(); //获取人名字
  19. if(name.isEmpty()){
  20. QMessageBox::warning(this, "提示", "请输入人名!");return;}
  21. Mat processedFrame = frame.clone();
  22. cvtColor(processedFrame,processedFrame,CV_BGR2GRAY);//将彩色图转换从灰度图
  23. equalizeHist(processedFrame,processedFrame);//进行均衡化处理
  24. // 将OpenCV灰度图转换为dlib图像
  25. dlib::cv_image<unsigned char> dlib_img(processedFrame);
  26. // 获取人脸关键点
  27. full_object_detection shape = sp(dlib_img, faces[0]);
  28. // 截取人脸部分并调整大小对齐
  29. matrix<rgb_pixel> face_chip;
  30. extract_image_chip(dlib_img, get_face_chip_details(shape, 150, 0.25), face_chip);
  31. // 提取人脸特征值
  32. matrix<float, 0, 1> face_descriptor = net(face_chip);
  33. // 转换为OpenCV格式
  34. cv::Mat faceMat(face_chip.nr(), face_chip.nc(), CV_8UC1); // 灰度图,单通道
  35. for(int r =0; r < face_chip.nr(); ++r){for(int c =0; c < face_chip.nc(); ++c){
  36. // 因为是灰度图,只取一个通道的值
  37. faceMat.at<uchar>(r, c)= static_cast<uchar>(face_chip(r, c).blue);}}
  38. // 调整图像大小
  39. cv::resize(faceMat, faceMat, cv::Size(processedFrame.cols, processedFrame.rows));
  40. // 更新图像显示
  41. updateImageDisplay(faceMat);
  42. QMessageBox msgBox;
  43. msgBox.setText("是否保存此人脸图形");
  44. msgBox.setStandardButtons(QMessageBox::Yes | QMessageBox::No);
  45. // 获取当前窗口的底部中心位置
  46. int x = this->x() + this->width() / 2 - msgBox.width() / 2; // 居中对齐
  47. int y = this->y() + this->height() - msgBox.height(); // 底部对齐
  48. // 将QMessageBox移动到窗口的底部中心
  49. msgBox.move(x, y);
  50. int reply = msgBox.exec();if(reply == QMessageBox::Yes){
  51. // 用户选择了“是”
  52. //保存人脸图片到文件夹
  53. string save_path ="/home/tj/faces/";
  54. string photo_name = save_path + name.toStdString()+ + ".jpg";
  55. imwrite(photo_name, faceMat);
  56. // 保存人脸特征值到文件
  57. string file_name = save_path + name.toStdString() + ".dat";
  58. ofstream outfile(file_name, ios::binary);
  59. serialize(face_descriptor, outfile);
  60. outfile.close();
  61. QMessageBox::information(this, "提示", name + "保存成功!");
  62. timer->start(30); //继续视频
  63. }else{
  64. // 用户选择了“否”
  65. timer->start(30); //继续视频
  66. return;}}
  67. //没有打开摄像头
  68. else{
  69. QMessageBox::information(this,"提示","请打开摄像头");}}
  70. //打开摄像头按钮
  71. void Widget::on_open_camera_clicked(){
  72. cap.open(0);if(!cap.isOpened()){
  73. qDebug()<<"Open camera failed!";return;}
  74. // 创建定时器,每隔一定时间更新图像
  75. connect(timer, &QTimer::timeout, this, &Widget::updateframe);
  76. timer->start(30); // 设置定时器间隔,单位毫秒
  77. }
  78. void Widget::updateframe(){
  79. cap >> frame;
  80. // 反转图像
  81. flip(frame, frame, 1);
  82. // 将OpenCV图像转换为dlib图像
  83. dlib::cv_image<dlib::bgr_pixel> dlibImage(frame);
  84. // 使用人脸检测器检测人脸
  85. faces = detector(dlibImage);
  86. // 对检测到的第人脸进行处理,有人脸才处理
  87. if(!faces.empty()){
  88. for(size_t i =0; i < faces.size(); ++i){
  89. // 在图像上绘制人脸框
  90. cv::rectangle(frame, cv::Point(faces[i].left(), faces[i].top()),
  91. cv::Point(faces[i].right(), faces[i].bottom()),
  92. cv::Scalar(0, 255, 0), 2);}}
  93. // 转换到label上显示
  94. QImage img(frame.data, frame.cols, frame.rows, frame.step, QImage::Format_RGB888);
  95. this->ui->label->setPixmap(QPixmap::fromImage(img.rgbSwapped()));
  96. this->ui->label->setFixedSize(frame.cols, frame.rows);}
  97. //关闭摄像头
  98. void Widget::on_close_camera_clicked(){
  99. cap.release();
  100. timer->stop();}
  101. //显示采集的图像
  102. void Widget::updateImageDisplay(const Mat &image){
  103. // 转换图像格式
  104. QImage img(image.data, image.cols, image.rows, image.step, QImage::Format_Grayscale8);
  105. currentImage = img.copy(); // 使用copy确保图像数据的独立性
  106. this->ui->label->setPixmap(QPixmap::fromImage(currentImage));
  107. this->ui->label->setFixedSize(image.cols, image.rows);}

serialize 和 deserialize 函数是 Dlib 库中用于序列化(将对象转换为字节流)和反序列化(将字节流还原为对象)的函数。这两个函数通常用于将对象保存到文件或从文件中加载对象。

serialize: 该函数用于将一个对象序列化为一个字节流,可以将这个字节流保存到文件、传输给其他系统等。在 Dlib 中,经过 serialize 处理的对象可以被保存为二进制文件。

deserialize: 该函数用于将一个字节流反序列化为原始对象。通过 deserialize,你可以还原之前使用 serialize序列化的对象。

这两个函数一起使用,使得你可以方便地保存和加载模型、数据、状态等信息。在本项目中, deserialize 函数用于加载训练好的人脸检测器、面部关键点检测器以及人脸识别器的模型

这里就成功将人脸特征保存至了指定目录下(代码做了详细注释)

3.人脸识别模块

人脸识别的主要流程:采集摄像头当前人脸数据生成128D人脸特征,再与人脸数据文件中的全部人脸特征进行对比,设置一个阈值,当两张人脸阈值小于指定阈值时,即判定属于同一个人

1.创建人脸识别模块

  1. dnn_face_recognition_ex.cpp代码参考dlib官网:参考
  2. 使用Dlib库定义的深度学习模型,基于残差网络(ResNet)结构的人脸识别模型。
  1. // ----------------------------------------------------------------------------------------
  2. template <template <int,template<typename>class,int,typename> class block, int N, template<typename>class BN, typename SUBNET>
  3. using residual = add_prev1<block<N,BN,1,tag1<SUBNET>>>;
  4. template <template <int,template<typename>class,int,typename> class block, int N, template<typename>class BN, typename SUBNET>
  5. using residual_down = add_prev2<avg_pool<2,2,2,2,skip1<tag2<block<N,BN,2,tag1<SUBNET>>>>>>;
  6. template <int N, template <typename> class BN, int stride, typename SUBNET>
  7. using block = BN<con<N,3,3,1,1,relu<BN<con<N,3,3,stride,stride,SUBNET>>>>>;
  8. template <int N, typename SUBNET> using ares = relu<residual<block,N,affine,SUBNET>>;
  9. template <int N, typename SUBNET> using ares_down = relu<residual_down<block,N,affine,SUBNET>>;
  10. template <typename SUBNET> using alevel0 = ares_down<256,SUBNET>;
  11. template <typename SUBNET> using alevel1 = ares<256,ares<256,ares_down<256,SUBNET>>>;
  12. template <typename SUBNET> using alevel2 = ares<128,ares<128,ares_down<128,SUBNET>>>;
  13. template <typename SUBNET> using alevel3 = ares<64,ares<64,ares<64,ares_down<64,SUBNET>>>>;
  14. template <typename SUBNET> using alevel4 = ares<32,ares<32,ares<32,SUBNET>>>;
  15. using anet_type = loss_metric<fc_no_bias<128,avg_pool_everything<
  16. alevel0<
  17. alevel1<
  18. alevel2<
  19. alevel3<
  20. alevel4<
  21. max_pool<3,3,2,2,relu<affine<con<32,7,7,2,2,
  22. input_rgb_image_sized<150>>>>>>>>>>>>>;
  23. // ----------------------------------------------------------------------------------------

代码(上文也是用到了该网络)中定义了一系列模板(template)和类型别名(using),这些构建块用于构造一个深层神经网络
1.block 模板: 一个基本的残差块,包含卷积层、批归一化层和激活函数。这是构建整个网络的基本单元。

2.residual 和 residual_down 类型别名: 分别表示残差块和降采样残差块。residual_down 在 avg_pool层之后执行降采样,其目的是减小特征图的尺寸。
3.ares, ares_down 类型别名: 表示使用残差块或降采样残差块的激活层。

4.alevel0 到 alevel4 类型别名: 表示不同层次的网络结构,每个层次都包含多个残差块或降采样残差块。

5.anet_type 类型别名: 表示整个神经网络模型,其中包含五个层次的残差块和一些池化层,最终使用 loss_metric进行人脸识别的损失计算。

总体而言,这段代码(应该位于处理人脸识别类的头文件中)定义了一个深度学习模型,用于人脸识别,其中使用了残差网络结构来提高模型的训练和推断性能。这是基于Dlib库的一种高级抽象。

我们利用上篇文章中的人脸68位特征值检测类来进行人脸识别,大概流程为:从当前摄像头人脸中提取出特征值,遍历人脸库(上一步保存的),利用欧氏距离的阈值作为是否是同一个人的判别

2.人脸识别逻辑实现

  1. 创建人脸识别函数recognizeFace,该函数功能就是在程序提取出当前人脸的128D特征时对人脸库中的人脸进行遍历,找到最符合要求的人脸输出名字
  1. //人脸识别函数
  2. string Faceprocess::recognizeFace(const matrix<rgb_pixel>&face_chip){
  3. // 从人脸形状中提取特征
  4. matrix<float, 0, 1> face_descriptor = net(face_chip);
  5. // 遍历已保存的人脸特征矩阵,计算特征之间的欧氏距离
  6. double min_distance = std::numeric_limits<double>::infinity();
  7. std::string recognized_name ="null"; // 默认值为 "null"for(size_t i =0; i < stored_descriptors.size(); ++i){
  8. double distance = length(face_descriptor - stored_descriptors[i]);
  9. // 如果距离小于阈值,认为是同一个人
  10. if(distance <0.6&& distance < min_distance){
  11. recognized_name = files[i].left(files[i].size() - 4).toStdString();
  12. min_distance = distance;}}
  13. // 如果没有找到匹配的人脸,返回"null"return recognized_name;}

可以在构造函数中进行一次性的遍历人脸库文件夹,然后将所有人脸特征存储到矩阵(stored_descriptors)中。这样,后续的识别过程就只需要比对当前人脸与这个矩阵即可,避免在每一帧都重新读取文件。

1.遍历人脸特征文件夹中的所有文件,读取每个文件的人脸特征,然后将这些特征存储到 stored_descriptors 矩阵中。files 用于迭代这些文件,获取文件路径,读取特征,最后将特征添加到 stored_descriptors 中。

2.在每一帧中,检测到人脸,并通过人脸形状提取特征,得到 face_descriptor。

3.遍历 stored_descriptors 中已保存的人脸特征,计算当前人脸特征与已保存特征之间的欧氏距离。
4.如果距离小于阈值(0.6),则认为当前人脸与已保存的某个人脸匹配,返回匹配的人脸名称。否则,返回空字符串。

  1. 总体识别代码

这里由于recognizeFace(face_chip);进行人脸识别的函数涉及到深度学习模型执行,相对于画面帧的更新速度而言,它要慢许多,因此我们在此使用QT所提供的简单的异步(多线程)操作,用来执行我们的深度学习人脸识别函数。
这里使用QtConcurrent::run来进行多线程操作,我的思路是这样的:由于QtConcurrent::run会将参数中的函数放在线程池中进行执行,并不会干扰主UI线程的执行,那么,如果我们在循环中调用QtConcurrent::run就会导致不间断重复创建和销毁线程,导致资源的浪费以及识别缓慢等问题,因此我们希望在打开摄像头后,就只创建一次线程,不断进行异步人脸识别,直到我们关闭摄像头或者关闭程序时才停止这个线程,这时候我们就需要将人脸识别函数进行修改,并将face_chip作为全局变量供人脸识别函数使用,关键代码如下:

  1. //人脸识别函数
  2. void Widget::recognizeFace(){
  3. //只要没点击停止,就一直识别
  4. while(start_face){
  5. try {
  6. if(face_chip.size()){//有人脸
  7. cout <<"人脸识别中"<< endl;
  8. // 从人脸形状中提取特征
  9. matrix<float, 0, 1> face_descriptor = net(face_chip);
  10. double min_distance = std::numeric_limits<double>::infinity();
  11. // 遍历已保存的人脸特征矩阵,计算特征之间的欧氏距离
  12. std::string recognized_name; // 默认值为空
  13. for(size_t i =0; i < stored_descriptors.size(); ++i){
  14. double distance = length(face_descriptor - stored_descriptors[i]);
  15. // 如果距离小于阈值,认为是同一个人
  16. if(distance <0.4&& distance < min_distance){
  17. // cout << distance <<endl;
  18. recognized_name = files[i].left(files[i].size() - 4).toStdString();
  19. min_distance = distance;}}
  20. // 如果没有找到匹配的人脸,返回"null"
  21. recognizedName = recognized_name;
  22. // 等待一段时间或者进行其他处理,然后再进行下一次识别
  23. QThread::msleep(100);}
  24. //如果没有人脸
  25. else{continue;}} catch (const std::exception& ex){
  26. qDebug()<<"Caught exception in recognizeFace: "<< ex.what();
  27. // 根据需要处理异常
  28. }}
  29. if(!start_face){return; // 结束线程
  30. }}
  31. //开启摄像头
  32. void Widget::on_open_camera_clicked(){
  33. // 打开摄像头
  34. cap.open(0);if(!cap.isOpened()){
  35. qDebug()<<"open camera faild!";}
  36. timer->start(10);
  37. start_face =true;//启动人脸识别线程
  38. // 异步人脸识别
  39. QtConcurrent::run([this]{ this->recognizeFace();});}
  40. //关闭摄像头
  41. void Widget::on_close_camera_clicked(){
  42. cap.release();
  43. timer->stop();
  44. start_face =false;//停止人脸识别线程
  45. }

用户开启摄像头,如果在当前帧中检测到了人脸(即face_chip不为0),那么就会通过QtConcurrent::run函数异步执行recognizeFace()来进行人脸识别,否则就空循环(线程闲逛)。recognizeFace()函数会一直运行,直到start_face被设置为false,这个变量在点击停止按钮的时候被设置,这个时候recognizeFace()也会返回,让线程结束。
因此,只要摄像头开启并且存在人脸,且未点击停止按钮,程序就会一直进行异步人脸识别。

下面是原始代码,相比之下,异步识别效率明显提升。

  1. #include "faceprocess.h"
  2. Faceprocess::Faceprocess(QObject *parent): QObject(parent){
  3. // 创建dlib的人脸检测器
  4. detector = get_frontal_face_detector();
  5. deserialize("/home/tj/shape_predictor_68_face_landmarks.dat")>> sp;
  6. // 创建dlib的人脸识别器
  7. deserialize("/home/tj/dlib_face_recognition_resnet_model_v1.dat")>> net;
  8. // 遍历人脸库文件夹,将所有人脸特征存储到矩阵中
  9. std::string folderPath ="/home/tj/faces/";
  10. QDir directory(QString::fromStdString(folderPath));
  11. files = directory.entryList(QStringList("*.dat"), QDir::Files);
  12. // 用于存储累积的人脸描述符的矩阵
  13. for(const auto &file: files){
  14. std::string filePath = folderPath + file.toStdString();
  15. std::ifstream infile(filePath, std::ios::binary);if(!infile.is_open()){
  16. qDebug()<<"人脸库文件打开失败 "<< filePath.c_str();}else{
  17. // 文件成功打开,继续读取数据
  18. dlib::matrix<float, 0, 1> stored_descriptor;
  19. dlib::deserialize(stored_descriptor, infile);
  20. infile.close();
  21. // 将存储的描述符添加到矩阵中
  22. stored_descriptors.push_back(stored_descriptor);}}}
  23. Faceprocess::~Faceprocess(){
  24. // 清理 stored_descriptors
  25. stored_descriptors.clear();}
  26. void Faceprocess::processFrame(const cv::Mat &frame){
  27. cv::Mat processedFrame = frame.clone();
  28. // qDebug()<<"处理帧";
  29. // 将OpenCV图像转换为dlib图像
  30. dlib::cv_image<dlib::bgr_pixel> dlibImage(processedFrame);
  31. // 使用人脸检测器检测人脸
  32. std::vector<dlib::rectangle> faces = detector(dlibImage);
  33. // 对检测到的第一个人脸进行处理,有人脸才处理
  34. if(!faces.empty()){
  35. // 获取面部关键点
  36. dlib::full_object_detection shape = sp(dlibImage, faces[0]);
  37. // 在图像上绘制人脸框
  38. cv::rectangle(processedFrame, cv::Point(faces[0].left(), faces[0].top()), cv::Point(faces[0].right(), faces[0].bottom()), cv::Scalar(0, 255, 0), 2);
  39. // 在图像上绘制面部关键点
  40. for(unsigned int i =0; i < shape.num_parts(); ++i){
  41. cv::circle(processedFrame, cv::Point(shape.part(i).x(), shape.part(i).y()), 2, cv::Scalar(255, 0, 0), -1);}
  42. // 截取人脸部分并调整大小
  43. matrix<rgb_pixel> face_chip;
  44. extract_image_chip(dlibImage, get_face_chip_details(shape, 150, 0.25), face_chip);
  45. // 传入截取的人脸进行人脸识别
  46. recognizedName = recognizeFace(face_chip);
  47. // 输出识别结果
  48. if(!recognizedName.empty()){
  49. qDebug()<<"识别到的人脸:"<< recognizedName.c_str();
  50. // 在人脸矩形框上显示识别结果
  51. cv::putText(processedFrame, recognizedName, cv::Point(faces[0].left(), faces[0].top() - 5),
  52. cv::FONT_HERSHEY_COMPLEX, 0.5, cv::Scalar(255, 0, 0), 1);}else{
  53. qDebug()<<"未识别到人脸";
  54. // 在人脸矩形框上显示"unknown"
  55. cv::putText(processedFrame, "unknown", cv::Point(faces[0].left(), faces[0].top() - 5),
  56. cv::FONT_HERSHEY_COMPLEX, 0.5, cv::Scalar(255, 0, 0), 1);}}
  57. // 转换到QImage
  58. QImage img(processedFrame.data, processedFrame.cols, processedFrame.rows, processedFrame.step, QImage::Format_RGB888);
  59. emit frameProcessed(img); //处理完一帧,触发信号传送出去
  60. }
  61. //人脸识别函数
  62. string Faceprocess::recognizeFace(const matrix<rgb_pixel>&face_chip){
  63. // 从人脸形状中提取特征
  64. matrix<float, 0, 1> face_descriptor = net(face_chip);
  65. // 遍历已保存的人脸特征矩阵,计算特征之间的欧氏距离
  66. double min_distance = std::numeric_limits<double>::infinity();
  67. std::string recognized_name ="null"; // 默认值为 "null"for(size_t i =0; i < stored_descriptors.size(); ++i){
  68. double distance = length(face_descriptor - stored_descriptors[i]);
  69. // 如果距离小于阈值,认为是同一个人
  70. if(distance <0.6&& distance < min_distance){
  71. recognized_name = files[i].left(files[i].size() - 4).toStdString();
  72. min_distance = distance;}}
  73. // 如果没有找到匹配的人脸,返回"null"return recognized_name;}

这里有个问题:cv::putText并不能显示中文,我们利用QT对中文的支持,定义一个函数进行转化(QImage上绘制中文再转换为Mat)即可:

  1. #include <QPainter>#include <QFont>
  2. //解决puttext无法显示中文
  3. void Faceprocess::PutTextChinese(Mat &inMat, QString drawText, Point wordCenter, int inWordSize, Scalar inColor){
  4. QImage image;
  5. switch (inMat.type()){case CV_8UC1:
  6. image = QImage(inMat.data, inMat.cols, inMat.rows, inMat.cols, QImage::Format_Mono);break;case CV_8UC3:
  7. image = QImage(inMat.data, inMat.cols, inMat.rows, inMat.cols * 3, QImage::Format_RGB888);break;case CV_8UC4:
  8. image = QImage(inMat.data, inMat.cols, inMat.rows, inMat.cols * 4, QImage::Format_ARGB32);break;}
  9. QColor drawColor(inColor[0], inColor[1], inColor[2]);
  10. QPainter painter(&image);
  11. QPen pen = QPen(drawColor, inWordSize);
  12. QBrush brush = QBrush(drawColor);
  13. painter.setPen(pen);
  14. painter.setBrush(brush);
  15. QFont font("KaiTi", inWordSize);
  16. painter.setFont(font);
  17. painter.drawText(QPointF(wordCenter.x, wordCenter.y), drawText);}
  18. //直接将cv::putText替换为PutTextChinese即可

效果图:
在这里插入图片描述
在这里插入图片描述在这里插入图片描述

3.灰度化,均衡化

在人脸识别中,灰度化和均衡化起着重要的作用:

灰度化(Grayscale):
将彩色图像转换为灰度图像的过程。在灰度图像中,每个像素的数值表示图像中的亮度,而不再包含颜色信息。在人脸识别中,灰度化可以简化图像的复杂性,减少了颜色对识别的干扰,同时也降低了数据处理的复杂度。

均衡化(Histogram Equalization):
通过调整图像的直方图来增强图像的对比度,使得图像中的像素分布更加均匀。在人脸识别中,均衡化可以提高图像的亮度和对比度,使得人脸特征更加突出,从而提高识别的准确性。

综合来看,灰度化和均衡化可以帮助人脸识别系统更好地理解图像中的信息,减少不必要的噪声和干扰,从而提高识别的准确性和鲁棒性。

  1. void Widget::updateFrame(){
  2. // 读取帧
  3. cap >> frame;
  4. flip(frame, frame, 1);
  5. cv::Mat processedFrame = frame.clone(); //克隆一个图像用于人脸识别,保留原本的彩色图像
  6. cvtColor(processedFrame,processedFrame,CV_BGR2GRAY);//将彩色图转换从灰度图
  7. equalizeHist(processedFrame,processedFrame);//进行均衡化处理
  8. // // 显示灰度图像
  9. // cv::imshow("灰度图", processedFrame);
  10. // cv::waitKey(0);if(detector.num_detectors()==0){
  11. qDebug()<<"人脸检测器未正确初始化!";
  12. // 处理错误情况
  13. return;}
  14. // // 将OpenCV彩色图像转换为dlib图像
  15. // dlib::cv_image<dlib::bgr_pixel> dlibImage(processedFrame);
  16. // 将OpenCV灰度图转换为dlib图像
  17. dlib::cv_image<unsigned char> dlibImage(processedFrame);
  18. // 使用人脸检测器检测人脸
  19. std::vector<dlib::rectangle> faces = detector(dlibImage);
  20. // cout << faces.size()<< endl;
  21. // 对检测到的第一个人脸进行处理,有人脸才处理
  22. if(!faces.empty()){
  23. // 获取面部关键点
  24. dlib::full_object_detection shape = sp(dlibImage, faces[0]);
  25. // 在图像上绘制人脸框(在frame上绘制,防止传入神经网络的照片上有标记)
  26. cv::rectangle(frame, cv::Point(faces[0].left(), faces[0].top()), cv::Point(faces[0].right(), faces[0].bottom()), cv::Scalar(0, 255, 0), 2);
  27. // 在图像上绘制面部关键点
  28. for(unsigned int i =0; i < shape.num_parts(); ++i){
  29. cv::circle(frame, cv::Point(shape.part(i).x(), shape.part(i).y()), 2, cv::Scalar(255, 0, 0), -1);}
  30. // 根据68位关键点截取人脸部分并调整大小,人脸对齐
  31. matrix<rgb_pixel> face_chip;//人脸部分截图
  32. extract_image_chip(dlibImage, get_face_chip_details(shape, 150, 0.25), face_chip);
  33. // // 转换为OpenCV格式
  34. // cv::Mat faceMat(face_chip.nr(), face_chip.nc(), CV_8UC3);
  35. // for(int r =0; r < face_chip.nr(); ++r){
  36. // for(int c =0; c < face_chip.nc(); ++c){
  37. // faceMat.at<cv::Vec3b>(r, c)= cv::Vec3b(face_chip(r, c).blue, face_chip(r, c).green, face_chip(r, c).red);
  38. // }
  39. // }
  40. // // 显示图像
  41. // cv::imshow("Face Chip", faceMat);
  42. // cv::waitKey(0);if(++frameCounter % RECOGNITION_INTERVAL ==0){
  43. recognizedName = recognizeFace(face_chip);
  44. frameCounter =0; // 重置计数器
  45. }
  46. // 输出识别结果
  47. if(!recognizedName.empty()){
  48. // 在人脸矩形框上显示识别结果
  49. PutTextChinese(frame, QString::fromStdString(recognizedName), cv::Point(faces[0].left(), faces[0].top() - 5),
  50. 20, cv::Scalar(255, 0, 0));}else{
  51. //qDebug()<<"未识别到人脸";
  52. // 在人脸矩形框上显示"unknown"
  53. cv::putText(frame, "unknown", cv::Point(faces[0].left(), faces[0].top() - 5),
  54. cv::FONT_HERSHEY_COMPLEX, 0.5, cv::Scalar(255, 0, 0), 1);}}
  55. // 转换到QImage
  56. QImage img(frame.data, frame.cols, frame.rows, frame.step, QImage::Format_RGB888);
  57. this->ui->label->setPixmap(QPixmap::fromImage(img.rgbSwapped()));
  58. this->ui->label->setFixedSize(img.height(), img.width());}

测试:
在这里插入图片描述

4.人脸对齐

通过后续学习,我发现之前我并未进行人脸对齐的工作,但是在检测时,我们对图像进行灰度化,均衡化以及人脸对齐过后会更加准确地提起人脸特征值,在查阅相关文献(这里可参考
[1]刘兆丰.Dlib在人脸识别技术中的运用[J].电子制作,2020(21):39-41+7.DOI:10.16589/j.cnki.cn11-3571/tn.2020.21.015.),我发现在上述程序中已经实现了人脸对齐的工作,如下:

我们可知 extract_image_chip(dlibImage, get_face_chip_details(shape, 150, 0.25), face_chip);主要完成的工作就是:根据人脸68位关键点截取人脸部分,并且实现人脸对齐工作,具体原理可参考文献,下面是验证程序:

  1. // 根据68位关键点截取人脸部分并调整大小以及人脸对齐
  2. matrix<rgb_pixel> face_chip;//人脸部分截图
  3. extract_image_chip(dlibImage, get_face_chip_details(shape, 150, 0.25), face_chip);
  4. // 转换为OpenCV格式
  5. cv::Mat faceMat(face_chip.nr(), face_chip.nc(), CV_8UC3);for(int r =0; r < face_chip.nr(); ++r){for(int c =0; c < face_chip.nc(); ++c){
  6. faceMat.at<cv::Vec3b>(r, c)= cv::Vec3b(face_chip(r, c).blue, face_chip(r, c).green, face_chip(r, c).red);}}
  7. // 显示图像
  8. cv::imshow("Face Chip", faceMat);
  9. cv::waitKey(0);

测试:
在这里插入图片描述

实际上就已经完成了人脸对齐。


本文转载自: https://blog.csdn.net/weixin_53186909/article/details/135971816
版权归原作者 好想吃干锅 所有, 如有侵权,请联系我们删除。

“【基于深度学习的人脸识别】(Dlib+ResNet残差神经网络)——QT(C++)+Linux”的评论:

还没有评论