形態學濾波（4）：使用分水嶺算法對圖像進行分割

一、分水嶺算法

　　分水嶺變換是一個流行的圖像處理算法，用於快速分割圖像為同類區域。它背后的原理是，將圖像視為拓撲結構的地圖，

那么均質區域對應的是被陡峭邊緣包圍的平坦盆地。

實現方法：

　　分水嶺分割的結果是通過watershed()函數獲取。

　　我們將圖片中已知屬於某個區域的像素進行標記，基於這個標記，分水嶺算法開始確定其他像素的歸屬區域。

#include<opencv2/opencv.hpp>
#include<iostream>

using namespace std;
using namespace cv;

/*
    我們將創建灰度格式的標記圖，然后將它轉化成一張整圖
    把這些步驟封裝在WatershedSegmenter類中
*/
class WatershedSegmenter {
private:
    Mat markers;
public:
    void setMarkers(const Mat &markerImage) {
        //轉換位整數圖像
        markerImage.convertTo(markers, CV_32S);
    }

    Mat process(const Mat &image) {
        //使用算法
        watershed(image, markers);
        return markers;
    }
};

int main()
{
    //載入原圖
    Mat image = imread("C:\\Users\\Nelsoner\\Desktop\\Camera Roll\\008.jpg");

    //移除噪點與微小物體
    Mat fg;
    erode(image, fg, Mat(), Point(-1, -1), 6);

    
    //識別不包含物體的像素
    Mat bg;
    dilate(image, bg, Mat(), Point(-1, -1), 6);
    threshold(bg, bg, 1, 128, THRESH_BINARY_INV);




    //創建標記圖像
    Mat markers(image.size(), CV_8U, Scalar(0));
    markers = fg + bg;

    
    //創建分水嶺分割對象
    WatershedSegmenter segmenter;
    //設置標記，並進行處理
    segmenter.setMarkers(markers);

    markers = segmenter.process(image);
    
    
namedWindow("hah");
    imshow("hah", markers);
    

    waitKey();
    
    return 0;
}