samples/cpp/facedetect.cpp

此程序演示如何使用级联分类器类

示例屏幕截图

#include "opencv2/objdetect.hpp"
#include "opencv2/highgui.hpp"
#include "opencv2/imgproc.hpp"
#include "opencv2/videoio.hpp"
#include <iostream>
 
using namespace std;
using namespace cv;
 
static void help(const char** argv)
{
cout << "\n此程序演示使用 cv::CascadeClassifier 类检测物体（面部 + 眼睛）。你可以使用 Haar 或 LBP 特征。\n"
            "此分类器可以识别很多种刚性物体，一旦训练了合适的分类器。\n"
            "它最知名的用途是人脸。\n"
            "用法：\n"
<< argv[0]
<< " [--cascade=<cascade_path> 这是主要训练分类器，例如正脸]\n"
            " [--nested-cascade[=nested_cascade_path 这一个可选的辅助分类器，例如眼睛]]\n"
            " [--scale=<图像缩放大于或等于 1，例如尝试 1.3>]\n"
            " [--try-flip]\n"
            " [文件名|摄像头索引]\n\n"
            "示例：\n"
<< argv[0]
<< " --cascade=\"data/haarcascades/haarcascade_frontalface_alt.xml\" --nested-cascade=\"data/haarcascades/haarcascade_eye_tree_eyeglasses.xml\" --scale=1.3\n\n"
            "执行期间：\n\t按任意键退出。\n"
            "\t正在使用 OpenCV 版本 " << CV_VERSION << "\n" << endl;
}
 
void detectAndDraw( Mat& img, CascadeClassifier& cascade,
                    CascadeClassifier& nestedCascade,
                    double scale, bool tryflip );
 
string cascadeName;
string nestedCascadeName;
 
int main( int argc, const char** argv )
{
    VideoCapture capture;
    Mat frame, image;
    string inputName;
    bool tryflip;
    CascadeClassifier cascade, nestedCascade;
    double scale;
 
    cv::CommandLineParser parser(argc, argv,
        "{help h||}"
        "{cascade|data/haarcascades/haarcascade_frontalface_alt.xml|}"
        "{nested-cascade|data/haarcascades/haarcascade_eye_tree_eyeglasses.xml|}"
        "{scale|1|}{try-flip||}{@filename||}"
    );
    if (parser.has("help"))
    {
help(argv);
        return 0;
    }
cascadeName = parser.get<string>("cascade");
nestedCascadeName = parser.get<string>("nested-cascade");
    scale = parser.get<double>("scale");
    if (scale < 1)
        scale = 1;
tryflip = parser.has("try-flip");
inputName = parser.get<string>("@filename");
    if (!parser.check())
    {
parser.printErrors();
        return 0;
    }
    if (!nestedCascade.load(samples::findFileOrKeep(nestedCascadeName)))
cerr << "WARNING: Could not load classifier cascade for nested objects" << endl;
    if (!cascade.load(samples::findFile(cascadeName)))
    {
cerr << "ERROR: Could not load classifier cascade" << endl;
help(argv);
        return -1;
    }
    if( inputName.empty() || (isdigit(inputName[0]) && inputName.size() == 1) )
    {
        int camera = inputName.empty() ? 0 : inputName[0] - '0';
        if(!capture.open(camera))
        {
cout << "Capture from camera #" << camera << " didn't work" << endl;
            return 1;
        }
    }
    else if (!inputName.empty())
    {
image = imread(samples::findFileOrKeep(inputName), IMREAD_COLOR);
        if (image.empty())
        {
            if (!capture.open(samples::findFileOrKeep(inputName)))
            {
cout << "Could not read " << inputName << endl;
                return 1;
            }
        }
    }
    else
    {
image = imread(samples::findFile("lena.jpg"), IMREAD_COLOR);
        if (image.empty())
        {
cout << "Couldn't read lena.jpg" << endl;
            return 1;
        }
    }
 
    if( capture.isOpened() )
    {
cout << "视频捕获已启动 ..." << endl;
 
        for(;;)
        {
capture >> frame;
            if( frame.empty() )
                break;
 
            Mat frame1 = frame.clone();
detectAndDraw( frame1, cascade, nestedCascade, scale, tryflip );
 
            char c = (char)waitKey(10);
            if( c == 27 || c == 'q' || c == 'Q' )
                break;
        }
    }
    else
    {
cout << "检测视频 " << inputName << endl;
        if( !image.empty() )
        {
detectAndDraw( image, cascade, nestedCascade, scale, tryflip );
            waitKey(0);
        }
        else if( !inputName.empty() )
        {
            /* 假设这是一个文本文件，其中包含
要处理的图像文件名的列表 - 每行一个 */
FILE* f = fopen( inputName.c_str(), "rt" );
            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';
cout << "文件 " << buf << endl;
image = imread( buf, IMREAD_COLOR );
                    if( !image.empty() )
                    {
detectAndDraw( image, cascade, nestedCascade, scale, tryflip );
                        char c = (char)waitKey(0);
                        if( c == 27 || c == 'q' || c == 'Q' )
                            break;
                    }
                    else
                    {
cerr << "啊呀, 无法读取图像 " << buf << endl;
                    }
                }
fclose(f);
            }
        }
    }
 
    return 0;
}
 
void detectAndDraw( Mat& img, CascadeClassifier& cascade,
                    CascadeClassifier& nestedCascade,
                    double scale, bool tryflip )
{
    double t = 0;
vector<Rect> faces, faces2;
    const static Scalar colors[] =
    {
        Scalar(255,0,0),
        Scalar(255,128,0),
        Scalar(255,255,0),
        Scalar(0,255,0),
        Scalar(0,128,255),
        Scalar(0,255,255),
        Scalar(0,0,255),
        Scalar(255,0,255)
    };
    Mat gray, smallImg;
 
    cvtColor( img, gray, COLOR_BGR2GRAY );
    double fx = 1 / scale;
    缩放(gray, smallImg, 尺寸(), fx, fx, INTER_LINEAR_EXACT );
    直方图均衡化(smallImg, smallImg );
 
t = (double)获得滴答数();
层叠.检测多尺度(smallImg, 面,
        1.1, 2, 0
        //|CASCADE_FIND_BIGGEST_OBJECT
        //|CASCADE_DO_ROUGH_SEARCH
|CASCADE_SCALE_IMAGE,
        尺寸(30, 30) );
    如果(尝试翻转 )
    {
        翻转(smallImg, smallImg, 1);
级联.检测多尺度( smallImg, 面2,
                                 1.1, 2, 0
                                 //|CASCADE_FIND_BIGGEST_OBJECT
                                 //|CASCADE_DO_ROUGH_SEARCH
|CASCADE_SCALE_IMAGE,
                                 尺寸(30, 30) );
        对于( 矢量<矩形>::const_iterator r = 面2.begin(); r != 面2.end(); ++r )
        {
面.push_back(矩形(smallImg.cols - r->x - r->width, r->y, r->width, r->height));
        }
    }
t = (double)获得滴答数() - t;
printf( "检测时间 = %g ms\n", t*1000/获得滴答频率());
    对于 ( 大小_t i = 0; i < 面.size(); i++ )
    {
        矩形 r = 面[i];
        垫子 smallImgROI;
矢量<矩形>嵌套对象;
        点 中心;
        标量 颜色 = 颜色[i%8];
        int 半径;
 
        double 长宽比 = (double)r.width/r.height;
        如果( 0.75 < 长宽比 && 长宽比 < 1.3 )
        {
中心.x = 四舍五入((r.x + r.width*0.5)*规模);
中心.y = 四舍五入((r.y + r.height*0.5)*规模);
半径 = 四舍五入((r.width + r.height)*0.25*规模);
            圆( img, 中心, 半径, 颜色, 3, 8, 0 );
        }
        else
            矩形( img,                        点(cvRound((r.x + r.width-1)*scale), cvRound((r.y + r.height-1)*scale)),
color, 3, 8, 0);
        if( nestedCascade.empty() )
            continue;
smallImgROI = smallImg( r );
nestedCascade.detectMultiScale( smallImgROI, nestedObjects,
            1.1, 2, 0
            //|CASCADE_FIND_BIGGEST_OBJECT
            //|CASCADE_DO_ROUGH_SEARCH
            //|CASCADE_DO_CANNY_PRUNING
|CASCADE_SCALE_IMAGE,
            尺寸(30, 30) );
        for ( size_t j = 0; j < nestedObjects.size(); j++ )
        {
            矩形 nr = nestedObjects[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);
            圆( img, 中心, 半径, 颜色, 3, 8, 0 );
        }
    }
    imshow( "result", img );
}