K-means算法應用：圖片壓縮

from sklearn.datasets import load_sample_image
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt

china = load_sample_image("china.jpg")
plt.imshow(china)
plt.show()

 

讀取一張示例圖片或本身準備的圖片，觀察圖片存放數據特色。

import matplotlib.image as img
ge = img.imread('F:\\ge.jpg')
plt.imshow(ge)
plt.show()

plt.imshow(ge[:,:,0])

print(ge.shape)
ge

 

根據圖片的分辨率，可適當下降分辨率

ges = ge[::3,::3]
plt.imshow(ges)
plt.show()

from sklearn.datasets import load_sample_image
from sklearn.cluster import KMeans
import matplotlib.pyplot as plt

china = load_sample_image("china.jpg")
plt.imshow(china)
plt.show()

image = china[::3, ::3]
X = image.reshape(-1,3)
print(china.shape,image.shape,X.shape)

n_colors = 64
model = KMeans(n_colors)
labels = model.fit_predict(X)
colors = model.cluster_centers_

new_image=colors[labels]

new_image=new_image.reshape(image.shape) 

plt.imshow(new_image.astype(np.uint8))

 plt.show()

 

 

import sys
print(sys.getsizeof(china))
print(sys.getsizeof(new_image))

 

 

理解貝葉斯定理