1 #include <opencv2/opencv.hpp> 2 #include <iostream> 3 4 using namespace cv; 5 using namespace std; 6 7 Mat watershedCluster(Mat &image, int &numSegments); 8 //参数分别为分割后图像,分割块数,输入原图 9 void createDisplaySegments(Mat &segments, int numSegments, Mat &image); 10 int main(int argc, char** argv) { 11 Mat src = imread("cvtest.png"); 12 if (src.empty()) { 13 printf("could not load image...\n"); 14 return -1; 15 } 16 namedWindow("input image", CV_WINDOW_AUTOSIZE); 17 imshow("input image", src); 18 19 int numSegments; 20 Mat markers = watershedCluster(src, numSegments); 21 createDisplaySegments(markers, numSegments, src); 22 waitKey(0); 23 return 0; 24 } 25 26 Mat watershedCluster(Mat &image, int &numComp) { 27 // 二值化 28 Mat gray, binary; 29 cvtColor(image, gray, COLOR_BGR2GRAY); 30 threshold(gray, binary, 0, 255, THRESH_BINARY | THRESH_OTSU); 31 // 形态学与距离变换 32 Mat k = getStructuringElement(MORPH_RECT, Size(3, 3), Point(-1, -1)); 33 morphologyEx(binary, binary, MORPH_OPEN, k, Point(-1, -1)); 34 Mat dist; 35 distanceTransform(binary, dist, DistanceTypes::DIST_L2, 3, CV_32F); 36 normalize(dist, dist, 0.0, 1.0, NORM_MINMAX); 37 38 // 开始生成标记 39 threshold(dist, dist, 0.1, 1.0, THRESH_BINARY); 40 normalize(dist, dist, 0, 255, NORM_MINMAX); 41 dist.convertTo(dist, CV_8UC1); 42 43 // 标记开始 44 vector<vector<Point>> contours; 45 vector<Vec4i> hireachy; 46 findContours(dist, contours, hireachy, RETR_CCOMP, CHAIN_APPROX_SIMPLE); 47 if (contours.empty()) { 48 return Mat(); 49 } 50 51 Mat markers(dist.size(), CV_32S); 52 markers = Scalar::all(0); 53 for (int i = 0; i < contours.size(); i++) { 54 drawContours(markers, contours, i, Scalar(i + 1), -1, 8, hireachy, INT_MAX); 55 } 56 circle(markers, Point(5, 5), 3, Scalar(255), -1); 57 58 // 分水岭变换 59 watershed(image, markers); 60 numComp = contours.size(); 61 return markers; 62 } 63 64 void createDisplaySegments(Mat &markers, int numSegments, Mat &image) { 65 // generate random color 66 vector<Vec3b> colors; 67 for (size_t i = 0; i < numSegments; i++) { 68 int r = theRNG().uniform(0, 255); 69 int g = theRNG().uniform(0, 255); 70 int b = theRNG().uniform(0, 255); 71 colors.push_back(Vec3b((uchar)b, (uchar)g, (uchar)r)); 72 } 73 74 // 颜色填充与最终显示 75 Mat dst = Mat::zeros(markers.size(), CV_8UC3); 76 int index = 0; 77 for (int row = 0; row < markers.rows; row++) { 78 for (int col = 0; col < markers.cols; col++) { 79 index = markers.at<int>(row, col); 80 if (index > 0 && index <= numSegments) { 81 dst.at<Vec3b>(row, col) = colors[index - 1]; 82 } 83 else { 84 dst.at<Vec3b>(row, col) = Vec3b(255, 255, 255); 85 } 86 } 87 } 88 imshow("分水岭图像分割-演示",dst); 89 return; 90 }标签:分割,dist,Mat,int,image,markers,图像,255,OpenCV3.2 From: https://www.cnblogs.com/ybqjymy/p/17639452.html