当前位置:   article > 正文

OpenCV--人脸识别入门_opencv人脸识别

opencv人脸识别

一、背景数据集

本次用的数据集是opencv给出的教程里面的第一个数据集:​​The AT&T Facedatabase​​,又称为ORL人脸数据库,40个人,每人10张照片。照片在不同时间、不同光照、不同表情(睁眼闭眼、笑或者不笑)、不同人脸细节(戴眼镜或者不戴眼镜)下采集。所有的图像都在一个黑暗均匀的背景下采集的,正面竖直人脸(有些有轻微旋转)。 

 

二、自己的人脸数据集

想要识别自己,单有别人的数据集还是不行的,还需要自己人脸的照片才行。这就需要我们收集自己的照片,然后和上面的那个数据集一起来训练模型。

Qt开发学习基地(免费报名学习):C/C++项目实战/Qt5/C语言/c++/数据库/OpenCV/MFC/QT项目-学习视频教程-腾讯课堂

2.1、人脸数据采集

下列程序调用openCV来拍照,按下P键拍照,按下Esc退出。

/** 拍照程序 **/

  1. #include "mainwindow.h"
  2. #include <QApplication>
  3. #include <QDebug>
  4. #include <opencv2/opencv.hpp>
  5. using namespace cv;
  6. int main(int argc, char *argv[])
  7. {
  8. QApplication a(argc, argv);
  9. namedWindow("photo",WINDOW_AUTOSIZE);
  10. VideoCapture cap(0);
  11. if(cap.isOpened())
  12. qDebug()<<"打开摄像头成功!";
  13. else
  14. qDebug()<<"打开摄像头失败!";
  15. Mat frame;
  16. int i=1;
  17. while (1)
  18. {
  19. char key = waitKey(100);
  20. cap >> frame;
  21. imshow("photo", frame);
  22. QString filename = QString("D:/Qt/Project/OpenCV/ORL_92x112/s41/s41_%1.bmp").arg(i);
  23. switch (key)
  24. {
  25. case 'p':
  26. imwrite(filename.toStdString(), frame);
  27. waitKey(500);
  28. i++;
  29. break;
  30. default:
  31. break;
  32. }
  33. int c = waitKey(0);
  34. if ((char)c == 27)
  35. return 0
  36. }
  37. return a.exec();
  38. }

注意变换角度,变换表情,拍完之后,精选十张帅照,用于自己的人脸数据集

 

2.2、预处理

在得到自己的人脸照片之后,还需要对这些照片进行一些预处理才能拿去训练模型。所谓预处理,其实就是检测并分割出人脸,并改变人脸图片的大小,需与下载的数据集中图片大小(92 x 112)一致。使用下列程序可以自动检测人脸、分割人脸、调整大小和存储。

注意:调用opencv训练好的分类器和自带的检测函数检测人脸,需要将OpenCV源代码中分类器事先放到自己自己的工程目录中去;

分类器位置:​​​D:\Qt\opencv-3.4.5\opencv-3.4.5\data\haarcascades\haarcascade_frontalface_default.xml​​ 移动到:自己工程的build目录下(必须这个目录)

 

  1. #include <QApplication>
  2. #include <QDebug>
  3. #include <opencv2/opencv.hpp>
  4. using namespace cv;
  5. int main(int argc, char *argv[])
  6. {
  7. QApplication a(argc, argv);
  8. QString face_cascade_name = "haarcascade_frontalface_default.xml";
  9. CascadeClassifier face_cascade; //定义人脸分类器
  10. QString window_name = "Capture - Face detection";
  11. namedWindow(window_name.toStdString(),WINDOW_AUTOSIZE);
  12. //-- 1. Load the cascades
  13. if (!face_cascade.load(face_cascade_name.toStdString()))
  14. {
  15. qDebug()<<"--(!)Error loading face cascade";
  16. return -1;
  17. }
  18. for(int i=1; i<=10; i++)
  19. {
  20. Mat img = imread(QString("D:/Qt/Project/OpenCV/ORL_92x112/s41/sources/s41_%1.bmp").arg(i).toStdString());
  21. std::vector<Rect> faces;
  22. Mat img_gray;
  23. cvtColor(img, img_gray, COLOR_BGR2GRAY);
  24. equalizeHist(img_gray, img_gray);
  25. //-- Detect faces
  26. face_cascade.detectMultiScale(img_gray, faces, 1.1, 3, CV_HAAR_DO_ROUGH_SEARCH, Size(50, 50));
  27. for (size_t j = 0; j < faces.size(); j++)
  28. {
  29. Mat faceROI = img_gray(faces[j]);
  30. Mat MyFace;
  31. if (faceROI.cols > 100)
  32. {
  33. resize(faceROI, MyFace, Size(92, 112));
  34. QString str = QString("D:/Qt/Project/OpenCV/ORL_92x112/s41/s41_%1.bmp").arg(i);
  35. imwrite(str.toStdString(), MyFace);
  36. imshow(window_name.toStdString(), MyFace);
  37. }
  38. waitKey(10);
  39. }
  40. }
  41. int c = waitKey(0);
  42. if ((char)c == 27)
  43. return 0;
  44. return a.exec();
  45. }

 

 

至此,我们就得到和ORL人脸数据库人脸大小一致的自己的人脸数据集。然后我们把自己的作为第41个人,在我们下载的人脸文件夹下建立一个s41的子文件夹,把自己的人脸数据放进去。就成了这样下面这样,最后一个文件夹里面是我自己的头像照片:

 

三、csv文件生成

当我们写人脸模型的训练程序的时候,我们需要读取人脸和人脸对应的标签。直接在数据库中读取显然是低效的,所以我们用csv文件读取。csv文件中包含两方面的内容:一是每一张图片的位置所在;二是每一个人脸对应的标签,就是为每一个人编号。如下图示:

 

这个工作可以自己纯手动去完成,但是400多张图片,也是很费时间的,珍惜生命,走下面这条路吧!

打开命令行,切换到人脸数据集目录下,输入如下指令:

dir /b/s *bmp > at.txt

 

 本文福利,费领取Qt开发学习资料包、技术视频,内容包括(C++语言基础,Qt编程入门,QT信号与槽机制,QT界面开发-图像绘制,QT网络,QT数据库编程,QT项目实战,QT嵌入式开发,Quick模块等等)↓↓↓↓↓↓见下面↓↓文章底部点击费领取↓↓

 

 

这个at.txt就是我们需要的csv文件,但是现在是只有路径没有标签的。使用下列程序,会生成一个at_temp.txt,这个既有路径又有标签,正是我们想要的,将之前的at.txt删除,将生成at_temp.txt改名为at.txt。

  1. #include <QApplication>
  2. #include <QDebug>
  3. #include <QFile>
  4. int main(int argc, char *argv[])
  5. {
  6. QApplication a(argc, argv);
  7. QString fileName1 = "D:/Qt/Project/OpenCV/ORL_92x112/at.txt";
  8. QString fileName2 = "D:/Qt/Project/OpenCV/ORL_92x112/at_temp.txt";
  9. QFile file1(fileName1);
  10. QFile file2(fileName2);
  11. file1.open(QIODevice::ReadOnly | QIODevice::Text);
  12. file2.open(QIODevice::WriteOnly | QIODevice::Text);
  13. int i=0;
  14. int count = 0;
  15. while (!file1.atEnd())
  16. {
  17. count++;
  18. QTextStream stream(&file2);
  19. QByteArray line = file1.readLine();
  20. QString str(line);
  21. str.replace('\\','/');
  22. str.replace('\n',';');
  23. stream << str<<QString("%1").arg(i)<<"\n";
  24. if(count%10 == 0)
  25. {
  26. i++;
  27. count=0;
  28. }
  29. }
  30. file1.close();
  31. file2.close();
  32. qDebug()<<"Done!!!";
  33. return a.exec();
  34. }

 

 

四、模型训练

  1. #include <QApplication>
  2. #include <opencv2\opencv.hpp>
  3. #include <iostream>
  4. #include <opencv2/face.hpp>
  5. #include <QDebug>
  6. using namespace cv;
  7. using namespace std;
  8. using namespace face;
  9. //使用CSV文件去读图像和标签,主要使用stringstream和getline方法
  10. static void read_csv(const string& filename, vector<Mat>& images, vector<int>& labels, char separator = ';') {
  11. std::ifstream file(filename.c_str(), ifstream::in);
  12. if (!file)
  13. {
  14. string error_message = "No valid input file was given, please check the given filename.";
  15. CV_Error(CV_StsBadArg, error_message);
  16. }
  17. string line, path, classlabel;
  18. while (getline(file, line))
  19. {
  20. stringstream liness(line);
  21. getline(liness, path, separator);
  22. getline(liness, classlabel);
  23. if (!path.empty() && !classlabel.empty())
  24. {
  25. images.push_back(imread(path, 0));
  26. labels.push_back(atoi(classlabel.c_str()));
  27. }
  28. }
  29. }
  30. int main(int argc, char *argv[])
  31. {
  32. QApplication a(argc, argv);
  33. //读取你的CSV文件路径.
  34. string fn_csv = "D:/Qt/Project/OpenCV/ORL_92x112/at.txt";
  35. // 2个容器来存放图像数据和对应的标签
  36. vector<Mat> images;
  37. vector<int> labels;
  38. // 读取数据. 如果文件不合法就会出错
  39. // 输入的文件名已经有了.
  40. try
  41. {
  42. read_csv(fn_csv, images, labels);
  43. }
  44. catch (cv::Exception& e)
  45. {
  46. cerr << "Error opening file \"" << fn_csv << "\". Reason: " << e.msg << endl;
  47. // 文件有问题,我们啥也做不了了,退出了
  48. exit(1);
  49. }
  50. // 如果没有读取到足够图片,也退出.
  51. if (images.size() <= 1)
  52. {
  53. string error_message = "This demo needs at least 2 images to work. Please add more images to your data set!";
  54. CV_Error(CV_StsError, error_message);
  55. }
  56. // 下面的几行代码仅仅是从你的数据集中移除最后一张图片
  57. //[gm:自然这里需要根据自己的需要修改,他这里简化了很多问题]
  58. Mat testSample = images[images.size() - 1];
  59. int testLabel = labels[labels.size() - 1];
  60. images.pop_back();
  61. labels.pop_back();
  62. // 下面几行创建了一个特征脸模型用于人脸识别,
  63. // 通过CSV文件读取的图像和标签训练它。
  64. // T这里是一个完整的PCA变换
  65. //如果你只想保留10个主成分,使用如下代码
  66. // cv::createEigenFaceRecognizer(10);
  67. //
  68. // 如果你还希望使用置信度阈值来初始化,使用以下语句:
  69. // cv::createEigenFaceRecognizer(10, 123.0);
  70. //
  71. // 如果你使用所有特征并且使用一个阈值,使用以下语句:
  72. // cv::createEigenFaceRecognizer(0, 123.0);
  73. Ptr<FaceRecognizer> model = EigenFaceRecognizer::create();
  74. model->train(images, labels);
  75. model->save("MyFacePCAModel.xml");
  76. Ptr<FaceRecognizer> model1 = FisherFaceRecognizer::create();
  77. model1->train(images, labels);
  78. model1->save("MyFaceFisherModel.xml");
  79. Ptr<FaceRecognizer> model2 = LBPHFaceRecognizer::create();
  80. model2->train(images, labels);
  81. model2->save("MyFaceLBPHModel.xml");
  82. // 下面对测试图像进行预测,predictedLabel是预测标签结果
  83. int predictedLabel = model->predict(testSample);
  84. int predictedLabel1 = model1->predict(testSample);
  85. int predictedLabel2 = model2->predict(testSample);
  86. QString result_message = QString("Predicted class = %1 | Actual class = %2").arg(predictedLabel).arg(testLabel);
  87. QString result_message1 = QString("Predicted class = %1 | Actual class = %2").arg(predictedLabel1).arg(testLabel);
  88. QString result_message2 = QString("Predicted class = %1 | Actual class = %2").arg(predictedLabel2).arg(testLabel);
  89. qDebug() << result_message << endl;
  90. qDebug() << result_message1 << endl;
  91. qDebug() << result_message2 << endl;
  92. waitKey(0);
  93. return a.exec();
  94. }

 

上述程序从人脸数据集中取了最后一张照片用于测试训练的模型。人脸数据集总共40+1个人,但是标签是从0开始的,所以最后的人脸标签是40,从上述程序运行结果可以看出,实际是40,预测也是40,说明模型训练成功,三个训练好的模型如下图:

 

五、人脸识别

  1. #include <QApplication>
  2. #include <opencv2\opencv.hpp>
  3. #include <iostream>
  4. #include <opencv2/face.hpp>
  5. #include <QDebug>
  6. using namespace cv;
  7. using namespace std;
  8. using namespace face;
  9. int main(int argc, char *argv[])
  10. {
  11. QApplication a(argc, argv);
  12. VideoCapture cap(0); //打开默认摄像头
  13. if (!cap.isOpened())
  14. {
  15. return -1;
  16. }
  17. Mat frame;
  18. Mat edges;
  19. Mat gray;
  20. CascadeClassifier cascade;
  21. bool stop = false;
  22. //训练好的文件名称,放置在可执行文件同目录下
  23. cascade.load("D:/Qt/opencv-3.4.5/opencv-3.4.5/data/haarcascades/haarcascade_frontalface_default.xml");
  24. //Ptr<FaceRecognizer> modelPCA = createEigenFaceRecognizer();
  25. //modelPCA->load("MyFacePCAModel.xml");
  26. Ptr<FaceRecognizer> modelPCA = EigenFaceRecognizer::create();
  27. modelPCA->read("D:/Qt/Project/build-OpenCV-Desktop_Qt_5_13_2_MinGW_64_bit-Debug/MyFacePCAModel.xml");//训练的模型
  28. while(!stop)
  29. {
  30. cap >> frame;
  31. //建立用于存放人脸的向量容器
  32. vector<Rect> faces(0);
  33. cvtColor(frame, gray, CV_BGR2GRAY);
  34. //改变图像大小,使用双线性差值
  35. //resize(gray, smallImg, smallImg.size(), 0, 0, INTER_LINEAR);
  36. //变换后的图像进行直方图均值化处理
  37. equalizeHist(gray, gray);
  38. cascade.detectMultiScale(gray, faces,
  39. 1.1, 2, 0
  40. //|CV_HAAR_FIND_BIGGEST_OBJECT
  41. //|CV_HAAR_DO_ROUGH_SEARCH
  42. | CV_HAAR_SCALE_IMAGE,
  43. Size(30, 30));
  44. Mat face;
  45. Point text_lb;
  46. for (size_t i = 0; i < faces.size(); i++)
  47. {
  48. if (faces[i].height > 0 && faces[i].width > 0)
  49. {
  50. face = gray(faces[i]);
  51. text_lb = Point(faces[i].x, faces[i].y);
  52. rectangle(frame, faces[i], Scalar(255, 0, 0), 1, 8, 0);
  53. }
  54. }
  55. Mat face_test;
  56. int predictPCA = 0;
  57. if (face.rows >= 120)
  58. {
  59. resize(face, face_test, Size(92, 112));
  60. imshow("缩放",face_test);
  61. }
  62. //Mat face_test_gray;
  63. //cvtColor(face_test, face_test_gray, CV_BGR2GRAY);
  64. if (!face_test.empty())
  65. {
  66. //测试图像应该是灰度图
  67. predictPCA = modelPCA->predict(face_test);
  68. }
  69. cout << predictPCA << endl;
  70. if (predictPCA == 40)
  71. {
  72. string name = "WangJiChuan";
  73. putText(frame, name, text_lb, FONT_HERSHEY_COMPLEX, 1, Scalar(255, 0, 0));
  74. }
  75. imshow("face", frame);
  76. if (waitKey(1) >= 0)
  77. stop = true;
  78. }
  79. return a.exec();
  80. }

从上图可知,自己的人脸标签是35,当识别的人脸为35时,识别成功为自己,标记人脸,贴上名字。 

 

 

C++ Qt是一个基于C++语言的跨平台应用程序开发框架,具有丰富的UI组件和API库。Qt包括了许多模块,如Qt Core、Qt GUI、Qt Network等,每个模块都提供了一系列的类和函数。Qt还提供了集成开发环境(IDE)Qt Creator,使得开发人员可以通过它来创建、编译、调试和部署Qt应用程序。 

学习Qt开发不知道做什么?我为大家整理的这些领域都涉及到Qt开发:嵌入式领域、桌面端开发、移动端、微控制器MCU、客户端(游戏、直播等等)、汽车领域行业、 消费类电子设备、医疗领域行业、工业自动化领域等等

Qt框架,GUI应用程序,跨平台开发,信号与槽机制,QML语言,模型视图编程,多线程编程,数据库编程,网络编程,XML解析,JSON解析,图形图像处理,用户界面设计,动画效果,OpenGL,WebKit,嵌入式开发,客户端/服务器应用程序.自定义控件QT6,C++语言基础,qt基础编程,qt软件开发,Qt架构设计,qt布局管理器,qt嵌入式开发,qt编程入门,qt数据库编程,qt跨平台框架,QT项目实战,Quick模块,OpenCV,Qt实战,OpenCV教程,QT界面开发,Qt框架,C++数据结构,Qt线程,桌面应用开发,qt桌面应用开发,Socket网络编程,qt开发工程师,qt开发,应用程序开发框架,图形视图框架,数据库编程,Qt开发编程,Qt开发控件,Qt开发工程师,QT开发必备技能栈,qt编码,qt网络编程,qt网络通信,Qt信号,Qt槽机制,qt字符串,qt数据类型,qt容器,qt客户端开发,qt软件工程师,qt页面绘制

 

 本文福利,费领取Qt开发学习资料包、技术视频,内容包括(C++语言基础,Qt编程入门,QT信号与槽机制,QT界面开发-图像绘制,QT网络,QT数据库编程,QT项目实战,QT嵌入式开发,Quick模块等等)↓↓↓↓↓↓见下面↓↓文章底部点击费领取↓↓

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

闽ICP备14008679号