else {
cascade_name = \ input_name = argc > 1 ? argv[1]:0; }
cascade = (CvHaarClassifierCascade*)cvLoad( cascade_name, 0, 0, 0 );
if( !cascade ) {
fprintf( stderr, \ fprintf( stderr,
\ return -1; }
storage = cvCreateMemStorage(0);
if( !input_name || (isdigit(input_name[0]) && input_name[1] == '\\0') )
capture = cvCaptureFromCAM( !input_name ? 0 : input_name[0] - '0' ); else
capture = cvCaptureFromAVI( input_name );
cvNamedWindow( \
if( capture ) {
for(;;) {
if( !cvGrabFrame( capture )) break;
frame = cvRetrieveFrame( capture ); if( !frame ) break;
if( !frame_copy )
frame_copy = cvCreateImage( cvSize(frame->width,frame->height),
IPL_DEPTH_8U, frame->nChannels ); if( frame->origin == IPL_ORIGIN_TL ) cvCopy( frame, frame_copy, 0 ); else
cvFlip( frame, frame_copy, 0 );
detect_and_draw( frame_copy );
18
if( cvWaitKey( 10 ) >= 0 ) break; }
cvReleaseImage( &frame_copy ); cvReleaseCapture( &capture ); } else {
const char* filename = input_name ? input_name : (char*)\ IplImage* image = cvLoadImage( filename, 1 );
if( image ) {
detect_and_draw( image ); cvWaitKey(0);
cvReleaseImage( &image ); } else {
/* assume it is a text file containing the
list of the image filenames to be processed - one per line */ FILE* f = fopen( filename, \ if( f ) {
char buf[1000+1];
while( fgets( buf, 1000, f ) ) {
int len = (int)strlen(buf);
while( len > 0 && isspace(buf[len-1]) ) len--; buf[len] = '\\0';
image = cvLoadImage( buf, 1 ); if( image ) {
detect_and_draw( image ); cvWaitKey(0);
cvReleaseImage( &image ); } }
fclose(f); } }
19
}
cvDestroyWindow(\
return 0; }
void detect_and_draw( IplImage* img ) {
static CvScalar colors[] = {
{{0,0,255}}, {{0,128,255}}, {{0,255,255}}, {{0,255,0}}, {{255,128,0}}, {{255,255,0}}, {{255,0,0}}, {{255,0,255}} };
double scale = 1.3;
IplImage* gray = cvCreateImage( cvSize(img->width,img->height), 8, 1 ); IplImage* small_img = cvCreateImage( cvSize( cvRound (img->width/scale), cvRound (img->height/scale)), 8, 1 ); int i;
cvCvtColor( img, gray, CV_BGR2GRAY );
cvResize( gray, small_img, CV_INTER_LINEAR ); cvEqualizeHist( small_img, small_img ); cvClearMemStorage( storage );
if( cascade ) {
double t = (double)cvGetTickCount();
CvSeq* faces = cvHaarDetectObjects( small_img, cascade, storage, 1.1, 0/*CV_HAAR_DO_CANNY_PRUNING*/,
cvSize(30, 30) ); t = (double)cvGetTickCount() - t;
printf( \
20
2,
for( i = 0; i < (faces ? faces->total : 0); i++ ) {
CvRect* r = (CvRect*)cvGetSeqElem( faces, i ); CvPoint center; int radius;
center.x = cvRound((r->x + r->width*0.5)*scale); center.y = cvRound((r->y + r->height*0.5)*scale); radius = cvRound((r->width + r->height)*0.25*scale); cvCircle( img, center, radius, colors[i%8], 3, 8, 0 ); } }
cvShowImage( \ cvReleaseImage( &gray );
cvReleaseImage( &small_img ); }
21