如何使用Python有效地识别图像中最主要颜色簇

import numpy as np
import cv2
from sklearn.cluster import KMeans

image = cv2.imread('image.jpg')

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

image = cv2.resize(image, (500, 500))

pixels = image.reshape(-1, 3)

k = 5  # 可根据需要调整聚类簇数
kmeans = KMeans(n_clusters=k)
kmeans.fit(pixels)

colors = kmeans.cluster_centers_

colors = colors.astype(int)

sorted_colors = sorted(colors, key=lambda x: np.linalg.norm(x))
main_colors = sorted_colors[-3:]  # 获取最主要的3个颜色簇

main_colors_image = np.zeros((100, 300, 3), dtype=np.uint8)
for i, color in enumerate(main_colors):
    main_colors_image[:, i * 100: (i + 1) * 100, :] = color
cv2.imshow('Main Colors', main_colors_image)
cv2.waitKey(0)
cv2.destroyAllWindows()