OpenCV3.2图像分割 实例9：K-Means-证件照背景替换

1 #include <opencv2/opencv.hpp> 2 #include <iostream> 3 4 using namespace cv; 5 using namespace std; 6 7 Mat mat_to_samples(Mat &image); 8 int main(int argc, char** argv) 14 namedWindow("输入图像", CV_WINDOW_AUTOSIZE); 15 imshow("输入图像", src); 16 17 // 组装数据 18 Mat points = mat_to_samples(src); 19 20 // 运行KMeans 21 int numCluster = 4; 22 Mat labels; 23 Mat centers; 24 TermCriteria criteria = TermCriteria(TermCriteria::EPS + TermCriteria::COUNT, 10, 0.1); 25 kmeans(points, numCluster, labels, criteria, 3, KMEANS_PP_CENTERS, centers); 26 27 // 去背景+遮罩生成 28 Mat mask=Mat::zeros(src.size(), CV_8UC1); 29 int index = src.rows*2 + 2; 30 int cindex = labels.at<int>(index, 0); 31 int height = src.rows; 32 int width = srcols; 33 //Mat dst; 34 //srcopyTo(dst); 35 for (int row = 0; row < height; row++) else 47 } 48 } 49 imshow("mask遮罩", mask); 50 51 // 腐蚀 + 高斯模糊 52 Mat k = getStructuringElement(MORPH_RECT, Size(3, 3), Point(1, 1));//用3*3像素进行腐蚀（减少3*3） 53 erode(mask, mask, k); 54 imshow("腐蚀mask", mask); 55 GaussianBlur(mask, mask, Size(3, 3), 0, 0);//用3*3像素进行高斯模糊（增加3*3） 所以边界不变 56 imshow("高斯模糊mask", mask); 57 58 // 通道混合 59 RNG rng(12345);//随机数 60 Vec3b color; 61 color[0] = 217;//rng.uniform(0, 255); 62 color[1] = 60;// rng.uniform(0, 255); color[2] = 160;// rng.uniform(0, 255); 64 Mat result(src.size(), src.type()); 65 66 double w = 0.0; 67 int b = 0, g = 0, r = 0; 68 int b1 = 0, g1 = 0, r1 = 0; 69 int b2 = 0, g2 = 0, r2 = 0; 70 71 for (int row = 0; row < height; row++) 77 else if (m == 0) 80 else 98 } 99 } 100 imshow("背景替换", result); 101 102 waitKey(0); 103 return 0; 104 } 105 106 Mat mat_to_samples(Mat &image) 122 } 123 return points; 124 }

上一篇：ITK 实例7 置信连接算法对脑部MHA文件进行三维分割

下一篇：ITK 实例17 阈值水平集算法对脑部MHA文件进行三维分割

---

OpenCV