X-Git-Url: https://vcs.maemo.org/git/?a=blobdiff_plain;f=samples%2Fc%2Ffacedetect.c;fp=samples%2Fc%2Ffacedetect.c;h=ec75505e827347a3a3a9a026ec52ce24a1811f13;hb=80cd7b93506cc1926882d5fd08a2c74ee9359e29;hp=f2606e8a30daed1038ef2a9bb74fe918993ad228;hpb=467a270adf12425827305759c0c4ea8f5b2b3854;p=opencv diff --git a/samples/c/facedetect.c b/samples/c/facedetect.c index f2606e8..ec75505 100644 --- a/samples/c/facedetect.c +++ b/samples/c/facedetect.c @@ -17,46 +17,82 @@ static CvMemStorage* storage = 0; static CvHaarClassifierCascade* cascade = 0; +static CvHaarClassifierCascade* nested_cascade = 0; +int use_nested_cascade = 0; void detect_and_draw( IplImage* image ); const char* cascade_name = - "haarcascade_frontalface_alt.xml"; + "../../data/haarcascades/haarcascade_frontalface_alt.xml"; /* "haarcascade_profileface.xml";*/ +const char* nested_cascade_name = + "../../data/haarcascades/haarcascade_eye_tree_eyeglasses.xml"; +// "../../data/haarcascades/haarcascade_eye.xml"; +double scale = 1; int main( int argc, char** argv ) { CvCapture* capture = 0; IplImage *frame, *frame_copy = 0; - int optlen = strlen("--cascade="); - const char* input_name; + IplImage *image = 0; + const char* scale_opt = "--scale="; + int scale_opt_len = (int)strlen(scale_opt); + const char* cascade_opt = "--cascade="; + int cascade_opt_len = (int)strlen(cascade_opt); + const char* nested_cascade_opt = "--nested-cascade"; + int nested_cascade_opt_len = (int)strlen(nested_cascade_opt); + int i; + const char* input_name = 0; - if( argc > 1 && strncmp( argv[1], "--cascade=", optlen ) == 0 ) - { - cascade_name = argv[1] + optlen; - input_name = argc > 2 ? argv[2] : 0; - } - else + for( i = 1; i < argc; i++ ) { - cascade_name = "../../data/haarcascades/haarcascade_frontalface_alt2.xml"; - input_name = argc > 1 ? argv[1] : 0; + if( strncmp( argv[i], cascade_opt, cascade_opt_len) == 0 ) + cascade_name = argv[i] + cascade_opt_len; + else if( strncmp( argv[i], nested_cascade_opt, nested_cascade_opt_len ) == 0 ) + { + if( argv[i][nested_cascade_opt_len] == '=' ) + nested_cascade_name = argv[i] + nested_cascade_opt_len + 1; + nested_cascade = (CvHaarClassifierCascade*)cvLoad( nested_cascade_name, 0, 0, 0 ); + if( !nested_cascade ) + fprintf( stderr, "WARNING: Could not load classifier cascade for nested objects\n" ); + } + else if( strncmp( argv[i], scale_opt, scale_opt_len ) == 0 ) + { + if( !sscanf( argv[i] + scale_opt_len, "%lf", &scale ) || scale < 1 ) + scale = 1; + } + else if( argv[i][0] == '-' ) + { + fprintf( stderr, "WARNING: Unknown option %s\n", argv[i] ); + } + else + input_name = argv[i]; } cascade = (CvHaarClassifierCascade*)cvLoad( cascade_name, 0, 0, 0 ); - + if( !cascade ) { fprintf( stderr, "ERROR: Could not load classifier cascade\n" ); fprintf( stderr, - "Usage: facedetect --cascade=\"\" [filename|camera_index]\n" ); + "Usage: facedetect [--cascade=\"\"]\n" + " [--nested-cascade[=\"nested_cascade_path\"]]\n" + " [--scale[=\n" + " [filename|camera_index]\n" ); 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 if( input_name ) + { + image = cvLoadImage( input_name, 1 ); + if( !image ) + capture = cvCaptureFromAVI( input_name ); + } else - capture = cvCaptureFromAVI( input_name ); + image = cvLoadImage( "lena.jpg", 1 ); cvNamedWindow( "result", 1 ); @@ -80,49 +116,50 @@ int main( int argc, char** argv ) detect_and_draw( frame_copy ); if( cvWaitKey( 10 ) >= 0 ) - break; + goto _cleanup_; } + cvWaitKey(0); +_cleanup_: cvReleaseImage( &frame_copy ); cvReleaseCapture( &capture ); } else { - const char* filename = input_name ? input_name : (char*)"lena.jpg"; - IplImage* image = cvLoadImage( filename, 1 ); - if( image ) { detect_and_draw( image ); cvWaitKey(0); cvReleaseImage( &image ); } - else + else if( input_name ) { /* assume it is a text file containing the list of the image filenames to be processed - one per line */ - FILE* f = fopen( filename, "rt" ); + FILE* f = fopen( input_name, "rt" ); if( f ) { char buf[1000+1]; while( fgets( buf, 1000, f ) ) { - int len = (int)strlen(buf); + int len = (int)strlen(buf), c; while( len > 0 && isspace(buf[len-1]) ) len--; buf[len] = '\0'; + printf( "file %s\n", buf ); image = cvLoadImage( buf, 1 ); if( image ) { detect_and_draw( image ); - cvWaitKey(0); + c = cvWaitKey(0); + if( c == 27 || c == 'q' || c == 'Q' ) + break; cvReleaseImage( &image ); } } fclose(f); } } - } cvDestroyWindow("result"); @@ -144,12 +181,12 @@ void detect_and_draw( IplImage* img ) {{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; + IplImage *gray, *small_img; + int i, j; + + gray = cvCreateImage( cvSize(img->width,img->height), 8, 1 ); + small_img = cvCreateImage( cvSize( cvRound (img->width/scale), + cvRound (img->height/scale)), 8, 1 ); cvCvtColor( img, gray, CV_BGR2GRAY ); cvResize( gray, small_img, CV_INTER_LINEAR ); @@ -160,19 +197,46 @@ void detect_and_draw( IplImage* img ) { double t = (double)cvGetTickCount(); CvSeq* faces = cvHaarDetectObjects( small_img, cascade, storage, - 1.1, 2, 0/*CV_HAAR_DO_CANNY_PRUNING*/, + 1.1, 2, 0 + //|CV_HAAR_FIND_BIGGEST_OBJECT + //|CV_HAAR_DO_ROUGH_SEARCH + |CV_HAAR_DO_CANNY_PRUNING + //|CV_HAAR_SCALE_IMAGE + , cvSize(30, 30) ); t = (double)cvGetTickCount() - t; printf( "detection time = %gms\n", t/((double)cvGetTickFrequency()*1000.) ); for( i = 0; i < (faces ? faces->total : 0); i++ ) { CvRect* r = (CvRect*)cvGetSeqElem( faces, i ); + CvMat small_img_roi; + CvSeq* nested_objects; CvPoint center; + CvScalar color = colors[i%8]; 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 ); + cvCircle( img, center, radius, color, 3, 8, 0 ); + if( !nested_cascade ) + continue; + cvGetSubRect( small_img, &small_img_roi, *r ); + nested_objects = cvHaarDetectObjects( &small_img_roi, nested_cascade, storage, + 1.1, 2, 0 + //|CV_HAAR_FIND_BIGGEST_OBJECT + //|CV_HAAR_DO_ROUGH_SEARCH + //|CV_HAAR_DO_CANNY_PRUNING + //|CV_HAAR_SCALE_IMAGE + , + cvSize(0, 0) ); + for( j = 0; j < (nested_objects ? nested_objects->total : 0); j++ ) + { + CvRect* nr = (CvRect*)cvGetSeqElem( nested_objects, j ); + center.x = cvRound((r->x + nr->x + nr->width*0.5)*scale); + center.y = cvRound((r->y + nr->y + nr->height*0.5)*scale); + radius = cvRound((nr->width + nr->height)*0.25*scale); + cvCircle( img, center, radius, color, 3, 8, 0 ); + } } }