图片颜色聚类分析--KMeans

最新推荐文章于 2025-09-25 21:36:53 发布

原创

最新推荐文章于 2025-09-25 21:36:53 发布 · 4.3k 阅读

26 ·

CC 4.0 BY-SA版权

本文探讨了如何使用KMeans聚类算法对图片中的颜色进行分析，通过将颜色值聚类并用聚类中心替换，创建出具有不同颜色效果的新图片。分别展示了k值为2、16和32时的图像结果，进一步讨论了优化的可能性。

用聚类分析将图片中所有的颜色值做聚类。
得到的聚类中心的颜色替换原来位置的颜色，形成新的图片。

from skimage import io
import matplotlib.image as img
from sklearn.cluster import KMeans
import numpy as np
from matplotlib import pyplot as plt

image = io.imread("E:/picture.jpg")
X = image.reshape(-1,3)  #-1表示自动通过总量和列数3补齐行数，3为颜色三元素

# 颜色聚类划分
mod = KMeans(n_clusters = 16)  #聚类个数
labels = mod.fit_predict(X