分水岭算法是比较经典的图像分割算法。最近看到一副区域检测和统计的图像,感觉可以通过分水岭算法进行实现,于是顺便对opencv的分水岭算法进行学习。如图需要分割的图像:
opencv有自带的分水岭分割示例,分割图像为硬币图像,如图:
由于示例是python的代码,没有C++的代码,所以打算先用C++实现示例中的功能,然后再对本文开头的图像进行分割。
基本步骤
- 标记背景区域;
- 标记前景和未知区域;
- 合并标记图像;
- 分水岭算法进行分割。
标记背景图像
基本操作:灰度化->阈值分割(OTSU)->形态学去噪->形态学膨胀得到背景图。
Mat src = imread("water_coins.jpg");
Mat frame;
cvtColor(src, frame, CV_BGR2GRAY); //灰度化
Mat marker = frame.clone();
threshold(marker, frame, 0,