当前位置:   article > 正文

OpenCV人脸识别实验(一)——特征脸(Eigenfaces)及其重构的源代码详解_teeigenfacerecognizer(10, 123.0);

teeigenfacerecognizer(10, 123.0);

1、介绍Introduction

从OpenCV2.4开始,加入了新的类FaceRecognizer,我们可以使用它便捷地进行人脸识别实验。

本实验采用的编程环境为:opencv3.0+VS2013。人脸识别的实验已经转移到face模块中,

face模块在我这里的路径为:D:\Program Files\opencv3.0\opencv\sources\modules\opencv_contrib-master\modules\face。

代码在OpenCV安装目录下为:D:\Program Files\opencv3.0\opencv\sources\modules\opencv_contrib-master\modules\face\samples\facerec_eigenfaces.cpp.

如果需要在高版本的opencv中使用face模块,需要自己进行编译,其编译和配置流程请参照题目为:

【图像处理】windows 10 + vs2015+ opencv3.0.0 +附加模块opencv_contrib编译和配置的博客,在此不在赘述。

face模块目前支持的算法有:

        (1)主成分分析(PCA)——Eigenfaces(特征脸)——函数:createEigenFaceRecognizer()

PCA:低维子空间是使用主元分析找到的,找具有最大方差的哪个轴。

缺点:若变化基于外部(光照),最大方差轴不一定包括鉴别信息,不能实行分类。

        (2)线性判别分析(LDA)——Fisherfaces特征脸——函数: createFisherFaceRecognizer()

LDA:线性鉴别的特定类投影方法,目标:实现类内方差最小,类间方差最大。

        (3)局部二值模式(LBP)——LocalBinary Patterns Histograms——函数:createLBPHFaceRecognizer()

PCA和LDA采用整体方法进行人脸辨别,LBP采用局部特征提取,除此之外,还有的局部特征提取方法为:

盖伯小波(Gabor Waelets)和离散傅里叶变换(DCT)。

2、人脸库及数据准备

在研究人脸识别,所以我们需要人脸图像!你可以自己创建自己的数据集,也可以从这里(http://face-rec.org/databases/)下载一个。本实验采用ORL人脸数据库,40个人,每人10张照片。本次实验ORL人脸库图像解压缩在的目录路径为:D:\Program Files\opencv3.0\opencv\sources\data\FaceData\ORL。CSV文件的目录路径为:

D:\Program Files\opencv3.0\opencv\sources\data\at.txt。我们需要在程序中读取它,我决定使用CSV文件读取它一个CSV文件包含文件名,紧跟一个标签

如:D:\Program Files\opencv3.0\opencv\sources\data\FaceData\ORL\s1\1.pgm;0

我们给它一个标签0,这个标签类似代表这个人的名字,所以同一个人的照片的标签都一样创建一个CSV文件,

at.txt文件的部分内容的截图如下:


3、算法描述

图像表示的问题是他的高维问题二维灰度图像p*q大小是一个m=qp维的向量空间

所以一个100*100像素大小的图像就是10,000维的图像空间。问题是,是不是所有的维数空间对我们来说都有用?

我们可以做一个决定,如果数据有任何差异,我们可以通过寻找主元来知道主要信息。

主成分分析(PrincipalComponent Analysis,PCA)KarlPearson (1901)独立发表的,

而 Harold Hotelling (1933)把一些可能相关的变量转换成一个更小的不相关的子集

想法是,一个高维数据集经常被相关变量表示,因此只有一些的维上数据才是有意义的,包含最多的信息

PCA方法寻找数据中拥有最大方差的方向,被称为主成分

令   表示一个随机特征,其中  .

1.   计算均值向量 

                             

2.   计算协方差矩阵 S

                                

3.   计算 的特征值    和对应的特征向量                

 

4.   特征值进行递减排序,特征向量和它顺序一致. K个主成分也就是k个最大的特征值对应的特征向量。

x的K个主成份:

         

其中  .

PCA基的重构:

                  

其中  .

然后特征脸通过下面的方式进行人脸识别:

A. 把所有的训练数据投影到PCA子空间

B. 把待识别图像投影到PCA子空间

C. 找到训练数据投影后的向量和待识别图像投影后的向量最近的那个。

4、代码实现及实验结果

  1. #include <opencv2/core/core.hpp>
  2. #include <opencv2/highgui/highgui.hpp>
  3. #include <opencv2/imgproc/imgproc.hpp>
  4. #include <opencv2/face/facerec.hpp>
  5. #include <iostream>
  6. #include <fstream> //文件操作的集合,以流的方式进行
  7. #include <sstream> //此库定义了stringstream类,即:流的输入输出操作。
  8. //使用string对象代替字符数组,避免缓冲区溢出的危险
  9. using namespace cv;
  10. using namespace std;
  11. using namespace cv::face;
  12. //归一化图像矩阵函数
  13. static Mat norm_0_255(InputArray _src)
  14. {
  15. Mat src = _src.getMat(); //将传入的类型为InputArray的参数转换为Mat的结构
  16. Mat dst; //创建和返回一个归一化后的图像
  17. switch (src.channels())
  18. {
  19. case 1:
  20. normalize(_src, dst, 0, 255, NORM_MINMAX, CV_8UC1);
  21. break;
  22. case 3:
  23. normalize(_src, dst, 0, 255, NORM_MINMAX, CV_8UC3);
  24. break;
  25. default:
  26. src.copyTo(dst);
  27. break;
  28. }
  29. return dst;
  30. }
  31. //使用CSV文件读取图像和标签,主要使用stringstream和getline方法
  32. static void read_csv(const string& filename, vector<Mat>& images, vector<int>& labels, char separator = ';')
  33. {
  34. ifstream file(filename.c_str(), ifstream::in); //以输入方式打开文件
  35. //c_str()函数将字符串转化为字符数组,返回指针
  36. if (!file)
  37. {
  38. string error_massage = "No valid input file was given,please check the given filename!";
  39. CV_Error(CV_StsBadArg, error_massage);
  40. }
  41. string line, path, classlabel;
  42. while (getline(file, line)) //getline(字符数组,字符个数n,终止标志字符)
  43. {
  44. stringstream liness(line);
  45. getline(liness, path, separator); //遇到分号就结束
  46. getline(liness, classlabel); //继续从分号后边开始,遇到换行结束
  47. if (!path.empty() && !classlabel.empty())
  48. {
  49. images.push_back(imread(path, 0));
  50. labels.push_back(atoi(classlabel.c_str())); //atoi函数将字符串转换为整数值
  51. }
  52. }
  53. }
  54. int main(int argc, const char* argv[])
  55. {
  56. //[1] 检测合法的命令,显示用法
  57. //如果没有参数输入,则退出
  58. //if (argc < 2)
  59. //{
  60. // cout << "usage:" << argv[0] << "<csv.ext> <output_folder>" << endl;
  61. // exit(1);
  62. //}
  63. string output_folder;
  64. output_folder = string("D:\\Program Files\\opencv3.0\\opencv\\sources\\data\\FaceData\\result6");
  65. //[2] 读取CSV文件路径
  66. string fn_csv = string("D:\\Program Files\\opencv3.0\\opencv\\sources\\data\\at.txt");
  67. //两个容器来存放图像数据和对应的标签
  68. vector<Mat> images;
  69. vector<int> labels;
  70. //读取数据,如果文件不合法就会出错。输入的文件名已经有了
  71. try{
  72. read_csv(fn_csv, images, labels);
  73. }
  74. catch (Exception& e)
  75. {
  76. cerr << "Error opening file" << fn_csv << ".Reason:" << e.msg << endl;
  77. exit(1);
  78. }
  79. //没有读取到足够多的图片,也需要退出
  80. if (images.size() <= 1)
  81. {
  82. string error_message = "This demo need at least 2 images,please add more images to your data set!";
  83. CV_Error(CV_StsError, error_message);
  84. }
  85. //[3] 得到第一张图片的高度,在下面对图像变形得到他们原始大小时需要
  86. int height = images[0].rows;
  87. //[4]下面代码仅从数据集中移除最后一张图片,用于做测试,需要根据自己的需要进行修改
  88. Mat testSample = images[images.size() - 1];
  89. int testLabel = labels[labels.size() - 1];
  90. images.pop_back(); //删除最后一张图片
  91. labels.pop_back(); //删除最后一个标签
  92. //[5] 创建一个特征脸模型用于人脸识别
  93. //通过CSV文件读取的图像和标签训练它
  94. //这里是一个完整的PCA 变换
  95. //如果想保留10个主成分,使用如下代码 cv::createEigenFaceRecognizer(10);
  96. //如果希望使用置信度阈值来初始化,使用代码 cv::createEigenFaceRecognizer(10, 123.0);
  97. //如果使用所有特征并使用一个阈值,使用代码 cv::createEigenFaceRecognizer(0, 123.0);
  98. Ptr<BasicFaceRecognizer> model = createEigenFaceRecognizer();
  99. model->train(images, labels);
  100. //[6] 对测试图像进行预测,predictedLabel是预测标签结果
  101. int predictedLabel = model->predict(testSample);
  102. // 还有一种调用方式,可以获取结果同时得到阈值:
  103. // int predictedLabel = -1;
  104. // double confidence = 0.0;
  105. // model->predict(testSample, predictedLabel, confidence);
  106. string result_message = format("Predicted class = %d / Actual class = %d.", predictedLabel, testLabel);
  107. cout << result_message << endl;
  108. //[7] 如何获取特征脸模型的特征值例子,使用getEigenValues方法
  109. Mat eigenvalues = model->getEigenValues();
  110. //[8] 获取特征向量
  111. Mat W = model->getEigenVectors();
  112. //[9] 得到训练图像的均值向量
  113. Mat mean = model->getMean();
  114. //[10] 显示或保存
  115. imshow("mean", norm_0_255(mean.reshape(1, images[0].rows)));
  116. imwrite(format("%s/mean.png", output_folder.c_str()), norm_0_255(mean.reshape(1, images[0].rows)));
  117. //[11] 显示或保存特征脸
  118. for (int i = 0; i < min(10, W.cols); i++) //修改数值10可以修改特征脸的数目
  119. {
  120. string msg = format("Eigenvalue #%d = %.5f", i,eigenvalues.at<double>(i));
  121. cout << msg << endl;
  122. //得到第i个特征向量
  123. Mat ev = W.col(i).clone();
  124. //把它变成原始大小,把数据显示归一化到0-255
  125. Mat grayscale = norm_0_255(ev.reshape(1, height));
  126. //使用伪彩色来显示结果,为了更好的观察
  127. Mat cgrayscale;
  128. applyColorMap(grayscale, cgrayscale, COLORMAP_JET);
  129. //显示或保存
  130. imshow(format("eigenface_%d", i), cgrayscale);
  131. imwrite(format("%s/eigenface_%d.png", output_folder.c_str(), i),norm_0_255(cgrayscale));
  132. }
  133. //[12] 预测过程中,显示或保存重建后的图像
  134. //修改值300可改变重构的图像的数目
  135. for (int num_components = min(W.cols, 10); num_components < min(W.cols,300); num_components += 15)
  136. {
  137. //从模型中的特征向量截取一部分
  138. Mat evs = Mat(W, Range::all(), Range(0, num_components));
  139. //在重构时,images[0]为ORL人脸库的第一张人脸图,
  140. //修改此数值0的大小可对其他人脸图像进行特征脸处理与重构的实验
  141. Mat projection = LDA::subspaceProject(evs, mean, images[0].reshape(1, 1));
  142. //投影样本到LDA子空间
  143. Mat reconstruction = LDA::subspaceReconstruct(evs, mean,projection);
  144. //重构来自于LDA子空间的投影
  145. //归一化结果
  146. reconstruction = norm_0_255(reconstruction.reshape(1, images[0].rows));
  147. //[13] 若不是存放到文件夹中就显示他,使用暂定等待键盘输入
  148. imshow(format("eigenface_reconstruction_%d", num_components),reconstruction);
  149. imwrite(format("%s/eigenface_reconstruction_%d.png",output_folder.c_str(), num_components), reconstruction);
  150. }
  151. waitKey(0);
  152. return 0;
  153. }

实验结果截图为:(特征值或特征脸数目为10,重构人脸图像为20)

     



修改代码:
(1)for (int i = 0; i < min(50, W.cols); i++)//修改数值50可以修改特征脸的数目
(2)//修改重构图像的范围或数目,修改值为400
for (int num_components = min(W.cols, 10); num_components < min(W.cols,400); num_components += 15)

实验运行结果截图为:(特征值或特征脸数目为50,重构人脸图像为26)








再次修改代码:(代码中红色注解部分)
//修改此数值0的大小为其他值,可对其他人脸图像进行特征脸处理与重构的实验
Mat projection = LDA::subspaceProject(evs, mean, images[50].reshape(1, 1)); 


实验截图为:



5、实验结论


(1)50个特征向量可以有效的编码出重要的人脸特征。

(2)对比下面两幅图,重构图像的范围或数目越大,重构的效果越好,重构的图片越清晰

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号