opencv kmeans (C++)

kmeans

函数原型

double cv::kmeans(
	InputArray 	data,
	int 	K,
	InputOutputArray 	bestLabels,
	TermCriteria 	criteria,
	int 	attempts,
	int 	flags,
	OutputArray 	centers = noArray()
)

参数说明

  • Parameters

    data待聚类的数据集,数据集的每一个样本是一个N维的点,点坐标都是float型的,例如:有m个样本,每个样本有n个维度,那data的格式就为cv::Mat dataSet(m,n,CV_32F)
    K聚类数,即要把数据集聚成k类.
    bestLabels存储data中每一个样本的标签,数据类型为int型
    criteriaopencv中迭代算法的终止条件,例如迭代的次数限制,或者迭代的精度达到要求时,算法迭代终止
    attempts使用不同的初始聚类中心执行算法的次数
    flagscv::KmeansFlags见下表,选择聚类中心的初始化方式
    centersOutput matrix of the cluster centers, one row per each cluster center.
  • cv::KmeansFlags

KMEANS_RANDOM_CENTERS Python: cv.KMEANS_RANDOM_CENTERSSelect random initial centers in each attempt.
KMEANS_PP_CENTERS Python: cv.KMEANS_PP_CENTERSUse kmeans++ center initialization by Arthur and Vassilvitskii [Arthur2007].
KMEANS_USE_INITIAL_LABELS Python: cv.KMEANS_USE_INITIAL_LABELSDuring the first (and possibly the only) attempt, use the user-supplied labels instead of computing them from the initial centers. For the second and further attempts, use the random or semi-random centers. Use one of KMEANS_*_CENTERS flag to specify the exact method.

示例

读取一张图片,把图片中每一个像素点的RGB值作为特征进行聚类(颜色量化),聚类数目根据需要进行调整。

#include "opencv.hpp"


int kmeansDemo(cv::Mat &srcImage, cv::Mat &dst, int clusterCount)
{
	if (srcImage.empty())
		return -1;
	if (clusterCount <= 0)
		return -1;

	//cv::GaussianBlur(srcImage, srcImage, cv::Size(0, 0), 2);
	int width = srcImage.cols;
	int height = srcImage.rows;

	//init
	int sampleCount = width * height;
	cv::Mat labels;//Input/output integer array that stores the cluster indices for every sample
	cv::Mat centers;//Output matrix of the cluster centers, one row per each cluster center.

	// convert image to kmeans data
	cv::Mat sampleData = srcImage.reshape(3, sampleCount);//every pixel is a sample
	cv::Mat data;
	sampleData.convertTo(data, CV_32F);

	//K-Means
	cv::TermCriteria criteria = cv::TermCriteria(cv::TermCriteria::EPS + cv::TermCriteria::COUNT, 5, 0.1);
	cv::kmeans(data, clusterCount, labels, criteria, clusterCount, cv::KMEANS_PP_CENTERS, centers);

	//create a color map
	std::vector<cv::Scalar> colorMaps;
	uchar b, g, r;;
	//clusterCount is equal to centers.rows
	for (int i = 0; i < centers.rows; i++)
	{
		b = (uchar)centers.at<float>(i, 0);
		g = (uchar)centers.at<float>(i, 1);
		r = (uchar)centers.at<float>(i, 2);
		colorMaps.push_back(cv::Scalar(b, g, r));
	}
	// Show  result
	int index = 0;
	dst = cv::Mat::zeros(srcImage.size(), srcImage.type());
	uchar *ptr=NULL;
	int *label = NULL;
	for (int row = 0; row < height; row++) {
		ptr = dst.ptr<uchar>(row);
		for (int col = 0; col < width; col++) {
			index = row * width + col;
			label = labels.ptr<int>(index);
			*(ptr + col * 3) = colorMaps[*label][0];
			*(ptr + col * 3 + 1) = colorMaps[*label][1];
			*(ptr + col * 3 + 2) = colorMaps[*label][2];
		}
	}
		
	return 0;
}

int main()
{
	int clusterCount = 8;//the number of clusters
	std::string path = "K:\\deepImage\\fruit.jpg";
	cv::Mat srcImage = cv::imread(path);
	cv::imshow("srcImage", srcImage);
	cv::Mat dst;
	
	kmeansDemo(srcImage,dst,clusterCount);

	std::string txt = "clusters:" + std::to_string(clusterCount);
	cv::putText(dst, txt, cv::Point(5, 35), 0, 1, cv::Scalar(0, 255, 250), 2);
	cv::imshow("result", dst);
	cv::waitKey(0);
	return 0;
}
  • 效果

聚类数为8时的效果

聚类数为8

聚类数为16

  • 6
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
C++中使用OpenCV进行kmeans聚类的代码示例如下: ```cpp #include <opencv2/opencv.hpp> #include <iostream> using namespace cv; using namespace std; int main() { Mat img = imread("image.jpg"); Mat samples(img.rows * img.cols, 3, CV_32F); for (int y = 0; y < img.rows; y++) { for (int x = 0; x < img.cols; x++) { for (int z = 0; z < 3; z++) { samples.at<float>(y + x * img.rows, z) = img.at<Vec3b>(y, x)\[z\]; } } } int clusterCount = 8; Mat labels; int attempts = 5; Mat centers; kmeans(samples, clusterCount, labels, TermCriteria(TermCriteria::EPS + TermCriteria::COUNT, 10, 1.0), attempts, KMEANS_RANDOM_CENTERS, centers); Mat new_image(img.size(), img.type()); for (int y = 0; y < img.rows; y++) { for (int x = 0; x < img.cols; x++) { int cluster_idx = labels.at<int>(y + x * img.rows, 0); new_image.at<Vec3b>(y, x)\[0\] = centers.at<float>(cluster_idx, 0); new_image.at<Vec3b>(y, x)\[1\] = centers.at<float>(cluster_idx, 1); new_image.at<Vec3b>(y, x)\[2\] = centers.at<float>(cluster_idx, 2); } } imshow("Original Image", img); imshow("Clustered Image", new_image); waitKey(0); return 0; } ``` 这段代码首先读取一张图像,然后将图像的像素值重塑为一个Mx3大小的数组,其中M是图像中的像素数。接下来,使用kmeans函数对像素值进行聚类,指定聚类的数量为8。最后,根据聚类结果将像素值替换为对应的聚类中心值,生成聚类后的图像。 #### 引用[.reference_title] - *1* *2* [[C++] Kmeans算法实现](https://blog.csdn.net/weixin_44179561/article/details/126208335)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [OpenCV52:OpenCV中的Kmeans聚类](https://blog.csdn.net/uncle_ll/article/details/122608227)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值