opencv 人脸识别 （一）训练样本的处理

本文实现基于eigenface的人脸检测与识别。给定一个图像数据库，进行以下步骤：

• 进行人脸检测，将检测出的人脸存入数据库2
• 对数据库2进行人脸建模
• 在测试集上进行recognition

环境：vs2010+opencv 2.4.6.0

特征：eigenface

Input：一个人脸数据库，15个人，每人20个样本（左右）。

Output：人脸检测，并识别出每张检测到的人脸。

===============================

本文完成第一步，数据预处理：自动检测所有文件夹中每个sample中的人脸，作为训练数据。

Input：一个color文件夹，每个文件夹中有1~N这N个子文件夹，每个子文件夹内有n张包括第n类人的照片，如图。

最终结果：

核心：face detection（detectAndDraw）

辅助：截图并保存部分图片(CutImg)，文件夹内图片遍历(read_img)，图片转换成相同大小（normalizeone）

括号内分别是函数名，下面分别给出代码及说明。

1. 遍历文件夹：CBrowseDir类和CStatDir类（具体见这篇），三个文件如下：

1.1 BrowseDir.h

[cpp]view plaincopy

1. #pragmaonce
2. #include"direct.h"
3. #include"string.h"
4. #include"io.h"
5. #include"stdio.h"
6. #include<vector>
7. #include<iostream>
8. usingnamespacestd;
9. classCBrowseDir
10. {
11. protected:
12. charm_szInitDir[_MAX_PATH];
14. public:
15. CBrowseDir();
16. boolSetInitDir(constchar*dir);
17. boolBeginBrowse(constchar*filespec);
18. vector<char*>BeginBrowseFilenames(constchar*filespec);
20. protected:
21. boolBrowseDir(constchar*dir,constchar*filespec);
22. vector<char*>GetDirFilenames(constchar*dir,constchar*filespec);
23. virtualboolProcessFile(constchar*filename);
24. virtualvoidProcessDir(constchar*currentdir,constchar*parentdir);
25. };

1.2 BrowseDir.cpp

[cpp]view plaincopy

1. #include"BrowseDir.h"
2. #include"direct.h"
3. #include"string.h"
4. #include"io.h"
5. #include"stdio.h"
6. #include<vector>
7. #include<iostream>
8. usingnamespacestd;
10. CBrowseDir::CBrowseDir()
11. {
12. getcwd(m_szInitDir,_MAX_PATH);
13. intlen=strlen(m_szInitDir);
14. if(m_szInitDir[len-1]!='\')
15. strcat(m_szInitDir,"\");
16. }
18. boolCBrowseDir::SetInitDir(constchar*dir)
19. {
20. if(_fullpath(m_szInitDir,dir,_MAX_PATH)==NULL)
21. returnfalse;
23. if(_chdir(m_szInitDir)!=0)
24. returnfalse;
25. intlen=strlen(m_szInitDir);
26. if(m_szInitDir[len-1]!='\')
27. strcat(m_szInitDir,"\");
29. returntrue;
30. }
32. vector<char*>CBrowseDir::BeginBrowseFilenames(constchar*filespec)
33. {
34. ProcessDir(m_szInitDir,NULL);
35. returnGetDirFilenames(m_szInitDir,filespec);
36. }
38. boolCBrowseDir::BeginBrowse(constchar*filespec)
39. {
40. ProcessDir(m_szInitDir,NULL);
41. returnBrowseDir(m_szInitDir,filespec);
42. }
44. boolCBrowseDir::BrowseDir(constchar*dir,constchar*filespec)
45. {
46. _chdir(dir);
47. longhFile;
48. _finddata_tfileinfo;
49. if((hFile=_findfirst(filespec,&fileinfo))!=-1)
50. {
51. do
52. {
53. if(!(fileinfo.attrib&_A_SUBDIR))
54. {
55. charfilename[_MAX_PATH];
56. strcpy(filename,dir);
57. strcat(filename,fileinfo.name);
58. cout<<filename<<endl;
59. if(!ProcessFile(filename))
60. returnfalse;
61. }
62. }while(_findnext(hFile,&fileinfo)==0);
63. _findclose(hFile);
64. }
65. _chdir(dir);
66. if((hFile=_findfirst("*.*",&fileinfo))!=-1)
67. {
68. do
69. {
70. if((fileinfo.attrib&_A_SUBDIR))
71. {
72. if(strcmp(fileinfo.name,".")!=0&&strcmp
73. (fileinfo.name,"..")!=0)
74. {
75. charsubdir[_MAX_PATH];
76. strcpy(subdir,dir);
77. strcat(subdir,fileinfo.name);
78. strcat(subdir,"\");
79. ProcessDir(subdir,dir);
80. if(!BrowseDir(subdir,filespec))
81. returnfalse;
82. }
83. }
84. }while(_findnext(hFile,&fileinfo)==0);
85. _findclose(hFile);
86. }
87. returntrue;
88. }
90. vector<char*>CBrowseDir::GetDirFilenames(constchar*dir,constchar*filespec)
91. {
92. _chdir(dir);
93. vector<char*>filename_vec;
94. filename_vec.clear();
96. longhFile;
97. _finddata_tfileinfo;
98. if((hFile=_findfirst(filespec,&fileinfo))!=-1)
99. {
100. do
101. {
102. if(!(fileinfo.attrib&_A_SUBDIR))
103. {
104. char*filename=newchar[_MAX_PATH];
105. strcpy(filename,dir);
106. //intst=0;while(dir[st++]!='