opencv - 带掩码的 OpenCV 泛洪填充

Question

OpenCV 的洪水填充功能的文档指出：

该函数使用和更新掩码，因此您负责初始化掩码内容。填充不能跨越掩码中的非零像素。例如，边缘检测器输出可用作掩码以停止在边缘填充。可以在对函数的多次调用中使用相同的掩码，以确保填充区域不重叠。

该函数如何更新掩码？它是否将填充中的所有像素设置为某个非零值？

Answer 1

与蒙版的种子点相同的连通分量中的所有零值像素都将替换为您指定的值。该值必须添加到flags参数中，左移 8 位：

uchar fillValue = 128;
cv::floodFill(img, mask, seed, cv::Scalar(255) ,0, cv::Scalar(), cv::Scalar(), 4 | cv::FLOODFILL_MASK_ONLY | (fillValue << 8));

下面是一个简单但也许很有启发性的例子。像这样创建图像：

//Create simple input image
cv::Point seed(4,4);
cv::Mat img = cv::Mat::zeros(100,100,CV_8UC1);
cv::circle(img, seed, 20, cv::Scalar(128),3);

此图像中的结果：

然后，创建一个蒙版并填充它：

//Create a mask from edges in the original image
cv::Mat mask;
cv::Canny(img, mask, 100, 200);
cv::copyMakeBorder(mask, mask, 1, 1, 1, 1, cv::BORDER_REPLICATE);

//Fill mask with value 128
uchar fillValue = 128;
cv::floodFill(img, mask, seed, cv::Scalar(255) ,0, cv::Scalar(), cv::Scalar(), 4 | cv::FLOODFILL_MASK_ONLY | (fillValue << 8));

给出这个结果：

掩码中的白色像素是边缘检测的结果，而灰色像素是洪水填充的结果。

更新： 作为对评论的回应，标志值4指定了用于比较颜色值差异的像素邻域。从文档中：

低位包含函数中使用的连接值 4（默认）或 8。连通性决定了考虑像素的哪些邻居。

当cv::FLOODFILL_MASK_ONLY标志未通过时，图像和掩码都会更新，但洪水填充将在任何非零掩码值处停止。

Answer 2

还有一个python版本

im = cv2.imread("seagull.jpg")
h,w,chn = im.shape
seed = (w/2,h/2)

mask = np.zeros((h+2,w+2),np.uint8)

floodflags = 4
floodflags |= cv2.FLOODFILL_MASK_ONLY
floodflags |= (255 << 8)

num,im,mask,rect = cv2.floodFill(im, mask, seed, (255,0,0), (10,)*3, (10,)*3, floodflags)

cv2.imwrite("seagull_flood.png", mask)

（来自维基媒体的海鸥图片：https ://commons.wikimedia.org/wiki/Commons:Quality_images#/media/File:Gull_portrait_ca_usa.jpg ）

结果：

Answer 3

根据 Aurelius 的回答，面具需要归零。

检查源中的评论，它指出

由于这既是输入参数又是输出参数，因此您必须负责对其进行初始化。填充不能跨越输入掩码中的非零像素。

掩码会影响结果，因此需要在使用前归零：

cv::Mat mask;
mask = cv::Mat::zeros(img.rows + 2, img.cols + 2, CV_8UC1);