Python数据分析与机器学习-使用Kmeans进行图像压缩

源码下载:

http://download.csdn.net/download/adam_zs/10194644

from skimage import io
from sklearn.cluster import KMeans
import numpy as np

image = io.imread("test2.jpg")
io.imshow(image)
io.show()

# print(image.shape) #(154, 160, 3)  每个点由3个元素组成

rows = image.shape[0]
cols = image.shape[1]

image = image.reshape(image.shape[0] * image.shape[1], 3)
kmeans = KMeans(n_clusters=128)  # 把集合分成128个簇
kmeans.fit(image)

labels = np.asarray(kmeans.labels_, dtype=np.uint8)  # labels_ 每个点的标签
labels = labels.reshape(rows, cols)
io.imsave('compressed_test.jpg', labels)

image = io.imread("compressed_test.jpg")
io.imshow(image)
io.show()


发布了265 篇原创文章 · 获赞 421 · 访问量 47万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览