Add files via upload

eastmountyxz · web-flow · commit 2ca45cbbe231 · 2021-08-20T17:31:38.000+08:00
diff --git a/blog44-ImageProcessingSummary/ImageProcessing_09_LH_KMeans.py b/blog44-ImageProcessingSummary/ImageProcessing_09_LH_KMeans.py
@@ -0,0 +1,44 @@
+# coding: utf-8
+# BY:Eastmount CSDN 2021-08-20
+import cv2
+import numpy as np
+import matplotlib.pyplot as plt
+
+#读取原始图像
+img = cv2.imread('nv.png') 
+
+#图像二维像素转换为一维
+data = img.reshape((-1,3))
+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类
+compactness, labels, centers = cv2.kmeans(data, 4, None, criteria, 10, flags)
+
+
+#图像转换回uint8二维类型
+centers = np.uint8(centers)
+res = centers[labels.flatten()]
+dst = res.reshape((img.shape))
+
+#图像转换为RGB显示
+img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+dst = cv2.cvtColor(dst, cv2.COLOR_BGR2RGB)
+
+#用来正常显示中文标签
+plt.rcParams['font.sans-serif']=['SimHei']
+
+#显示图像
+titles = ['原始图像', '聚类量化 K=4']  
+images = [img, dst]  
+for i in range(2):  
+   plt.subplot(1,2,i+1), plt.imshow(images[i], 'gray'), 
+   plt.title(titles[i])  
+   plt.xticks([]),plt.yticks([])  
+plt.show()
