Opencv4+VS2017实现灰度图片kmeans++聚类分割

本文环境:opencv4+VS2017+C++
搜了一些使用opencv实现kmeans算法的文章,没有一个是正常能用的。
其中不乏一些高阅读高收藏文章,比如
将3通道彩色图像转化为3维数据进行聚类,丢失了空间语义
信息,实质上变成了根据颜色聚类。导致明明不相连的部分,由于颜色一样,被分成了同一个label

同样是基于颜色的聚类,根本没考虑连通性,跟阈值分割没啥区别。这篇文章高达2W阅读数,却连头文件都加错了。

网上找的都不能用,于是在他们基础上改进一下
opencv中的kmeans其实是kmeans++算法,改进了kmeans受初始点影响较大的缺点,

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

using namespace cv;
using namespace std;

int main()
{
	Mat src = imread("E:\\code\\image\\cam0101.png");
	Mat src_gray;
	cvtColor(src, src_gray, COLOR_BGR2GRAY);
	morphologyEx(src_gray, src_gray, MORPH_OPEN, getStructuringElement(MORPH_RECT, Size(15, 15)));
	morphologyEx(src_gray, src_gray, MORPH_CLOSE, getStructuringElement(MORPH_RECT, Size(15, 15)));
	int colorTab[] = {0,100,200,255};
	int width = src.cols;
	int height = src.rows;
	int sampleCount = width * height;
	int clusterCount = 4;
	Mat points(sampleCount, 3, CV_32FC1);
	Mat labels;
	TermCriteria criteria(TermCriteria::EPS + TermCriteria::COUNT, 100, 0.1);
	int index = 0;
	for (int row = 0; row < height; row++)
	{
		for (int col = 0; col < width; col++)
		{
			index = row * width + col;
			uchar gray = src_gray.at<uchar>(row, col);
			points.at<float>(index, 0) = static_cast<int>(gray);
			points.at<float>(index, 1) = static_cast<int>(row);
			points.at<float>(index, 2) = static_cast<int>(col);
		}
	}

	kmeans(points, clusterCount, labels, criteria, 100, 0);

	Mat result = Mat::zeros(src_gray.size(), src_gray.type());
	for (int row = 0; row < height; row++)
	{
		for (int col = 0; col < width; col++)
		{
			index = row * width + col;
			int label = labels.at<int>(index, 0);
			result.at<uchar>(row, col) = colorTab[label];
		}
	}
	waitKey(0);
	return 0;
}

分为三类。结果使用imagewatch查看即可。
输入:原灰度图
Opencv4+VS2017实现灰度图片kmeans++聚类分割
输出:文章1结果
Opencv4+VS2017实现灰度图片kmeans++聚类分割
输出:本文结果
Opencv4+VS2017实现灰度图片kmeans++聚类分割
这才是大家实际想要的结果。

上一篇:win10下编译安装ncnn


下一篇:SAS ESCAPECHAR_2