Python opencv 机器学习 8. KMeans k值聚类颜色量化

最新推荐文章于 2024-05-03 22:52:43 发布

一头秀发的假程序猿

最新推荐文章于 2024-05-03 22:52:43 发布

阅读量930

点赞数

分类专栏： Python OpenCV 机器学习

原文链接：https://blog.csdn.net/fontthrone/article/details/72330737

版权

Python OpenCV 机器学习专栏收录该内容

8 篇文章 4 订阅

订阅专栏

转载自https://blog.csdn.net/fontthrone/article/details/72330737的第三个，加到自己这记录下来

import numpy as np
import cv2

img = cv2.imread('pic1.png')
Z = img.reshape((-1, 3))
# convert to np.float32
Z = np.float32(Z)
j = 0
# define criteria, number of clusters(K) and apply kmeans()
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10, 1.0)
Klist = [2, 4, 6, 8, 10]
for i in Klist:
    ret, label, center = cv2.kmeans(Z, i, None, criteria, 10, cv2.KMEANS_RANDOM_CENTERS)
    j += 2
    # Now convert back into uint8, and make original image
    center = np.uint8(center)
    res = center[label.flatten()]
    res2 = res.reshape((img.shape))
    cv2.imshow(str(("spaceship K=", i)), res2)
    cv2.waitKey(0)
cv2.imshow('quondam image', img)
cv2.waitKey(0)
cv2.destroyAllWindows()