OPenCV floodfill 泛洪填充的使用方法

• 目的

        在做表情识别项目过程中遇到脸分区的问题，本以为可以用OPenCV floodfill 方法解决，对floodfill 不甚理解，于是乎写了一段例子来测试它的功能，为了不让大家重复劳动，在此分享给大家。

• 程序实现

        本代码是在Qt中实现，在此假设您已经会在Qt中配置OPenCV，如果不会也没关系，底部我会上传我的代码，从 .pro文件中您应该会看出我是如何在Qt上配OpenCV的，当然还有一个假设，您会编译Qt版的Opencv。

        如果Qt版的OpenCV你也不会配，那么没有关系，如果你是用的VS跑OpenCV直接把main.cpp中的代码拷到您的项目中也可以跑起来。

话不多说，直接上图：

第一步：release目录下输入 test.exe faceGrid.png启动程序

第二步：选中图像任意区域，floodfill会根据您选择的像素允许偏差范围，进行上色并统计这种区域像素个数

这就是floodfill方法的用处，中文应该叫泛洪填充，下面是代码：

#include <opencv2/imgproc/imgproc.hpp>

#include <opencv2/imgcodecs.hpp>

#include <opencv2/videoio/videoio.hpp>

#include <opencv2/highgui/highgui.hpp>

#include <iostream>

using namespace cv;

using namespace std;

static void help()

{

    cout << "\nThis program demonstrated the floodFill() function\n"

            "Call:\n"

            "./ffilldemo [image_name -- Default: ../data/fruits.jpg]\n" << endl;

    cout << "Hot keys: \n"

            "\tESC - quit the program\n"

            "\tc - switch color/grayscale mode\n"

            "\tm - switch mask mode\n"

            "\tr - restore the original image\n"

            "\ts - use null-range floodfill\n"

            "\tf - use gradient floodfill with fixed(absolute) range\n"

            "\tg - use gradient floodfill with floating(relative) range\n"

            "\t4 - use 4-connectivity mode\n"

            "\t8 - use 8-connectivity mode\n" << endl;

}

Mat image0, image, gray, mask;

int ffillMode = 1;

int loDiff = 20, upDiff = 20;

int connectivity = 4;

int isColor = true;

bool useMask = false;

int newMaskVal = 255;

static void onMouse( int event, int x, int y, int, void* )

{

    if( event != EVENT_LBUTTONDOWN )

        return;

    Point seed = Point(x,y);

    int lo = ffillMode == 0 ? 0 : loDiff;

    int up = ffillMode == 0 ? 0 : upDiff;

    int flags = connectivity + (newMaskVal << 8) +

                (ffillMode == 1 ? FLOODFILL_FIXED_RANGE : 0);

    int b = (unsigned)theRNG() & 255;

    int g = (unsigned)theRNG() & 255;

    int r = (unsigned)theRNG() & 255;

    Rect ccomp;

    Scalar newVal = isColor ? Scalar(b, g, r) : Scalar(r*0.299 + g*0.587 + b*0.114);

    Mat dst = isColor ? image : gray;

    int area;

    if( useMask )

    {

        threshold(mask, mask, 1, 128, THRESH_BINARY);

        area = floodFill(dst, mask, seed, newVal, &ccomp, Scalar(lo, lo, lo),

                  Scalar(up, up, up), flags);

        imshow( "mask", mask );

    }

    else

    {

        area = floodFill(dst, seed, newVal, &ccomp, Scalar(lo, lo, lo),

                  Scalar(up, up, up), flags);

    }

    imshow("image", dst);

    cout << area << " pixels were repainted\n";

}

int main( int argc, char** argv )

{

    cv::CommandLineParser parser (argc, argv,

        "{help h | | show help message}{@image|../data/fruits.jpg| input image}"

    );

    if (parser.has("help"))

    {

        parser.printMessage();

        return 0;

    }

    string filename = parser.get<string>("@image");

    image0 = imread(filename, 1);

    if( image0.empty() )

    {

        cout << "Image empty\n";

        parser.printMessage();

        return 0;

    }

    help();

    image0.copyTo(image);

    cvtColor(image0, gray, COLOR_BGR2GRAY);

    mask.create(image0.rows+2, image0.cols+2, CV_8UC1);

    namedWindow( "image", 0 );

    createTrackbar( "lo_diff", "image", &loDiff, 255, 0 );

    createTrackbar( "up_diff", "image", &upDiff, 255, 0 );

    setMouseCallback( "image", onMouse, 0 );

    for(;;)

    {

        imshow("image", isColor ? image : gray);

        int c = waitKey(0);

        if( (c & 255) == 27 )

        {

            cout << "Exiting ...\n";

            break;

        }

        switch( (char)c )

        {

        case 'c':

            if( isColor )

            {

                cout << "Grayscale mode is set\n";

                cvtColor(image0, gray, COLOR_BGR2GRAY);

                mask = Scalar::all(0);

                isColor = false;

            }

            else

            {

                cout << "Color mode is set\n";

                image0.copyTo(image);

                mask = Scalar::all(0);

                isColor = true;

            }

            break;

        case 'm':

            if( useMask )

            {

                destroyWindow( "mask" );

                useMask = false;

            }

            else

            {

                namedWindow( "mask", 0 );

                mask = Scalar::all(0);

                imshow("mask", mask);

                useMask = true;

            }

            break;

        case 'r':

            cout << "Original image is restored\n";

            image0.copyTo(image);

            cvtColor(image, gray, COLOR_BGR2GRAY);

            mask = Scalar::all(0);

            break;

        case 's':

            cout << "Simple floodfill mode is set\n";

            ffillMode = 0;

            break;

        case 'f':

            cout << "Fixed Range floodfill mode is set\n";

            ffillMode = 1;

            break;

        case 'g':

            cout << "Gradient (floating range) floodfill mode is set\n";

            ffillMode = 2;

            break;

        case '4':

            cout << "4-connectivity mode is set\n";

            connectivity = 4;

            break;

        case '8':

            cout << "8-connectivity mode is set\n";

            connectivity = 8;

            break;

        }

    }

    return 0;

}

最后附上我建的项目链接：http://download.csdn.net/detail/searobbers_duck/9552284