Create blog19-image01.py

eastmountyxz · web-flow · commit a30867ab6a95 · 2019-10-17T23:44:38.000+08:00
diff --git a/blog19-kmeans/blog19-image01.py b/blog19-kmeans/blog19-image01.py
@@ -0,0 +1,40 @@
+# coding: utf-8
+import cv2
+import numpy as np
+import matplotlib.pyplot as plt
+
+#读取原始图像灰度颜色
+img = cv2.imread('scenery.png', 0) 
+print img.shape
+
+#获取图像高度、宽度
+rows, cols = img.shape[:]
+
+#图像二维像素转换为一维
+data = img.reshape((rows * cols, 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类
+compactness, labels, centers = cv2.kmeans(data, 4, None, criteria, 10, flags)
+
+#生成最终图像
+dst = labels.reshape((img.shape[0], img.shape[1]))
+
+#用来正常显示中文标签
+plt.rcParams['font.sans-serif']=['SimHei']
+
+#显示图像
+titles = [u'原始图像', u'聚类图像']  
+images = [img, dst]  
+for i in xrange(2):  
+   plt.subplot(1,2,i+1), plt.imshow(images[i], 'gray'), 
+   plt.title(titles[i])  
+   plt.xticks([]),plt.yticks([])  
+plt.show()
