![](https://img2020.cnblogs.com/blog/1612966/202004/1612966-20200427122627434-995861376.png)
Kmeans
# coding: utf-8
import cv2
import numpy as np
img = cv2.imread("/home/hichens/Datasets/pic/11.jpg")
if len(img.shape) == 3:
data = img.reshape(-1, 3)
else:
data = img.reshape(-1, 1)
data = np.float32(data)
#定义中心 (type,max_iter,epsilon)
criteria = (cv2.TERM_CRITERIA_EPS +
cv2.TERM_CRITERIA_MAX_ITER, 10, 1.0)
#设置标签
flags = cv2.KMEANS_RANDOM_CENTERS
#K-Means聚类 聚集成4类
Ncenter = 2
compactness, labels, centers = cv2.kmeans(data, Ncenter, None, criteria, 10, flags)
#生成最终图像
res = centers[labels.flatten()]
dst = res.reshape(img.shape)
print(dst.shape)
dst = np.array(dst, dtype=np.uint8)
dst = cv2.cvtColor(dst, cv2.COLOR_BGR2RGB)
while True:
cv2.imshow("img", img)
cv2.imshow("dst", dst)
if cv2.waitKey(1) == 27:
break
cv2.destroyAllWindows()
![](https://img2020.cnblogs.com/blog/1612966/202004/1612966-20200427122645852-1206168254.png)
FCM
import cv2
import skfuzzy as fuzz
import numpy as np
img = cv2.imread('/home/hichens/Datasets/pic/11.jpg')
if len(img.shape) == 3:
data = img.reshape(-1, 3)
else:
data = img.reshape(-1, 1)
data = np.float32(data)
Ncenter = 4
cntr, u, _, _, _, _, fpc = fuzz.cluster.cmeans(
data.T, Ncenter, 2, error=0.05, maxiter=100, init=None)
labels = np.argmax(u, axis=0)
res = cntr[labels.flatten()]
dst = res.reshape(img.shape)
print(dst.shape)
print(fpc)
dst = np.array(dst, dtype=np.uint8)
dst = cv2.cvtColor(dst, cv2.COLOR_BGR2RGB)
while True:
cv2.imshow("img", img)
cv2.imshow("dst", dst)
if cv2.waitKey(1) == 27:
break
cv2.destroyAllWindows()
![](https://img2020.cnblogs.com/blog/1612966/202004/1612966-20200427123304829-1981436315.png)