00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024 #include <classifiers/faces.h>
00025
00026 #include <core/exception.h>
00027 #include <core/exceptions/software.h>
00028 #include <fvutils/color/colorspaces.h>
00029 #include <fvutils/color/conversions.h>
00030 #include <fvutils/adapters/iplimage.h>
00031
00032 #include <opencv/cv.h>
00033
00034 namespace firevision {
00035 #if 0
00036 }
00037 #endif
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063 FacesClassifier::FacesClassifier(const char *haarcascade_file,
00064 unsigned int pixel_width, unsigned int pixel_height,
00065 IplImage *image,
00066 float haar_scale_factor, int min_neighbours, int flags)
00067 : Classifier("FacesClassifier")
00068 {
00069 __haar_scale_factor = haar_scale_factor;
00070 __min_neighbours = min_neighbours;
00071 __flags = flags;
00072
00073 __cascade = (CvHaarClassifierCascade *) cvLoad(haarcascade_file);
00074 if ( ! __cascade ) {
00075 throw fawkes::Exception("Could not load Haar casca via OpenCV");
00076 }
00077
00078 __storage = cvCreateMemStorage(0);
00079 if ( ! __storage ) {
00080 cvReleaseHaarClassifierCascade(&__cascade);
00081 throw fawkes::Exception("Could not initialize OpenCV memory");
00082 }
00083
00084 if ( image ) {
00085 __image = image;
00086 __own_image = false;
00087 } else {
00088 __image = cvCreateImage(cvSize(pixel_width, pixel_height), IPL_DEPTH_8U, 3);
00089 __own_image = true;
00090 }
00091 }
00092
00093
00094
00095 FacesClassifier::~FacesClassifier()
00096 {
00097 cvReleaseHaarClassifierCascade(&__cascade);
00098 cvReleaseMemStorage(&__storage);
00099 if ( __own_image ) {
00100 cvReleaseImage(&__image);
00101 }
00102 }
00103
00104
00105 std::list< ROI > *
00106 FacesClassifier::classify()
00107 {
00108 std::list< ROI > *rv = new std::list< ROI >();
00109
00110 if ( __own_image ) {
00111 IplImageAdapter::convert_image_bgr(_src, __image);
00112 }
00113
00114 CvSeq *face_seq = cvHaarDetectObjects(__image, __cascade, __storage,
00115 __haar_scale_factor, __min_neighbours, __flags);
00116
00117 for ( int i = 0; i < face_seq->total; ++i) {
00118 CvAvgComp el = *(CvAvgComp*)cvGetSeqElem(face_seq, i);
00119 ROI r(el.rect.x, el.rect.y, el.rect.width, el.rect.height, _width, _height);
00120 r.num_hint_points = el.rect.width * el.rect.height;
00121 rv->push_back(r);
00122 }
00123
00124
00125 rv->sort();
00126
00127 return rv;
00128 }
00129
00130 }