K-Means图片颜色点分类
1. 首先读取图片
和以往不同,这里使用数据集中的load_sample_image()函数
这是一个宽427,高640的427x640像素的三通道彩色图片
图片类型为uint8类型
2. 要把427*640个像素划分成64个聚类
将rgb值变为小数,便于计算
将三维数组变为二维数组,最后一个维度是彩色,不变,变前面两个维度
3. 由于像素点特别,没有办法全部研究,我们采取随机取样法
- 将像素随机打乱
- 随机取样1000个
- 这个16个像素就是我们最终的聚类中心,我们用各个聚类聚类中心来代表聚类
4. 画一下16个聚类中心的颜色
5. 把labels和对应的像素点重新组合起来
- 预测所整个图片中所有的像素点的聚类,把labels和对应的像素点重新组合起来
- 对比图片
和原图比起来,聚类后的第二张图只有16种颜色