OpenCV——距離變換與分水嶺算法的（圖像分割）

 

 C++: void distanceTransform(InputArray src, OutputArray dst, int distanceType, int maskSize)

 

參數詳解：

InputArray src：輸入的圖像，一般為二值圖像

 OutputArray dst：輸出的圖像

int distanceType：所用的求解距離的類型、

It can be CV_DIST_L1, CV_DIST_L2 , or CV_DIST_C

mask_size  距離變換掩模的大小，可以是 3 或 5. 對 CV_DIST_L1 或 CV_DIST_C 的情況，參數值被強制設定為 3, 因為 3×3 mask 給出 5×5 mask 一樣的結果，而且速度還更快。 

 

mask

用戶自定義距離情況下的 mask。 在 3×3 mask 下它由兩個數(水平/垂直位量，對角線位移量）組成， 5×5 mask 下由三個數組成(水平/垂直位移量，對角位移和 國際象棋里的馬步(馬走日))

 

 

 

 

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

using namespace cv;
using namespace std;

Mat src;

int main(int argc, char** argv)
{
    src = imread("分水嶺.jpg");
    if (src.empty())
    {
        printf("Can not load Image...");
        return -1;
    }
    imshow("input Image",src);

    //白色背景變成黑色
    for (int row=0;row<src.rows;row++)
    {
        for (int col = 0; col < src.cols; col++) {
            if (src.at<Vec3b>(row, col) == Vec3b(255, 255, 255)) {
                src.at<Vec3b>(row, col)[0] = 0;
                src.at<Vec3b>(row, col)[1] = 0;
                src.at<Vec3b>(row, col)[2] = 0;
            }
        }
    }
    imshow("black backgroung", src);

    //sharpen(提高對比度)
    Mat kernel = (Mat_<float>(3, 3) << 1, 1, 1, 1, -8, 1, 1, 1, 1);

    //make it more sharp
    Mat imgLaplance;
    Mat sharpenImg = src;
    //拉普拉斯算子實現邊緣提取
    filter2D(src, imgLaplance, CV_32F, kernel, Point(-1, -1), 0, BORDER_DEFAULT);//拉普拉斯有浮點數計算，位數要提高到32
    src.convertTo(sharpenImg, CV_32F);

    //原圖減邊緣（白色）實現邊緣增強
    Mat resultImg = sharpenImg - imgLaplance;

    resultImg.convertTo(resultImg,CV_8UC3);
    imgLaplance.convertTo(imgLaplance, CV_8UC3);
    imshow("sharpen Image", resultImg);

    //轉換成二值圖
    Mat binary;
    cvtColor(resultImg, resultImg, CV_BGR2GRAY);
    threshold(resultImg, binary,40,255,THRESH_BINARY|THRESH_OTSU);
    imshow("binary image",binary);

    //距離變換
    Mat distImg;
    distanceTransform(binary,distImg,DIST_L1,3,5);
    normalize(distImg, distImg, 0, 1, NORM_MINMAX);    
    imshow("dist image",distImg);

    //二值化
    threshold(distImg, distImg, 0.4, 1, THRESH_BINARY);
    imshow("dist binary image", distImg);

    //腐蝕(使得連在一起的部分分開)
    Mat k1 = Mat::ones(3, 3, CV_8UC1);
    erode(distImg, distImg, k1);
    imshow("分開", distImg);

    //標記
    Mat dist_8u;
    distImg.convertTo(dist_8u,CV_8U);
    vector<vector<Point>> contours;
    findContours(dist_8u, contours, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE, Point(0, 0));

    //創建標記
    Mat marker = Mat::zeros(src.size(),CV_32SC1);

    //畫標記
    for (size_t i = 0; i < contours.size(); i++)
    {
        drawContours(marker,contours,static_cast<int>(i),Scalar(static_cast<int>(i)+1),-1);
    }

    circle(marker, Point(5, 5), 3, Scalar(255, 255, 255), -1);
    imshow("marker",marker*1000);

    //分水嶺變換
    watershed(src,marker);//根據距離變換的標記，在原圖上分離
    Mat water = Mat::zeros(marker.size(),CV_8UC1);
    marker.convertTo(water,CV_8UC1);
    bitwise_not(water, water,Mat());//取反操作
    //imshow("源 image", src);
    imshow("watershed Image", water);

    // generate random color
    vector<Vec3b> colors;
    for (size_t i = 0; i < contours.size(); i++) {
        int r = theRNG().uniform(0, 255);
        int g = theRNG().uniform(0, 255);
        int b = theRNG().uniform(0, 255);
        colors.push_back(Vec3b((uchar)b, (uchar)g, (uchar)r));
    }

    // fill with color and display final result
    Mat dst = Mat::zeros(marker.size(), CV_8UC3);
    for (int row = 0; row < marker.rows; row++) {
        for (int col = 0; col < marker.cols; col++) {
            int index = marker.at<int>(row, col);
            if (index > 0 && index <= static_cast<int>(contours.size())) {
                dst.at<Vec3b>(row, col) = colors[index - 1];
            }
            else {
                dst.at<Vec3b>(row, col) = Vec3b(0, 0, 0);
            }
        }
    }
    imshow("Final Result", dst);
    waitKey(0);
    return 0;
}