【opencv实践】人脸识别匹配——模型训练

本文由@星沉阁冰不语出品，转载请注明作者和出处。

文章链接：http://blog.csdn.net/xingchenbingbuyu/article/details/51407336

在之前的博客人脸识别之一数据收集和预处理之中，已经下载了ORL人脸数据库，并且为了识别自己的人脸写了一个拍照程序自拍。之后对拍的照片进行人脸识别和提取，最后我们得到了一个包含自己的人脸照片的文件夹s41。在博客的最后我们提到了一个非常重要的文件——at.txt。

一、csv文件的生成

当我们写人脸模型的训练程序的时候，我们需要读取人脸和人脸对应的标签。直接在数据库中读取显然是低效的。所以我们用csv文件读取。csv文件中包含两方面的内容，一是每一张图片的位置所在，二是每一个人脸对应的标签，就是为每一个人编号。这个at.txt就是我们需要的csv文件。生成之后它里面是这个样子的：

前面是图片的位置，后面是图片所属人脸的人的标签。

要生成这样一个文件直接用手工的方式一个一个输入显然不可取的，毕竟这里有400多张图片。而且这种重复性的工作估计也没人想去做。所以我们可以用命令行的方式简化工作量；或者用OpenCV自带的Python脚本来自动生成。

命令行方式是这样的。比如我的数据集在C:\Users\bingbuyu\Downloads\att_faces文件夹下面，我就用下面两行命令：

然后数据集文件夹下面就多出了一个at.txt文件，但是现在是只有路径没有标签的。像下面这样：

标签需要手动敲上去。。。也挺麻烦的。

好在opencv教程里面为我们提供了自动生成csv文件的脚本。路径类似这样：F:\opencv\sources\modules\contrib\doc\facerec\src\create_csv.py。我不知道怎么用命令行参数的形式运行Python脚本。

用这个和大神的代码都没有实现，自己想了一个办法，非常简单，大家可以用这个方法来实现。先生成at.txt，接下来对它进行更改就可以了。代码如下：（按照自己的文件目录更改）

#include<iostream>
#include<fstream>
#include<stdio.h>
using namespace std;

int main()
{

    ofstream in;
    in.open("at.txt",ios::trunc);
    for(int i=1;i<=23;i++)
       for(int j=1;j<=9;j++)
          in<<"C:\\Users\\bingbuyu\\Downloads\\att_faces\\s"<<i<<"\\"<<j<<".jpg;"<<i<<"\n";
    in.close();

    return 0;
}

然后运行这个脚本就可以生成一个既有路径又有标签的at.txt了。

在这个过程中途换了图片，本来以为不影响生成的路径标签，结果。。程序报错，重新生成后就好了。大家注意一下这个问题！

二、训练模型

现在数据集、csv文件都已经准备好了。接下来要做的就是训练模型了。