人脸识别经典算法实现

原文链接: https://blog.csdn.net/guoyunfei123/article/details/81261792

 

一、原理

opencv支持3种人脸识别的算法,分别是:
1.    Eigen Faces    PCA(特征脸方法)
2.    Fisher Faces    LDA(线性判别分析)
3.    Local Binary Pattern Histograms(LBP 局部二值模式直方图)

 

详细介绍可以参照以下链接

https://blog.csdn.net/wanghz999/article/details/78817265

https://blog.csdn.net/u010006643/article/details/46417127

 

以下是部分摘抄:

Eigenfaces的原理

Eigenfaces就是特征脸的意思,是一种从主成分分析(Principal Component Analysis,PCA)中导出的人脸识别和描述技术。特征脸方法的主要思路就是将输入的人脸图像看作一个个矩阵,通过在人脸空间中一组正交向量,并选择最重要的正交向量,作为“主成分”来描述原来的人脸空间。为了更好地理解特征脸方法,需要先了解PCA的主要过程。

PCA主要过程

在很多应用中,需要对大量数据进行分析计算并寻找其内在的规律,但是数据量巨大造成了问题分析的复杂性,因此我们需要一些合理的方法来减少分析的数据和变量同时尽量不破坏数据之间的关联性。于是这就有了主成分分析方法,PCA作用:

  • 数据降维。减少变量个数;确保变量独立;提供一个合理的框架解释。
  • 去除噪声,发现数据背后的固有模式。

PCA的主要过程:

  1. 特征中心化:将每一维的数据(矩阵A)都减去该维的均值,使得变换后(矩阵B)每一维均值为0;
  2. 计算变换后矩阵B的协方差矩阵C;
  3. 计算协方差矩阵C的特征值和特征向量;
  4. 选取大的特征值对应的特征向量作为”主成分”,并构成新的数据集;

特征脸方法

特征脸方法就是将PCA方法应用到人脸识别中,将人脸图像看成是原始数据集,使用PCA方法对其进行处理和降维,得到“主成分”——即特征脸,然后每个人脸都可以用特征脸的组合进行表示。这种方法的核心思路是认为同一类事物必然存在相同特性(主成分),通过将同一目标(人脸图像)的特性寻在出来,就可以用来区分不同的事物了。人脸识别嘛,就是一个分类的问题,将不同的人脸区分开来。特征脸方法的过程(先计算特征脸,然后识别人脸):

  1. 将训练集中的N个人脸拉成一列(reshape(1,1)),然后组合在一起形成一个大矩阵A。若人脸图像大小为m * m,则矩阵A的维度是m * m * N;
  2. 将N个人脸在对应的维度求平均,得到一个“平均脸”;
  3. 将矩阵A中N个图像都减去“平均脸”,得到新矩阵B;
  4. 计算B的协方差矩阵;
  5. 计算协方差矩阵的特征值和特征向量(特征脸);
  6. 将训练集图像和测试集图像都投影到特征向量空间中,再使用聚类方法(最近邻或k近邻等)得到里测试集中的每个图像最近的图像,进行分类即可。

在opencv中void EigenFaceRecognizer::train();就是对训练集进行处理最终得到特征向量和阈值的过程,其中的阈值用于后面对测试图像的识别。int EigenFaceRecognizer:: predict(InputArray src) const;是对测试图像进行识别。

特征脸识别的局限性

要让系统准确识别需要保证人脸图像满足:

  1. 待识别图像中人脸尺寸接近特征脸中人脸的尺寸;
  2. 待识别人脸图像必须为正面人脸图像。

若不满足此条件,识别错误率很高。从PCA方法的过程可以看出,特征脸识别的方法是以每张人脸的一个维度(可以看出是矩阵的一列)为单位进行处理的,求得的特征向量(特征脸)中包含训练集每个纬度的绝大部分信息。但是若测试集中人脸尺寸不同,那么与特征脸中维度的也就没法对应起来。

二、程序

人脸训练集的获取与制作

在人脸识别之前需要获取人脸的训练集,这里利用的是The ORL Database of Faces’数据库,也可以自己采集人脸图像进行检测然后调整大小,具体的过程可以参照:http://blog.csdn.net/wanghz999/article/details/78751406

cvs文件获取

人脸识别的过程就是对人脸图像进行训练然后读取人脸并进行识别。在训练时,计算机会对每张人脸的多张图像进行特征提取,然后根据特征值来区分不同的人脸。然而计算机并不知道两种不同的人脸彼此是哪张,因此需要告诉预先告诉计算机一个标签——区分不同的人脸。cvs文件就是包含人脸图像路径和标签的文件,格式如下:


   
   
   
   
  1. D: /workspace/VS/opencv/face_recg/face_recg/face/s1/ 1.pgm; 0
  2. D: /workspace/VS/opencv/face_recg/face_recg/face/s1/ 10.pgm; 0
  3. D: /workspace/VS/opencv/face_recg/face_recg/face/s1/ 2.pgm; 0
  4. D: /workspace/VS/opencv/face_recg/face_recg/face/s1/ 3.pgm; 0
  5. D: /workspace/VS/opencv/face_recg/face_recg/face/s1/ 4.pgm; 0
  6. ...

;之前的表示人脸图像的路径,;之后的0表示标签。opencv官方提供了create_cvs.py,该文件位于源码目录(OPENCV_SOURCE_DIR)的opencv_contrib模块目录下:

OPENCV_SOURCE_DIR/opencv_contrib/modules/face/samples/etc
   
   
   
   

也可以直接从github上获取:https://github.com/opencv/opencv_contrib/tree/master/modules/face/samples/etc

这里给出我使用的(python3.*适用): 
#!/usr/bin/env python


   
   
   
   
  1. import sys
  2. import os.path
  3. # This is a tiny script to help you creating a CSV file from a face
  4. # database with a similar hierarchie:
  5. #
  6. # philipp@mango:~/facerec/data/at$ tree
  7. # .
  8. # |-- README
  9. # |-- s1
  10. # | |-- 1.pgm
  11. # | |-- ...
  12. # | |-- 10.pgm
  13. # |-- s2
  14. # | |-- 1.pgm
  15. # | |-- ...
  16. # | |-- 10.pgm
  17. # ...
  18. # |-- s40
  19. # | |-- 1.pgm
  20. # | |-- ...
  21. # | |-- 10.pgm
  22. #
  23. if __name_ _ == "__main__":
  24. #if len(sys.argv) != 2:
  25. # print "usage: create_csv "
  26. # sys.exit(1)
  27. #BASE_PATH=sys.argv[1]
  28. BASE_PATH= "D:/workspace/VS\opencv/face_recg/face_recg/face/"
  29. SEPARATOR= ";"
  30. fh = open( "at.txt", 'w')
  31. label = 0
  32. for dirname, dirnames, filenames in os.walk(BASE_PATH):
  33. for subdirname in dirnames:
  34. subject_path = os.path.join(dirname, subdirname)
  35. for filename in os.listdir(subject_path):
  36. abs_path = "%s/%s" % (subject_path, filename)
  37. print ( "%s%s%d" % (abs_path, SEPARATOR, label))
  38. fh.write(abs_path)
  39. fh.write(SEPARATOR)
  40. fh.write(str(label))
  41. fh.write( "\n")
  42. label = label + 1
  43. fh.close()

其中,BASE_PATH表示存放人脸图像的文件夹,在该文件夹下,每个文件夹代表一个人脸,一个人脸应该有多张大小一样的图。该文件夹如下: 

opencv人脸识别模块简介

opencv支持3种人脸识别的算法,分别是:

  1. Eigen Faces
  2. Fisher Faces
  3. Local Binary Pattern Histograms(局部二值模式直方图)

自opencv3.0开始,人脸识别的模块也有许多变动,实现跟以前一些很多不一样。上述的3种算法在opencv3.*中对应三个类:EigenFaceRecognizerFisherFaceRecognizerLBPHFaceRecognizer,它们都是从BasicFaceRecognizer继承过来的子类。这三个类有相似的API函数(以EigenFaceRecognizer为例):


   
   
   
   
  1. /*创建人脸识别的模型,若该两个参数为空,会以默认值创建
  2. @param num_components:进行训练的人脸图片数
  3. @param threshold: 阈值*/
  4. Ptr EigenFaceRecognizer::create(int num_components,
  5. double threshold);
  6. /*对模型进行训练
  7. @param src:进行训练的人脸图像
  8. @param labels:标签值,每张相同的人脸都有标签值来区分*/
  9. void train(InputArrayOfArrays src, InputArray labels);
  10. /*利用训练好的模型进行人脸识别,返回标签值
  11. @param src:需要进行识别的图像/
  12. CV_WRAP_AS(predict_label) int predict(InputArray src) const;

opencv人脸识别代码实现


   
   
   
   
  1. #include
  2. #include
  3. #include
  4. #include
  5. #include
  6. #include
  7. #include
  8. using namespace std;
  9. using namespace cv;
  10. using namespace cv:: face;
  11. //使用CVS文件读取图片和标签
  12. void read_cvs( const string& filename, vector& img, vector& lable, char separator = ';')
  13. {
  14. Mat tmp_img;
  15. ifstream cvs_file(filename.c_str());
  16. assert(cvs_file);
  17. string line, path, tag;
  18. while (getline(cvs_file, line))
  19. {
  20. stringstream lines(line);
  21. getline(lines, path, separator);
  22. getline(lines, tag);
  23. if (!path. empty() && !tag. empty())
  24. {
  25. tmp_img = imread(path, IMREAD_GRAYSCALE); //读入图像时转成灰度图
  26. assert(!tmp_img. empty());
  27. img.push_back(tmp_img);
  28. lable.push_back(atoi(tag.c_str()));
  29. }
  30. }
  31. tmp_img.release();
  32. }
  33. int main()
  34. {
  35. vector faces;
  36. vector labels;
  37. string cvs_path = ".\\face_recg\\face\\at.txt";
  38. try
  39. {
  40. read_cvs(cvs_path, faces, labels); //读取人脸图像和标签
  41. }
  42. catch (cv::Exception e)
  43. {
  44. printf( "Error open file, reason: %s\n", e.msg);
  45. return -1;
  46. }
  47. if (faces.size() <= 1)
  48. {
  49. printf( "Too few face images\n");
  50. return -1;
  51. }
  52. /* 从训练集中读取一张图片作为测试图 */
  53. Mat test_face;
  54. int sample_num = 0;
  55. int test_label;
  56. sample_num = (int)faces.size();
  57. test_face = faces[sample_num - 1];
  58. test_label = labels[sample_num - 1];
  59. faces.pop_back();
  60. labels.pop_back();
  61. double start1, start2, start3;
  62. double end1, end2, end3;
  63. /* 创建人脸识别的模型并进行训练,之后保存训练结果 */
  64. //PCA特征脸算法
  65. start1 = (double)getTickCount();
  66. Ptr eigen_model = EigenFaceRecognizer::create();
  67. eigen_model->train(faces, labels);
  68. eigen_model->save( "my_eigen_face_model.xml");
  69. end1 = (getTickCount() - start1) / getTickFrequency();
  70. //LDA线性判别分析
  71. start2 = (double)getTickCount();
  72. Ptr fisher_model = FisherFaceRecognizer::create();
  73. fisher_model->train(faces, labels);
  74. fisher_model->save( "my_fisher_face_model.xml");
  75. end2 = (getTickCount() - start2) / getTickFrequency();
  76. //LBP局部二值模式直方图
  77. start3 = (double)getTickCount();
  78. Ptr lbph_model = LBPHFaceRecognizer::create();
  79. lbph_model->train(faces, labels);
  80. lbph_model->save( "my_lbph_face_model.xml");
  81. end3 = (getTickCount() - start3) / getTickFrequency();
  82. printf( "PAC算法训练所用时间: %f s\n", end1);
  83. printf( "LDA算法训练所用时间: %f s\n", end2);
  84. printf( "LBP算法训练所用时间: %f s\n", end3);
  85. //对人脸进行测试,查看是否能够识别
  86. int p_label;
  87. p_label = eigen_model->predict(test_face);
  88. printf( "(PCA)Test label is: %d, predict label is: %d\n", test_label, p_label);
  89. p_label = fisher_model->predict(test_face);
  90. printf( "(LDA)Test label is: %d, predict label is: %d\n", test_label, p_label);
  91. p_label = lbph_model->predict(test_face);
  92. printf( "(LBP)Test label is: %d, predict label is: %d\n", test_label, p_label);
  93. waitKey( 0);
  94. return 0;
  95. }

人脸识别程序说明

训练和测试使用灰度图。EigenFaceRecognizer、 FisherFaceRecognizer两个类在调用train训练时可以使用BGR图像,但是训练结果中保存的是单通道的灰度图结果。可以查看生成的训练结果文件my_eigen_face_model.xml: 
 
可以看到在训练结果中保存的训练集是1*10304大小的图片,而我训练集图片大小为92 * 112(92 * 112 = 10304)。在对测试图进行人脸识别时,必须确保测试图大小与保存的训练集大小一致,否则会报错。而LBPHFaceRecognizer类必须使用灰度图进行训练,不然也会报错。

结果

可以看到程序可以对人脸进行识别,正确给出标签值。 

人脸识别经典算法实现_第1张图片

从算法耗时来看,PAC最耗时,LDA次之,LBP最省时。

从生成的训练结果文件来看,eigen算法最大,Fisher算法最小。

 

你可能感兴趣的:(算法)