• 文档
main content

使用 k 均值聚类实现基于颜色的分割 -凯发k8网页登录

此示例说明如何使用 k 均值聚类自动分割颜色。

聚类是一种分离对象组的方法。k 均值聚类将每个对象视为在空间中有一个位置。它将对象划分为若干分区，使每个簇中的对象尽可能彼此靠近，并尽可能远离其他簇中的对象。您可以使用 imsegkmeans 函数将图像像素按值分成一个颜色空间内的若干个簇。此示例在 rgb 和 l*a*b* 颜色空间中执行图像的 k 均值聚类，以显示使用不同颜色空间如何改进分割结果。

步骤 1：读取图像

在 hestain.png 中读取，这是一个带有苏木精和曙红染色组织 (h&e) 的图像。这种染色方法有助于病理学家区分染成蓝-紫色和粉红色的组织类型。

he = imread("hestain.png");
imshow(he)
title("h&e image")
text(size(he,2),size(he,1) 15, ...
     "image courtesy of alan partin, johns hopkins university", ...
     fontsize=7,horizontalalignment="right")

figure contains an axes object. the axes object with title h&e image contains 2 objects of type image, text.

步骤 2：用 k 均值聚类对 rgb 颜色空间的颜色进行分类

在 rgb 颜色空间中使用 k 均值聚类将图像分割成三个区域。对于输入图像中的每个像素，imsegkmeans 函数返回一个对应的簇标签。

将标注图像叠加显示在原始图像上。标注图像将白色、浅蓝-紫色和浅粉色区域组合在一起，这是不正确的。由于 rgb 颜色空间合并了每个通道（红、绿、蓝）内的亮度和颜色信息，因此两种不同颜色的较亮版本比这两种颜色的较暗版本更接近，也更难分割。

numcolors = 3;
l = imsegkmeans(he,numcolors);
b = labeloverlay(he,l);
imshow(b)
title("labeled image rgb")

figure contains an axes object. the axes object with title labeled image rgb contains an object of type image.

步骤 3：将图像从 rgb 颜色空间转换为 l*a*b* 颜色空间

l*a*b* 颜色空间将图像的光度和颜色分开。这使得按颜色分割区域变得更加容易并且与亮度无关。颜色空间也更符合人类对图像中不同的白色、蓝-紫色和粉色区域的视觉感知。

l*a*b* 颜色空间是从 cie xyz 三色值派生的。l*a*b* 空间包含光度层 l*、色度层 a*（表示颜色落在沿红-绿轴的位置）和色度层 b*（表示颜色落在沿蓝-黄轴的位置）。所有颜色信息都在 a* 和 b* 层。

使用 rgb2lab 函数将图像转换为 l*a*b* 颜色空间。

lab_he = rgb2lab(he);

步骤 4：用 k 均值聚类对基于 a*b* 空间的颜色进行分类

要仅使用颜色信息分割图像，请将图像限制为 lab_he 中的 a* 和 b* 值。将图像转换为 single 数据类型，以便于 imsegkmeans 函数使用。使用 imsegkmeans 函数将图像像素分成三个簇。将 numattempts 名称-值参数的值设置为使用不同的初始簇质心位置重复聚类三次，以避免拟合局部最小值。

ab = lab_he(:,:,2:3);
ab = im2single(ab);
pixel_labels = imsegkmeans(ab,numcolors,numattempts=3);

将标注图像叠加显示在原始图像上。新标注图像将白色、蓝紫色和粉色染色组织区域更清晰地区分开来。

b2 = labeloverlay(he,pixel_labels);
imshow(b2)
title("labeled image a*b*")

figure contains an axes object. the axes object with title labeled image a*b* contains an object of type image.

步骤 5：创建按颜色分割 h&e 图像的图像

使用 pixel_labels，您可以按颜色分离原始图像 hestain.png 中的对象，从而产生三个掩膜图像。

mask1 = pixel_labels == 1;
cluster1 = he.*uint8(mask1);
imshow(cluster1)
title("objects in cluster 1");

figure contains an axes object. the axes object with title objects in cluster 1 contains an object of type image.

mask2 = pixel_labels == 2;
cluster2 = he.*uint8(mask2);
imshow(cluster2)
title("objects in cluster 2");

figure contains an axes object. the axes object with title objects in cluster 2 contains an object of type image.

mask3 = pixel_labels == 3;
cluster3 = he.*uint8(mask3);
imshow(cluster3)
title("objects in cluster 3");

figure contains an axes object. the axes object with title objects in cluster 3 contains an object of type image.

步骤 6：分割核

簇 3 中仅包含蓝色对象。请注意，有深蓝色和浅蓝色对象。您可以使用 l*a*b* 颜色空间中的 l* 层来分离深蓝色和浅蓝色。细胞核为深蓝色。

l* 层包含每个像素的亮度值。提取此簇中像素的亮度值，并使用 imbinarize 函数用全局阈值对其设置阈值。掩膜 idx_light_blue 给出了浅蓝色像素的索引。 

l = lab_he(:,:,1);
l_blue = l.*double(mask3);
l_blue = rescale(l_blue);
idx_light_blue = imbinarize(nonzeros(l_blue));

复制蓝色对象的掩膜 mask3，然后从掩膜中删除浅蓝色像素。将新掩膜应用于原始图像并显示结果。只有深蓝色细胞核可见。

blue_idx = find(mask3);
mask_dark_blue = mask3;
mask_dark_blue(blue_idx(idx_light_blue)) = 0;
blue_nuclei = he.*uint8(mask_dark_blue);
imshow(blue_nuclei)
title("blue nuclei")

figure contains an axes object. the axes object with title blue nuclei contains an object of type image.

另请参阅

相关主题

    网站地图