OpenCV EigenFaceRecognizer 人脸识别

目录

一、人脸检测vs人脸识别

二、数据库

三、准备需要识别的人脸

四、人脸识别(OpenCV)

---

一、人脸检测vs人脸识别

人脸检测只是需要检测图像中是否有人脸，而人脸识别是一个程序能够识别出给定图像或者视频中的人脸。

二、数据库

人脸数据库，这个博客有网盘链接。

三、准备需要识别的人脸

准备一个人的10张左右数量的人脸，放到刚刚下载ORL的数据库中：

 

捕获人脸代码：

#include <iostream>
#include <opencv2/opencv.hpp>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <dirent.h>

using namespace std;
using namespace cv;

void detectFace(Mat frame, string path)
{
    // 级联分类器加载文件
    CascadeClassifier facefier;
    facefier.load("/home/jason/file/opencv/haarcascade_frontalface_alt.xml");

    VideoCapture capture(0);
    vector<Rect> faces;
    Mat gray_frame;

    int number=0;
    while (1)
    {
        capture >> frame;
        imshow("detect face", frame);
        cvtColor(frame,gray_frame,COLOR_BGR2GRAY);
        equalizeHist(gray_frame,gray_frame);

        // 检测人脸
        facefier.detectMultiScale(gray_frame, faces);

        // 保存人脸
        for (size_t i=0; i < faces.size(); i++)
        {

            // 画框标出人脸
            Mat temp = frame.clone();
            rectangle(temp,faces[i], Scalar(255, 0,0), 1, 8);
            imshow("detect face", temp);

            // 扣出人脸
            Mat faceROI = frame(faces[i]);

            // 过滤
            if (faceROI.cols * faceROI.rows > 100 * 100)
            {
                resize(faceROI, faceROI, Size(92, 112));
                string newpath = path + "/" + to_string(number) + ".png";
                imwrite(newpath,faceROI);
                cout << "file  save :" << newpath << endl;
                imshow("ROI", faceROI);
                number++;
            }

        }

        if(waitKey() == 27)
            break;

    }
    cout << "you save " << number << " face images" << endl;


}

void generator(string path)
{

    // 输入文件夹名字
    string name;
    cout << "Please key the name:\n";
    getline(cin, name);


    // 新建文件夹
    string newpath = path + "/" + name;
    cout << "image will save the path:" << newpath.c_str() << endl;

    if (!newpath.empty())
    {
        if(NULL != opendir(newpath.c_str()))
        {
            cout << "fin a dir same to your input! delete it now!" << endl;
            rmdir(newpath.c_str());
        }

        if (NULL == opendir(newpath.c_str()))
        {
            mkdir(newpath.c_str(),0777);
        }

    }

    // 检测人脸并扣出来保存到指定位置
    Mat frame;
    detectFace(frame, newpath);

}




int main()
{
    string path = "/home/jason/file";
    generator(path);
    cout << "Hello World!" << endl;
    return 0;
}

四、人脸识别(OpenCV)

 

实现代码：

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <dirent.h>
#include <sys/stat.h>

#include <opencv2/opencv.hpp>
#include <opencv2/face/facerec.hpp>
using namespace cv;
using namespace std;

int label=0;
std::string nam;

vector<Mat> images;
vector<int> labels;
vector<string> names;
void Load_data(const std::string& path)
{

    DIR* dir = opendir(path.c_str());
    if (dir == nullptr)
    {
        std::cerr << "Cannot open directory: " << path << std::endl;
        return;
    }

    struct dirent* entry;
    while ((entry = readdir(dir)) != nullptr)
    {
        std::string name = entry->d_name;
        if (name == "." || name == "..")
        {
            continue;
        }

        std::string fullPath = path + "/" + name;
        struct stat stat_buf;
        if (stat(fullPath.c_str(), &stat_buf) != 0)
        {
            std::cerr << "Cannot stat file: " << fullPath << std::endl;
            return;
        }
        // Determine whether it is a folder
        if (S_ISDIR(stat_buf.st_mode))
        {
            std::cout << "Directory: " << fullPath  <<std::endl;
            Load_data(fullPath);


            // label name
            label+=1;
            nam=name;
        }
        else if (S_ISREG(stat_buf.st_mode))
        {
            std::cout << "File: " << fullPath
                      << "  name:" << nam
                      << "  label: "<<label<<std::endl;

            // 添加到容器中
            Mat img = imread(fullPath);

            images.push_back(img);
            labels.push_back(label);
            names.push_back(nam);
        }
    }
    closedir(dir);

}

void FaceRec(const string &path, string person_name="jack")
{
    // 准备数据
    Load_data(path);

    // 训练
//    Ptr<cv::face::FaceRecognizer> model = cv::face::FisherFaceRecognizer::create();
    Ptr<cv::face::EigenFaceRecognizer> model = cv::face::EigenFaceRecognizer::create();
//    Ptr<cv::face::LBPHFaceRecognizer> model = cv::face::LBPHFaceRecognizer(1, 8,8,8,);
    model->train(images, labels);

    // 创建一个级联分类器 加载一个 .xml 分类器文件. 它既可以是Haar特征也可以是LBP特征的分类器.
    CascadeClassifier casc;
    casc.load("/home/jason/file/opencv/haarcascade_frontalface_alt.xml");

    int jack_label;

    for (size_t i=0; i<names.size(); i++)
    {
        if (names[i] == person_name)
            jack_label=labels[i];
    }

    VideoCapture capture(0);
    Mat frame, gray;
    vector<Rect> faces;

    while (1)
    {
        // 获取图像
        capture >> frame;

        imshow("recognizer", frame);

        // 预处理
        cvtColor(frame,gray,COLOR_RGB2GRAY);
        equalizeHist(gray, gray);

        // 检测人脸
        casc.detectMultiScale(gray, faces);

        // 扣出人脸判断是否为目标人物
        for(size_t i=0; i<faces.size(); i++)
        {
            Mat temp = frame.clone();
            rectangle(temp, faces[i], Scalar(255, 0, 0), 1, 8);

            Mat roi_gray = gray(faces[i]);
            resize(roi_gray, roi_gray, Size(92, 112));
            imshow("person",roi_gray);

            int result;
            double confidence=0.8;

            // 预测
            model->predict(roi_gray, result, confidence);
            cout<< "result: " << result
                <<"confidence: " << confidence<< endl;
            if (result==jack_label)
            {
                putText(temp, person_name ,Point(faces[i].x, faces[i].y),FONT_ITALIC,
                        0.5, Scalar(255, 0, 0),1,8);
            }
            else
            {
                cout << "not  find " << person_name << endl;

            }
            imshow("recognizer", temp);
        }

        if (waitKey(10)==27) break;

    }
}



int main()
{
    std::string path = "/home/jason/file/Face/ORL_Faces";
    FaceRec(path,"jack");
    return 1;
}

参考资料：

C++创建文件夹和写文件_林多的博客-CSDN博客

C++11标准中遍历文件夹方法_c++11 遍历目录_HotPot_Dumplings的博客-CSDN博客

OpenCV: cv::face::FaceRecognizer Class Reference

https://www.cnblogs.com/zyly/p/9498857.html