图像处理项目——制作自己的人脸数据集

制作自己的人脸数据集

*开发环境为visual studio2010
*使用的是opencv中的Haart特征分类器，harr Cascades
*检测对象为人脸照片 

一：主要步骤

1、加载人脸检测器
2、打开电脑摄像头，载入视频流，抽取帧，进行灰度化处理
3、识别人脸，采集20张保存到文件中

二 ：代码

/****************************************************
制作自己的人脸数据集:
1、加载人脸检测器
2、打开电脑摄像头，载入视频流，抽取帧，进行灰度化处理
3、识别人脸，采集20张保存到文件中
*****************************************************/
#include "opencv2/objdetect.hpp"
#include "opencv2/videoio.hpp"
#include "opencv2/highgui.hpp"
#include "opencv2/imgproc.hpp"
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
using namespace std;
using namespace cv;

int main()
{ 
   Mat img;                      //用于存储每一帧的图像
   Mat imgGray;                  //用于存储灰度图
   Mat myFace;                   //用于存储我的人脸数据集
   int count=1;                //图片计数 
   CascadeClassifier cascades;   //定义人脸分类器,opencv中的harr Cascades
   cascades.load("models/lbpcascade_frontalface.xml"); //加载人脸检测器
   VideoCapture cap;
   cap.open(0);                   //打开摄像头
   //cap.open("E：//demo.avi");   //打开视频 
   std::vector<Rect> faces;       //定义容器存储人脸数据
   //循环20张人脸保存
  while(1)
   {
     cap >> img;                   //从视频中读取当前帧        
     cvtColor(img,imgGray,COLOR_BGR2GRAY); //灰度化
     //检测出图片中的人脸，用vector保存各个人脸的坐标、大小（用矩形表示）到faces
     cascades.detectMultiScale(imgGray,faces, 1.1, 5,0);
     for(size_t i = 0; i < faces.size(); i++){
        rectangle(img, faces[i], Scalar(0, 255, 0), 3, 8, 0);
     }
     //如果检测到人脸
    if(faces.size() == 1) 
     {     
         //如果图片小于等于20张
        if(count<=20)
         {
             Mat faceROI = imgGray(faces[0]);    
             resize(faceROI, myFace,Size(92,112));//opencv中resize()函数将识别到的人脸图像自动改变尺寸为92x112
                putText(img, "face_zsc", faces[0].tl(), FONT_HERSHEY_DUPLEX, 1.2, Scalar(71, 99, 255), 2, LINE_8);
             string filename = format("faces/myface/%d.jpg", count);      //保存到此文件
             imwrite(filename, myFace);      //通过imwrite函数将resize()过后的图片保存到背景数据
             imshow(filename,myFace);   
             waitKey(100);
             destroyWindow(filename);             
             count++;
         } 
     }
    imshow("frame",img);
    waitKey(100);   
    
   }
   return 0;
   system("pause");
}

三 ： 制作结果