机器学习 day06(三)

K-Means图片颜色点分类

1. 首先读取图片

和以往不同,这里使用数据集中的load_sample_image()函数

这是一个宽427,高640的427x640像素的三通道彩色图片
图片类型为uint8类型

2. 要把427*640个像素划分成64个聚类

将rgb值变为小数,便于计算
将三维数组变为二维数组,最后一个维度是彩色,不变,变前面两个维度

3. 由于像素点特别，没有办法全部研究，我们采取随机取样法

1. 将像素随机打乱
   
2. 随机取样1000个
   
3. 这个16个像素就是我们最终的聚类中心，我们用各个聚类聚类中心来代表聚类
   

4. 画一下16个聚类中心的颜色

5. 把labels和对应的像素点重新组合起来

1. 预测所整个图片中所有的像素点的聚类,把labels和对应的像素点重新组合起来
2. 对比图片
   
   和原图比起来,聚类后的第二张图只有16种颜色