Python案例 | 使用K-means 聚类算法提取图像中的颜色

假如我们需要提取下图中的颜色,可以通过使用 K-means 聚类算法对图像进行颜色聚类分析,并生成一个基于聚类中心(即最具代表性的颜色)的RGB值和调色板。

复制代码
# 通过使用 K-means 聚类算法对图像进行颜色聚类分析,并生成一个基于聚类中心(即最具代表性的颜色)的RGB值和调色板。
# 导入必要的库
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
from sklearn.cluster import KMeans

# 读取图像
image = mpimg.imread('poland-landscape.jpg')

# 获取图像尺寸
w, h, d = image.shape
pixels = image.reshape(w * h, d)

# K-means聚类,n_init设置为auto
n_colors = 15
kmeans = KMeans(n_clusters=n_colors, random_state=42, n_init='auto').fit(pixels)

# 获取聚类中心(调色板)
palette = np.uint8(kmeans.cluster_centers_)
print(palette)
# 显示调色板
plt.imshow([palette])
plt.axis('off')
plt.savefig('rgb_color_chart.png', dpi=300, bbox_inches='tight')
plt.show()

运行结果如下:

聚类得到的15个RGB值如下:

复制代码
[[  9  43  77]
 [ 70 165 209]
 [145 118  72]
 [ 47 122 160]
 [154 157 153]
 [ 11  26  38]
 [ 79  91  93]
 [212 164  37]
 [ 43  58  64]
 [231 227 217]
 [  6 125 184]
 [216 172 110]
 [ 95  78  39]
 [105 133 146]
 [ 20  78 118]]

参考资料:

https://www.douyin.com/note/7413545987140095241

相关推荐
快乐的钢镚子2 小时前
【leetcode hot 100】49.字母异位词分组
python·leetcode
honeysuckle_luo3 小时前
RandLA-net-pytorch 复现
人工智能·pytorch·python
SunnyDays10114 小时前
Python 高效实现 Excel 与 TXT 文本文件之间的数据转换
python·excel转txt·文本转excel·excel转文本·txt转excel
一念&4 小时前
每日一个C语言知识:C 头文件
c语言·开发语言·算法
硬件人某某某5 小时前
python基于卷积神经网络的桥梁裂缝检测系统(django),附可视化界面,源码
python·cnn·django
java1234_小锋5 小时前
PyTorch2 Python深度学习 - 自动微分(Autograd)与梯度优化
开发语言·python·深度学习·pytorch2
java1234_小锋5 小时前
PyTorch2 Python深度学习 - 简介以及入门
python·深度学习·pytorch2
tao3556677 小时前
【Python刷力扣hot100】42. Trapping Rain Water
开发语言·python·leetcode
一只安7 小时前
从零开发AI(不依赖任何模型)
人工智能·python
Miraitowa_cheems7 小时前
LeetCode算法日记 - Day 88: 环绕字符串中唯一的子字符串
java·数据结构·算法·leetcode·深度优先·动态规划