图片人脸检测 (I)【sample改版 人眼检测】

1. / face_detect.cpp : 定义控制台应用程序的入口点。  
2. //  
3.   
4. //#include "stdafx.h"  
5.   
6. #include "opencv2/objdetect/objdetect.hpp"  
7. #include "opencv2/highgui/highgui.hpp"  
8. #include "opencv2/imgproc/imgproc.hpp"  
9. #include "opencv2/ml/ml.hpp"  
10.   
11. #include <iostream>  
12. #include <stdio.h>  
13.   
14. using namespace std;  
15. using namespace cv;  
16.   
17.   
18. String cascadeName = "./haarcascade_frontalface_alt2.xml";//人脸的训练数据  
19. //String nestedCascadeName = "./haarcascade_eye_tree_eyeglasses.xml";//人眼的训练数据  
20. String nestedCascadeName = "./haarcascade_eye.xml";//人眼的训练数据  
21.   
22.   
23.   
24.   
25. void detectAndDraw( Mat& img,  
26.     CascadeClassifier& cascade, CascadeClassifier& nestedCascade,  
27.     double scale)  
28. {  
29.     int i = 0;  
30.     double t = 0;  
31.     vector<Rect> faces;  
32.     const static Scalar colors[] =  { CV_RGB(0,0,255),  
33.         CV_RGB(0,128,255),  
34.         CV_RGB(0,255,255),  
35.         CV_RGB(0,255,0),  
36.         CV_RGB(255,128,0),  
37.         CV_RGB(255,255,0),  
38.         CV_RGB(255,0,0),  
39.         CV_RGB(255,0,255)} ;//用不同的颜色表示不同的人脸  
40.   
41.     Mat gray, smallImg( cvRound (img.rows/scale), cvRound(img.cols/scale), CV_8UC1 );//将图片缩小，加快检测速度  
42.   
43.     cvtColor( img, gray, CV_BGR2GRAY );//因为用的是类haar特征，所以都是基于灰度图像的，这里要转换成灰度图像  
44.     resize( gray, smallImg, smallImg.size(), 0, 0, INTER_LINEAR );//将尺寸缩小到1/scale,用线性插值  
45.     equalizeHist( smallImg, smallImg );//直方图均衡  
46.   
47.     t = (double)cvGetTickCount();//用来计算算法执行时间  
48.   
49.   
50.     //检测人脸  
51.     //detectMultiScale函数中smallImg表示的是要检测的输入图像为smallImg，faces表示检测到的人脸目标序列，1.1表示  
52.     //每次图像尺寸减小的比例为1.1，2表示每一个目标至少要被检测到3次才算是真的目标(因为周围的像素和不同的窗口大  
53.     //小都可以检测到人脸),CV_HAAR_SCALE_IMAGE表示不是缩放分类器来检测，而是缩放图像，Size(30, 30)为目标的  
54.     //最小最大尺寸  
55.     cascade.detectMultiScale( smallImg, faces,  
56.         1.1, 2, 0  
57.         //|CV_HAAR_FIND_BIGGEST_OBJECT  
58.         //|CV_HAAR_DO_ROUGH_SEARCH  
59.         |CV_HAAR_SCALE_IMAGE  
60.         ,  
61.         Size(30, 30) );  
62.   
63.     t = (double)cvGetTickCount() - t;//相减为算法执行的时间  
64.     printf( "detection time = %g ms\n", t/((double)cvGetTickFrequency()*1000.) );  
65.     for( vector<Rect>::const_iterator r = faces.begin(); r != faces.end(); r++, i++ )  
66.     {  
67.         Mat smallImgROI;  
68.         vector<Rect> nestedObjects;  
69.         Point center;  
70.         Scalar color = colors[i%8];  
71.         int radius;  
72.         center.x = cvRound((r->x + r->width*0.5)*scale);//还原成原来的大小  
73.         center.y = cvRound((r->y + r->height*0.5)*scale);  
74.         radius = cvRound((r->width + r->height)*0.25*scale);  
75.         circle( img, center, radius, color, 3, 8, 0 );  
76.   
77.         //检测人眼，在每幅人脸图上画出人眼  
78.         if( nestedCascade.empty() )  
79.             continue;  
80.         smallImgROI = smallImg(*r);  
81.   
82.         //和上面的函数功能一样  
83.         nestedCascade.detectMultiScale( smallImgROI, nestedObjects,  
84.             1.1, 2, 0  
85.             //|CV_HAAR_FIND_BIGGEST_OBJECT  
86.             //|CV_HAAR_DO_ROUGH_SEARCH  
87.             //|CV_HAAR_DO_CANNY_PRUNING  
88.             |CV_HAAR_SCALE_IMAGE  
89.             ,  
90.             Size(30, 30) );  
91.         for( vector<Rect>::const_iterator nr = nestedObjects.begin(); nr != nestedObjects.end(); nr++ )  
92.         {  
93.             center.x = cvRound((r->x + nr->x + nr->width*0.5)*scale);  
94.             center.y = cvRound((r->y + nr->y + nr->height*0.5)*scale);  
95.             radius = cvRound((nr->width + nr->height)*0.25*scale);  
96.             circle( img, center, radius, color, 3, 8, 0 );//将眼睛也画出来，和对应人脸的图形是一样的  
97.         }  
98.     }  
99.     cv::imshow( "result", img );  
100. }  
101.   
102.   
103.   
104.   
105. int main( int argc, const char** argv )  
106. {  
107.     Mat image;  
108.     CascadeClassifier cascade, nestedCascade;//创建级联分类器对象  
109.     double scale = 1.3;  
110.   
111.     //image = imread( "lena.jpg", 1 );//读入lena图片  
112.     image = imread("0055.jpg",1);  
113.     namedWindow( "result", 1 );//opencv2.0以后用namedWindow函数会自动销毁窗口  
114.   
115.     if( !cascade.load( cascadeName ) )//从指定的文件目录中加载级联分类器  
116.     {  
117.         cerr << "ERROR: Could not load classifier cascade" << endl;  
118.         return 0;  
119.     }  
120.   
121.     if( !nestedCascade.load( nestedCascadeName ) )  
122.     {  
123.         cerr << "WARNING: Could not load classifier cascade for nested objects" << endl;  
124.         return 0;  
125.     }  
126.   
127.     if( !image.empty() )//读取图片数据不能为空  
128.     {  
129.         detectAndDraw( image, cascade, nestedCascade, scale );  
130.         waitKey(0);  
131.     }  
132.   
133.     return 0;  
134. }