floodfill分割

floodfill分割

floodfill泛洪填充算法是在很多图形绘制软件中常用的填充算法，通常来说是自动选中与种子像素相关的区域，利用指定的颜色区域颜色替换，可用于标记或分离图像的某部分。windows的图像编辑软件中的油漆桶这一功能，以及类似photoshop的魔术棒选择工具，都是通过floodfill泛洪填充来改进和延伸的。
示例代码如下：

#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/highgui/highgui.hpp"
#include <iostream>
using namespace cv;
using namespace std;
// 初识化参数
Mat 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 != CV_EVENT_LBUTTONDOWN)
        return;
    // floodfill参数设置
    Point seed = Point(x, y);
    int lo = ffillMode == 0 ? 0 : loDiff;
    int up = ffillMode == 0 ? 0 : upDiff;
    int flags = connectivity + (newMaskVal << 8) +
        (ffillMode == 1 ? CV_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, CV_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);
}
int main()
{
    cv::Mat srcImage = cv::imread("..\\images\\sea.jpg");
    if (srcImage.empty())
        return 0;
    srcImage.copyTo(image);
    cvtColor(srcImage, gray, CV_BGR2GRAY);
    mask.create(srcImage.rows + 2, srcImage.cols + 2, CV_8UC1);
    // 鼠标响应回调函数
    namedWindow("image", 0);
    setMouseCallback("image", onMouse, 0);
    for (;;)
    {
        imshow("image", isColor ? image : gray);
        int c = waitKey(0);
        if ((c & 255) == 27)
        {
            cout << "Exiting ...\n";
            break;
        }
        if (c == 'r')//清除操作
        {
            cout << "Original image is restored\n";
            srcImage.copyTo(image);
            cvtColor(image, gray, CV_BGR2GRAY);
            mask = Scalar::all(0);
        }
    }
    return 0;
}