opencv入门漫水填充算法(Floodfill)
Posted long-w
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了opencv入门漫水填充算法(Floodfill)相关的知识,希望对你有一定的参考价值。
一、引言 · 漫水填充的定义
漫水填充法是一种用特定的颜色填充联通区域,通过设置可连通像素的上下限以及连通方式来达到不同的填充效果的方法。漫水填充经常被用来标记或分离图像的一部分以便对其进行进一步处理或分析,也可以用来从输入图像获取掩码区域,掩码会加速处理过程,或只处理掩码指定的像素点,操作的结果总是某个连续的区域。
二、漫水填充法的基本思想
所谓漫水填充,简单来说,就是自动选中了和种子点相连的区域,接着将该区域替换成指定的颜色,这是个非常有用的功能,经常用来标记或者分离图像的一部分进行处理或分析。漫水填充也可以用来从输入图像获取掩码区域,掩码会加速处理过程,或者只处理掩码指定的像素点。
以此填充算法为基础,类似photoshop的魔术棒选择工具就很容易实现了。漫水填充(FloodFill)是查找和种子点联通的颜色相同的点,魔术棒选择工具则是查找和种子点联通的颜色相近的点,将和初始种子像素颜色相近的点压进栈作为新种子。
在OpenCV中,漫水填充是填充算法中最通用的方法。且在OpenCV 2.X中,使用C++重写过的FloodFill函数有两个版本。一个不带掩膜mask的版本,和一个带mask的版本。这个掩膜mask,就是用于进一步控制哪些区域将被填充颜色(比如说当对同一图像进行多次填充时)。这两个版本的FloodFill,都必须在图像中选择一个种子点,然后把临近区域所有相似点填充上同样的颜色,不同的是,不一定将所有的邻近像素点都染上同一颜色,漫水填充操作的结果总是某个连续的区域。当邻近像素点位于给定的范围(从loDiff到upDiff)内或在原始seedPoint像素值范围内时,FloodFill函数就会为这个点涂上颜色。
三、floodFill函数详解
第一个版本的floodFill:
int floodFill(InputOutputArray image, Point seedPoint, Scalar newVal, Rect* rect=0, Scalar loDiff=Scalar(), Scalar upDiff=Scalar(), int flags=4 )
第二个版本的floodFill:
int floodFill(InputOutputArray image, InputOutputArray mask, Point seedPoint,Scalar newVal, Rect* rect=0, Scalar loDiff=Scalar(), Scalar upDiff=Scalar(), int flags=4 )
参数详解:
第一个参数,InputOutputArray类型的image, 输入/输出1通道或3通道,8位或浮点图像,具体参数由之后的参数具体指明。
第二个参数, InputOutputArray类型的mask,这是第二个版本的floodFill独享的参数,表示操作掩模,。它应该为单通道、8位、长和宽上都比输入图像
image 大两个像素点的图像。第二个版本的floodFill需要使用以及更新掩膜,所以这个mask参数我们一定要将其准备好并填在此处。需要注意的是,
漫水填充不会填充掩膜mask的非零像素区域。例如,一个边缘检测算子的输出可以用来作为掩膜,以防止填充到边缘。同样的,也可以在多次的函数调用
中使用同一个掩膜,以保证填充的区域不会重叠。另外需要注意的是,掩膜mask会比需填充的图像大,所以 mask 中与输入图像(x,y)像素点相对应的点的
坐标为(x+1,y+1)。
第三个参数,Point类型的seedPoint,漫水填充算法的起始点。
第四个参数,Scalar类型的newVal,像素点被染色的值,即在重绘区域像素的新值。
第五个参数,Rect*类型的rect,有默认值0,一个可选的参数,用于设置floodFill函数将要重绘区域的最小边界矩形区域。
第六个参数,Scalar类型的loDiff,有默认值Scalar( ),表示当前观察像素值与其部件邻域像素值或者待加入该部件的种子像素之间的亮度或颜色之负差
(lower brightness/color difference)的最大值。
第七个参数,Scalar类型的upDiff,有默认值Scalar( ),表示当前观察像素值与其部件邻域像素值或者待加入该部件的种子像素之间的亮度或颜色之正差
(lower brightness/color difference)的最大值。
第八个参数,int类型的flags,操作标志符,此参数包含三个部分,比较复杂,我们一起详细看看。
低八位(第0~7位)用于控制算法的连通性,可取4 (4为缺省值) 或者 8。如果设为4,表示填充算法只考虑当前像素水平方向和垂直方向的相邻点;如果
设为 8,除上述相邻点外,还会包含对角线方向的相邻点。
高八位部分(16~23位)可以为0 或者如下两种选项标识符的组合:
FLOODFILL_FIXED_RANGE - 如果设置为这个标识符的话,就会考虑当前像素与种子像素之间的差,否则就考虑当前像素与其相邻像素的差。也就
是说,这个范围是浮动的。
FLOODFILL_MASK_ONLY - 如果设置为这个标识符的话,函数不会去填充改变原始图像 (也就是忽略第三个参数newVal), 而是去填充掩模图像
(mask)。这个标识符只对第二个版本的floodFill有用,因第一个版本里面压根就没有mask参数。
中间八位部分,上面关于高八位FLOODFILL_MASK_ONLY标识符中已经说的很明显,需要输入符合要求的掩码。Floodfill的flags参数的中间八位的值就
是用于指定填充掩码图像的值的。但如果flags中间八位的值为0,则掩码会用1来填充。
而所有flags可以用or操作符连接起来,即“|”。例如,如果想用8邻域填充,并填充固定像素值范围,填充掩码而不是填充源图像,以及设填充值为38,那么输入的参数是这样:
flags=8 | FLOODFILL_MASK_ONLY | FLOODFILL_FIXED_RANGE | (38<<8)
【示例】
1 //漫水填充
2 #include <opencv2/opencv.hpp>
3 #include <opencv2/imgproc/imgproc.hpp>
4
5 using namespace cv;
6
7 int main( )
8 {
9 Mat src = imread("1.jpg");
10 imshow("【原始图】",src);
11 Rect ccomp;
12 floodFill(src, Point(50,300), Scalar(155, 255,55), &ccomp, Scalar(20, 20, 20),Scalar(20, 20, 20));
13 imshow("【效果图】",src);
14 imwrite("效果图.jpg", src);
15 waitKey(0);
16 return 0;
17 }
效果展示:
四、关于SetMouseCallback函数
因为下面示例程序中有用到SetMouseCallback函数,我们在这里讲一讲。SetMouseCallback函数为指定的窗口设置鼠标回调函数。
C++: void setMouseCallback(conststring& winname, MouseCallback onMouse, void* userdata=0 )
第一个参数,const string&类型的winname,为窗口的名字。
第二个参数,MouseCallback类型的onMouse,指定窗口里每次鼠标时间发生的时候,被调用的函数指针。这个函数的原型应该为
voidFoo(int event, int x, int y, int flags, void* param);其中event是 CV_EVENT_*变量之一, x和y
是鼠标指针在图像坐标系的坐标(不是窗口坐标系), flags是CV_EVENT_FLAG的组合, param是用户定义的传递到
cvSetMouseCallback函数调用的参数。
第三个参数,void*类型的userdata,用户定义的传递到回调函数的参数,有默认值0。
五、综合示例
1 //漫水填充,综合示例
2 #include "opencv2/imgproc/imgproc.hpp"
3 #include "opencv2/highgui/highgui.hpp"
4 #include <iostream>
5
6 using namespace cv;
7 using namespace std;
8
9 Mat g_srcImage, g_dstImage, g_grayImage, g_maskImage;//定义原始图、目标图、灰度图、掩模图
10 int g_nFillMode = 1;//漫水填充的模式
11 int g_nLowDifference = 20, g_nUpDifference = 20;//负差最大值、正差最大值
12 int g_nConnectivity = 4;//表示floodFill函数标识符低八位的连通值
13 int g_bIsColor = true;//是否为彩色图的标识符布尔值
14 bool g_bUseMask = false;//是否显示掩膜窗口的布尔值
15 int g_nNewMaskVal = 255;//新的重新绘制的像素值
16
17 static void ShowHelpText()
18 {
19 //输出一些帮助信息
20 cout << "
欢迎来到漫水填充示例程序~
";
21 cout << "
按键操作说明:
"
22 " 鼠标点击图中区域- 进行漫水填充操作
"
23 " 键盘按键【ESC】- 退出程序
"
24 " 键盘按键【1】- 切换彩色图/灰度图模式
"
25 " 键盘按键【2】- 显示/隐藏掩膜窗口
"
26 " 键盘按键【3】- 恢复原始图像
"
27 " 键盘按键【4】- 使用空范围的漫水填充
"
28 " 键盘按键【5】- 使用渐变、固定范围的漫水填充
"
29 " 键盘按键【6】- 使用渐变、浮动范围的漫水填充
"
30 " 键盘按键【7】- 操作标志符的低八位使用4位的连接模式
"
31 " 键盘按键【8】- 操作标志符的低八位使用8位的连接模式
"
32 "
n
";
33 }
34
35 static void onMouse( int event, int x, int y, int, void* )
36 {
37 // 若鼠标左键没有按下,便返回
38 if( event != CV_EVENT_LBUTTONDOWN )
39 return;
40
41 Point seed = Point(x,y);
42 int LowDifference = g_nFillMode == 0 ? 0 : g_nLowDifference;//空范围的漫水填充,此值设为0,否则设为全局的g_nLowDifference
43 int UpDifference = g_nFillMode == 0 ? 0 : g_nUpDifference;//空范围的漫水填充,此值设为0,否则设为全局的g_nUpDifference
44 int flags = g_nConnectivity + (g_nNewMaskVal << 8) +
45 (g_nFillMode == 1 ? CV_FLOODFILL_FIXED_RANGE : 0);//标识符的0~7位为g_nConnectivity,8~15位为g_nNewMaskVal左移8位的值,16~23位为CV_FLOODFILL_FIXED_RANGE或者0。
46
47 //随机生成bgr值
48 int b = (unsigned)theRNG() & 255;//随机返回一个0~255之间的值
49 int g = (unsigned)theRNG() & 255;//随机返回一个0~255之间的值
50 int r = (unsigned)theRNG() & 255;//随机返回一个0~255之间的值
51 Rect ccomp;//定义重绘区域的最小边界矩形区域
52
53 Scalar newVal = g_bIsColor ? Scalar(b, g, r) : Scalar(r*0.299 + g*0.587 + b*0.114);//在重绘区域像素的新值,若是彩色图模式,取Scalar(b, g, r);若是灰度图模式,取Scalar(r*0.299 + g*0.587 + b*0.114)
54
55 Mat dst = g_bIsColor ? g_dstImage : g_grayImage;//目标图的赋值
56 int area;
57
58 if( g_bUseMask )
59 {
60 threshold(g_maskImage, g_maskImage, 1, 128, CV_THRESH_BINARY);
61 area = floodFill(dst, g_maskImage, seed, newVal, &ccomp, Scalar(LowDifference, LowDifference, LowDifference),
62 Scalar(UpDifference, UpDifference, UpDifference), flags);
63 imshow( "mask", g_maskImage );
64 }
65 else
66 {
67 area = floodFill(dst, seed, newVal, &ccomp, Scalar(LowDifference, LowDifference, LowDifference),
68 Scalar(UpDifference, UpDifference, UpDifference), flags);
69 }
70
71 imshow("效果图", dst);
72
73 cout << area << " 个像素被重绘
";
74 }
75
76 int main( int argc, char** argv )
77 {
78 //载入原图
79 g_srcImage = imread("1.jpg", 1);
80
81 //if( !g_srcImage.data ) { printf("Oh,no,读取图片image0错误~!
"); return false; }
82
83 //显示帮助文字
84 ShowHelpText();
85
86 g_srcImage.copyTo(g_dstImage);//拷贝源图到目标图
87 cvtColor(g_srcImage, g_grayImage, COLOR_BGR2GRAY);//转换三通道的image0到灰度图
88 g_maskImage.create(g_srcImage.rows+2, g_srcImage.cols+2, CV_8UC1);//利用image0的尺寸来初始化掩膜mask
89
90 namedWindow( "效果图",CV_WINDOW_AUTOSIZE );
91
92 //创建Trackbar
93 createTrackbar( "负差最大值", "效果图", &g_nLowDifference, 255, 0 );
94 createTrackbar( "正差最大值" ,"效果图", &g_nUpDifference, 255, 0 );
95
96 //鼠标回调函数
97 setMouseCallback( "效果图", onMouse, 0 );
98
99 //循环轮询按键
100 while(1)
101 {
102 //先显示效果图
103 imshow("效果图", g_bIsColor ? g_dstImage : g_grayImage);
104
105 //获取键盘按键
106 int c = waitKey(0);
107 //判断ESC是否按下,若按下便退出
108 if( (c & 255) == 27 )
109 {
110 cout << "程序退出...........
";
111 break;
112 }
113
114 //根据按键的不同,进行各种操作
115 switch( (char)c )
116 {
117 //如果键盘“1”被按下,效果图在在灰度图,彩色图之间互换
118 case ‘1‘:
119 if( g_bIsColor )//若原来为彩色,转为灰度图,并且将掩膜mask所有元素设置为0
120 {
121 cout << "键盘“1”被按下,切换彩色/灰度模式,当前操作为将【彩色模式】切换为【灰度模式】
";
122 cvtColor(g_srcImage, g_grayImage, COLOR_BGR2GRAY);
123 g_maskImage = Scalar::all(0); //将mask所有元素设置为0
124 g_bIsColor = false; //将标识符置为false,表示当前图像不为彩色,而是灰度
125 }
126 else//若原来为灰度图,便将原来的彩图image0再次拷贝给image,并且将掩膜mask所有元素设置为0
127 {
128 cout << "键盘“1”被按下,切换彩色/灰度模式,当前操作为将【彩色模式】切换为【灰度模式】
";
129 g_srcImage.copyTo(g_dstImage);
130 g_maskImage = Scalar::all(0);
131 g_bIsColor = true;//将标识符置为true,表示当前图像模式为彩色
132 }
133 break;
134 //如果键盘按键“2”被按下,显示/隐藏掩膜窗口
135 case ‘2‘:
136 if( g_bUseMask )
137 {
138 destroyWindow( "mask" );
139 g_bUseMask = false;
140 }
141 else
142 {
143 namedWindow( "mask", 0 );
144 g_maskImage = Scalar::all(0);
145 imshow("mask", g_maskImage);
146 g_bUseMask = true;
147 }
148 break;
149 //如果键盘按键“3”被按下,恢复原始图像
150 case ‘3‘:
151 cout << "按键“3”被按下,恢复原始图像
";
152 g_srcImage.copyTo(g_dstImage);
153 cvtColor(g_dstImage, g_grayImage, COLOR_BGR2GRAY);
154 g_maskImage = Scalar::all(0);
155 break;
156 //如果键盘按键“4”被按下,使用空范围的漫水填充
157 case ‘4‘:
158 cout << "按键“4”被按下,使用空范围的漫水填充
";
159 g_nFillMode = 0;
160 break;
161 //如果键盘按键“5”被按下,使用渐变、固定范围的漫水填充
162 case ‘5‘:
163 cout << "按键“5”被按下,使用渐变、固定范围的漫水填充
";
164 g_nFillMode = 1;
165 break;
166 //如果键盘按键“6”被按下,使用渐变、浮动范围的漫水填充
167 case ‘6‘:
168 cout << "按键“6”被按下,使用渐变、浮动范围的漫水填充
";
169 g_nFillMode = 2;
170 break;
171 //如果键盘按键“7”被按下,操作标志符的低八位使用4位的连接模式
172 case ‘7‘:
173 cout << "按键“7”被按下,操作标志符的低八位使用4位的连接模式
";
174 g_nConnectivity = 4;
175 break;
176 //如果键盘按键“8”被按下,操作标志符的低八位使用8位的连接模式
177 case ‘8‘:
178 cout << "按键“8”被按下,操作标志符的低八位使用8位的连接模式
";
179 g_nConnectivity = 8;
180 break;
181 }
182 }
183
184 return 0;
185 }
以上是关于opencv入门漫水填充算法(Floodfill)的主要内容,如果未能解决你的问题,请参考以下文章