1 #include "opencv2/objdetect.hpp"
2 #include "opencv2/videoio.hpp"
3 #include "opencv2/highgui.hpp"
4 #include "opencv2/imgproc.hpp"
5
6 #include <iostream>
7 #include <stdio.h>
8
9 using namespace std;
10 using namespace cv;
11
12 /** Function Headers */
13 void detectAndDisplay( Mat frame );
14
15 /** Global variables */
16 String face_cascade_name = "haarcascade_frontalface_alt.xml";
17 String eyes_cascade_name = "haarcascade_eye_tree_eyeglasses.xml";
18 CascadeClassifier face_cascade;
19 CascadeClassifier eyes_cascade;
20 String window_name = "Capture - Face detection";
21
22 /** @function main */
main(void)23 int main( void )
24 {
25 VideoCapture capture;
26 Mat frame;
27
28 //-- 1. Load the cascades
29 if( !face_cascade.load( face_cascade_name ) ){ printf("--(!)Error loading face cascade\n"); return -1; };
30 if( !eyes_cascade.load( eyes_cascade_name ) ){ printf("--(!)Error loading eyes cascade\n"); return -1; };
31
32 //-- 2. Read the video stream
33 capture.open( -1 );
34 if ( ! capture.isOpened() ) { printf("--(!)Error opening video capture\n"); return -1; }
35
36 while ( capture.read(frame) )
37 {
38 if( frame.empty() )
39 {
40 printf(" --(!) No captured frame -- Break!");
41 break;
42 }
43
44 //-- 3. Apply the classifier to the frame
45 detectAndDisplay( frame );
46
47 int c = waitKey(10);
48 if( (char)c == 27 ) { break; } // escape
49 }
50 return 0;
51 }
52
53 /** @function detectAndDisplay */
detectAndDisplay(Mat frame)54 void detectAndDisplay( Mat frame )
55 {
56 std::vector<Rect> faces;
57 Mat frame_gray;
58
59 cvtColor( frame, frame_gray, COLOR_BGR2GRAY );
60 equalizeHist( frame_gray, frame_gray );
61
62 //-- Detect faces
63 face_cascade.detectMultiScale( frame_gray, faces, 1.1, 2, 0|CASCADE_SCALE_IMAGE, Size(30, 30) );
64
65 for ( size_t i = 0; i < faces.size(); i++ )
66 {
67 Point center( faces[i].x + faces[i].width/2, faces[i].y + faces[i].height/2 );
68 ellipse( frame, center, Size( faces[i].width/2, faces[i].height/2 ), 0, 0, 360, Scalar( 255, 0, 255 ), 4, 8, 0 );
69
70 Mat faceROI = frame_gray( faces[i] );
71 std::vector<Rect> eyes;
72
73 //-- In each face, detect eyes
74 eyes_cascade.detectMultiScale( faceROI, eyes, 1.1, 2, 0 |CASCADE_SCALE_IMAGE, Size(30, 30) );
75
76 for ( size_t j = 0; j < eyes.size(); j++ )
77 {
78 Point eye_center( faces[i].x + eyes[j].x + eyes[j].width/2, faces[i].y + eyes[j].y + eyes[j].height/2 );
79 int radius = cvRound( (eyes[j].width + eyes[j].height)*0.25 );
80 circle( frame, eye_center, radius, Scalar( 255, 0, 0 ), 4, 8, 0 );
81 }
82 }
83 //-- Show what you got
84 imshow( window_name, frame );
85 }
86