opencv 图像分水岭算法

本章内容

1.给图像做标记

2.利用分水岭算法寻找标记的边界

输出结果

消除背景的影响

输出结果

代码

#include <ostream>
#include <opencv.hpp>
#include <math.h>

int main(int argc, char *argv[])
{
    /*
     本章内容:
        图像分水岭实验
    */
    cv::String fileName = "/home/wang/dev/Image/pingpang.jpeg";
    cv::String fileName1 = "/home/wang/dev/Image/hei.png";
    cv::Mat src = cv::imread(fileName);
//    cv::Mat src1 = cv::imread(fileName1);
    if(src.data == NULL){
        printf("图像读入失败\n");
        return -1;
    }
    cv::imshow("src",src);

    cv::Mat gray;
    cv::cvtColor(src,gray,cv::COLOR_BGR2GRAY);
    for (int row = 0; row < src.rows; row++) {
        for (int col = 0; col < src.cols; col++) {
            if (gray.at<uchar>(row, col) < 50) {
                src.at<cv::Vec3b>(row, col)[0] = 0;
                src.at<cv::Vec3b>(row, col)[1] = 0;
                src.at<cv::Vec3b>(row, col)[2] = 0;
            }
        }
    }
    cv::imshow("src1",src);

    // convert to binary
    cv::Mat binaryImg;
//    cv::Mat gray;
    cv::cvtColor(src, gray,  cv::COLOR_BGR2GRAY);
    cv::threshold(gray, binaryImg, 40, 255,  cv::THRESH_BINARY |  cv::THRESH_OTSU);
    cv::imshow("binary image", binaryImg);

    cv::Mat distImg;
    cv::distanceTransform(binaryImg, distImg, cv::DIST_L1, 3, 5);
    cv::normalize(distImg, distImg, 0, 1, cv::NORM_MINMAX);
    cv::imshow("distance result", distImg);

    // binary again
    cv::threshold(distImg, distImg, .4, 1, cv::THRESH_BINARY);
    cv::Mat k1 = cv::Mat::ones(13, 13, CV_8UC1);
    cv::erode(distImg, distImg, k1, cv::Point(-1, -1));
    cv::imshow("distance binary image", distImg);

    // markers
    cv::Mat dist_8u;
    distImg.convertTo(dist_8u, CV_8U);
    std::vector<std::vector<cv::Point>> contours;
    cv::findContours(dist_8u, contours, cv::RETR_EXTERNAL, cv::CHAIN_APPROX_SIMPLE);
    cv::imshow("dist_8u",dist_8u*255);
    // create makers
    cv::Mat markers = cv::Mat::zeros(src.size(), CV_32SC1);
    cv::Mat mymarkers= cv::Mat::zeros(src.size(), CV_8UC3);;
    cv::RNG rng(1234);
    for (size_t i = 0; i < contours.size(); i++) {
        cv::drawContours(markers, contours, static_cast<int>(i), cv::Scalar::all(static_cast<int>(i) + 1), -1);
        cv::drawContours(mymarkers, contours, i, cv::Scalar(rng.uniform(0,255),rng.uniform(0,255),rng.uniform(0,255)), -1);
    }
    cv::circle(markers, cv::Point(5, 5), 3, cv::Scalar(255, 255, 255), -1);
    cv::imshow("my markers", mymarkers);

//    // perform watershed
    cv::watershed(src, markers);
    cv::Mat mark = cv::Mat::zeros(markers.size(), CV_8UC1);
    markers.convertTo(mark, CV_8UC1);
    cv::bitwise_not(mark, mark, cv::Mat());
    cv::imshow("watershed image", mark);

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

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

    cv::waitKey(0);
    return 1;
}