基于harr的人脸检测

来源:互联网 发布:python实现cnn算法 编辑:程序博客网 时间:2024/05/21 18:07
这是一个超级简单的人脸检测程序,关键代码才两句
头文件:
#pragma once
#include "cv.h"
#include "highgui.h"

class MFace
{
public:
    MFace(void);
    ~MFace(void);
    CvHaarClassifierCascade* Loadhaar(CString path);
    int FindAndDraw(CString image,CvHaarClassifierCascade* cascade,int pyrdown);
};
源文件:
#include "StdAfx.h"
#include "MFace.h"
#include "opencv2/objdetect/objdetect.hpp"
#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
using namespace std;
using namespace cv;

MFace::MFace(void)
{
    CvHaarClassifierCascade* cascade = Loadhaar("haarcascade_frontalface_alt.xml");
    FindAndDraw("D:\\tmp\\physical.jpg",cascade,0);
}

MFace::~MFace(void)
{
}
CvHaarClassifierCascade* MFace::Loadhaar(CString path){
    return (CvHaarClassifierCascade*)cvLoad(path);
}

int MFace::FindAndDraw(CString image,CvHaarClassifierCascade* cascade,int pyrdown){
    CvMemStorage* mem= cvCreateMemStorage(0);
    CvSeq* seq = NULL;

    //加载检测图片
    IplImage* mainImage = cvLoadImage(image,0);
    //如果金子塔变换加载小图片
    IplImage* smallImage;
    if(pyrdown > 0){
        smallImage = cvCreateImage(cvSize(mainImage->width/2,mainImage->height/2),mainImage->depth,mainImage->nChannels);
        cvPyrDown(mainImage,smallImage,CV_GAUSSIAN_5x5);
    }
    else{
        smallImage = cvCreateImage(cvGetSize(mainImage),mainImage->depth,mainImage->nChannels);
        cvCopy(mainImage,smallImage);
    }
     //绘制找到的地方
    seq = cvHaarDetectObjects(smallImage,cascade,mem,1.2,2,CV_HAAR_DO_CANNY_PRUNING);
    for (int i = 0; i < seq->total;i++){
        CvRect center = *(CvRect*)cvGetSeqElem(seq,i);
        cvDrawRect(mainImage,cvPoint(center.x,center.y),cvPoint(center.x+center.width,center.y + center.height),cvScalarAll(255),2);
    }
    //保存图像
    cvSaveImage("D:\\result.bmp",mainImage);

    return 0;
}
0 0
原创粉丝点击