机器学习-学习记录:KMeans聚类算法实现图像分割
原图片:代码:import numpy as npimport matplotlib.pyplot as pltfrom skimage import io as iofrom sklearn.cluster import KMeansimport pandas as pdimg = io.imread('D:\\1.jpg')plt.imshow(img)plt.show()# 查看数据结果和维
·
原图片:
代码:
import numpy as np
import matplotlib.pyplot as plt
from skimage import io as io
from sklearn.cluster import KMeans
import pandas as pd
img = io.imread('D:\\1.jpg')
plt.imshow(img)
plt.show()
# 查看数据结果和维度
print(type(img))
print(img.shape)
# 维度存储
img_width = img.shape[1]
img_height = img.shape[0]
# 数据维度转换
img_data = img.reshape(-1,3)
print(img_data.shape)
# 模型建立与训练
model = KMeans(n_clusters=2,random_state=0)
model.fit(img_data)
# 结果预测
label = model.predict(img_data)
print(label)
print(pd.value_counts(label))
# 结果数据维度转换
label = label.reshape([img_height,img_width])
print(label)
print(label.shape)
# 后续灰度处理
label = 1/(label+1)
print(label)
# 结果可视化
plt.imshow(label)
plt.show()
# 图像存储到本地
io.imsave('result_kmeans.png',label)
结果图片:
K值得不同,结果图片是不同的,可以试着修改K值,看一下结果图片的不同之初。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)