K-Mean 聚类中的质心

Question

我正在尝试对灰度图像进行 k 均值聚类。

代码如下：

I = im2double(imread('sample.png'));
J = rgb2gray(I);
r = size(J,1);
c = size(J,2);
J = reshape(J,r*c,1);

[cluster_idx, cluster_center] = kmeans(J,k,'start','uniform','distance','sqEuclidean', 'Replicates', 3);

这里 cluster_idx 有输入图像每一行的簇索引。并且 cluster_center 具有集群质心位置。但这将是一个 k x 1 矩阵。我不明白这些值（double）如何表示输入图像集群的质心？

Answer 1

cluster_center 表示每个簇的平均灰度级，而cluster_idx 为每个像素分配一个簇成员。

只需将reshape您的聚类结果恢复为原始图像大小，并使用cluster_idx作为cluster_center索引：

J = im2double(imread('cameraman.tif'));
r = size(J,1);
c = size(J,2);
J = reshape(J,r*c,1);
k = 3;
[cluster_idx, cluster_center] = kmeans(J,k,'start','uniform','distance','sqEuclidean', 'Replicates', 3);

II = reshape(cluster_idx,[r c]);
JJ = cluster_center(II);
imshow(JJ);