0penCV_（Watershed Segmenter）使用分水岭算法对图像进行分割

柔光的暖阳◎ 2022-09-29 04:29 160阅读 0赞

//使用 分水岭算法 对图像进行分割

cv::Mat image = cv::imread("../../aTestImage/cow.jpg", 1);
    cv::Mat image1 = cv::imread("../../aTestImage/cow.jpg", 0);
    cv::Mat binary;
    
    
    cv::threshold(image1, binary, 60, 255, cv::THRESH_BINARY_INV);//阈值化得到二值化图像binary
    
    
    cv::namedWindow("image", 1);
    cv::imshow("image", image);
    
    
    cv::Mat fg;
    cv::erode(binary, fg, cv::Mat(), cv::Point(-1, -1), 6);//用点Point对图像腐蚀6次
         // cv::namedWindow("fg", 1);
    //cv::imshow("fg", fg);
    
    
    cv::Mat bg;
    cv::dilate(binary, bg, cv::Mat(), cv::Point(-1, -1), 6);//用点Point对图像膨胀6次
    cv::threshold(bg, bg, 1, 128, cv::THRESH_BINARY_INV);
    //cv::namedWindow("bg", 1);
    //cv::imshow("bg", bg);
    
    
    cv::Mat markers(binary.size(), CV_8U, cv::Scalar(0));
    markers = fg + bg;
    cv::namedWindow("markers", 1);
        cv::imshow("markers", markers);
    
    
    WatershedSegmenter segmenter;
    segmenter.setMarkers(markers);//通过对二值图像的腐蚀+膨胀，获得标记图像
    markers = segmenter.process(image); // markers是一个数组，图像显示为纯黑
    
    
    cv::namedWindow("Segmentation");//显示分割图像
    cv::imshow("Segmentation", segmenter.getSegmentation());
    
    
    cv::namedWindow("Watersheds");//显示边界图像
    cv::imshow("Watersheds", segmenter.getWatersheds());
    
    
    cv::waitKey(0);
    return 0;
    }
    
    
    
    class WatershedSegmenter
    {
    public:
    WatershedSegmenter();
    ~WatershedSegmenter();
    void setMarkers(const cv::Mat &markerImage)
    {
    markerImage.convertTo(markers,CV_32S);
    }
    cv::Mat process(const cv::Mat &image)
    {
    cv::watershed(image, markers);
    return markers;
    }
    cv::Mat getSegmentation()
    {
    cv::Mat tmp;
    markers.convertTo(tmp, CV_8U);//等价于(tmp, CV_8U，1, 0)  //double alpha=1, double beta=0
    return tmp;
    
    
    }
    cv::Mat getWatersheds()
    {
    cv::Mat tmp;
    markers.convertTo(tmp, CV_8U,255,255);//只显示边界...
    return tmp;
    }
    private:
    cv::Mat markers;
    };

![Center][]

[Center]: /images/20220708/aefc05cac50f4bfe819c136968b36e75.png