8.5 分水岭算法(watershed algorithm)
1.基于拓扑理论的数学形态学的分割方法。
2.基本思想:把图像看作测地学上的拓扑地貌,图像中每一点像素的灰度值表示该点的海拔高度,每一个局部极小值及其影响区域称为集水盆,分水岭变换得到的是输入图像的集水盆图像,集水盆边界形成分水岭(输入图像的极大值点)。
3.计算过程:一个迭代标注过程,包括排序过程和淹没过程,先对每个像素的灰度级进行从低到高的排序,然后在从低到高实现淹没的过程中,对每一个局部最小值在h阶高度的影响域采用先进先出(FIFO)结构进行判断及标注。
8.5.1 实现分水岭算法:watershed()函数
1.基本操作:
基于标记的分割算法,在把图像传给函数之前,需要大致标记出图像中期望进行分割的区域,每个区域被标记为像素值1、2、3等,表示成为一个或多个连接组件,标记的值可使用findContours()函数和drawContours()函数由二进制的掩码检索出来,函数输出中,每一个标记中的像素被设置为标记的值,区域间的值被设置为-1。
2.函数原型:
void watershed(InputArray image,InputOutputArray markers)
3.参数说明:
(1)输入图像,8位三通道彩色图像
(2)函数调用后运算结果,输入/输出32位单通道图像的标记结果
8.5.2 综合示例
/*
程序说明:鼠标大致标记出图像中期望进行分割的区域
键盘按键【1】启动分水岭算法
按键【2】恢复原始图重新标记
*/
#include<opencv2/opencv.hpp>
#include<opencv2/imgproc/imgproc.hpp>
#include<opencv2/highgui/highgui.hpp>
#include<iostream>
using namespace cv;
using namespace std;
//定义辅助宏
#define WINDOW_NAME1 "【原始图窗口】"
#define WINDOW_NAME2 "【恢复原始图窗口】"
#define WINDOW_NAME3 "【分水岭变换窗口】"
//全局变量
Mat g_srcImage, g_maskImage;
Point prevPt(-1, -1);
//全局函数
static void on_Mouse(int event, int x, int y, int flags, void*);
static void ShowHelpText();
int main()
{
//【0】显示帮助信息
ShowHelpText();
//【1】载入原图并显示
g_srcImage = imread("night.jpg");
if (!g_srcImage.data)
{
printf("载入原图像失败~!\n");
return false;
}
imshow(WINDOW_NAME1, g_srcImage);
//【2】初始化掩模和灰度图
Mat srcImage, grayImage;
g_srcImage.copyTo(srcImage);
cvtColor(srcImage, g_maskImage, COLOR_BGR2GRAY);
cvtColor(g_maskImage, grayImage, COLOR_GRAY2BGR);
g_maskImage = Scalar: