声明:如果有写的不对的地方欢迎指正!
一、分水岭算法
关于分水岭算法的具体原理我就不说了,网上搜一下很多。OpenCV中的watershed函数实现的分水岭算法是基于“标记”的分割算法,用于解决传统的分水岭算法过度分割的问题。试想,一副图片中肯定有N多个“山谷”,它们中的很多是我们不想要的。
对于标记的原则,我总结是:你认为它们属于一个区域,就用标记将它们连接起来,对于另一个区域,再用另一个标记连接。就像这样:
图片中我认为有三个区域,所以做了三个标记。看到这里,你就可以把文章结尾的代码和图片拷到你的工程中试一试效果了。
二、代码分析
要想watershed函数,我们先要做一些准备工作:
1. 做标记
做标记的原则在上面已经说过了,具体对应代码中on_Mouse函数里面的内容,这是一个鼠标事件回调函数。else if (event == EVENT_MOUSEMOVE && (flags & EVENT_FLAG_LBUTTON))
{
Point pt(x, y);
if (previousPoint.x < 0)
previousPoint = pt;
//绘制白色线条
line(inpaintMask, previousPoint, pt, Scalar::all(255), 5, 8, 0);
line(srcImage1, previousPoint, pt, Scalar::all(255), 5, 8, 0);
previousPoint = pt;
imshow(WINDOW_NAME1, srcImage1);
}
else if (event == EVENT_MOUSEMOVE && (flags & EVENT_FLAG_LBUTTON))
{
Point pt(x, y);
if (previousPoint.x < 0)
previousPoint = pt;
//绘制白色线条
line(inpaintMask, previousPoint, pt, Scalar::all(255), 5, 8, 0);
line(srcImage1, previousPoint, pt, Scalar::all(255), 5, 8, 0);
previousPoint = pt;
imshow(WINDOW_NAME1, srcImage1);
}
在鼠标左键点按并移动时画线,其中的maskImage是一个CV_8UC1类型的掩模,绘制完的结果就是黑色背景上有几条线(标记),srcImage用于实时显示我们做标记的结果。
2. 寻找轮廓
对我们做过标记的maskImage寻找轮廓,这部分代码写在if ((char)c == '1')中,findContours函数这里不展开说明。
vector<vector<Point>> contours;
vector<Vec4i> hierarchy;
findContours(maskImage,