1 //use the camera to aid the decision to sleep.
2 //Copyright 2009 Daniel Watson
4 This file is part of oble.
6 oble is free software: you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation, either version 3 of the License, or
9 (at your option) any later version.
11 oble is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with oble. If not, see <http://www.gnu.org/licenses/>.
24 CvHaarClassifierCascade
* load_object_detector(const char* cascade_path
)
26 return (CvHaarClassifierCascade
*)cvLoad(cascade_path
);
29 void detect_and_draw_objects(IplImage
* image
,
30 CvHaarClassifierCascade
* cascade
,
33 IplImage
* small_image
= image
;
34 CvMemStorage
* storage
= cvCreateMemStorage(0);
38 /* if the flag is specified, down-scale the input image to get a
39 performance boost w/o loosing quality (perhaps) */
42 small_image
= cvCreateImage(cvSize(image
->width
/2,image
->height
/2),
44 cvPyrDown(image
, small_image
, CV_GAUSSIAN_5x5
);
48 /* use the fastest variant */
49 faces
= cvHaarDetectObjects(small_image
, cascade
, storage
, 1.2, 2,
50 CV_HAAR_DO_CANNY_PRUNING
);
52 /* draw all the rectangles */
53 for(i
= 0; i
< faces
->total
; i
++)
55 /* extract the rectangles only */
56 CvRect face
= *(CvRect
*)cvGetSeqElem(faces
, i
);
57 CvPoint upperLeft
= cvPoint(face
.x
*scale
,face
.y
*scale
);
58 CvPoint bottomRight
= cvPoint((face
.x
+face
.width
)*scale
,
59 (face
.y
+face
.height
)*scale
);
60 cvRectangle(image
, upperLeft
, bottomRight
, CV_RGB(255,0,0), 3);
63 if(small_image
!= image
)
64 cvReleaseImage(&small_image
);
65 cvReleaseMemStorage(&storage
);
68 // A Simple Camera Capture Framework.
69 int main(int argc
, char** argv
)
71 CvHaarClassifierCascade
* cascade
= load_object_detector(argv
[1]);
73 CvCapture
* capture
= cvCaptureFromCAM(CV_CAP_ANY
);
75 fprintf(stderr
, "ERROR: capture is NULL \n");
80 // Create a window in which the captured images will be presented
81 cvNamedWindow("mywindow", CV_WINDOW_AUTOSIZE
);
83 // Show the image captured from the camera in the window and repeat
86 IplImage
* frame
= cvQueryFrame(capture
);
88 fprintf(stderr
, "ERROR: frame is null...\n");
93 //flip the image so displayed right/left corresponds to physical right/left
94 cvFlip(frame
, frame
, 1);
96 detect_and_draw_objects(frame
, cascade
, 1);
97 cvShowImage("mywindow", frame
);
98 // Do not release the frame!
100 //If ESC key pressed, Key=0x10001B under OpenCV 0.9.7(linux version),
101 //remove higher bits using AND operator
102 if((cvWaitKey(10) & 255) == 27) break;
105 // Release the capture device housekeeping
106 cvReleaseCapture(&capture
);
107 cvDestroyWindow("mywindow");