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=\"<cascade_path>\" [filename|camera_index]\n" );
+ "Usage: facedetect [--cascade=\"<cascade_path>\"]\n"
+ " [--nested-cascade[=\"nested_cascade_path\"]]\n"
+ " [--scale[=<image 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 );
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");
{{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 );
{
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 );
+ }
}
}