博客
关于我
kmeans图像分割
阅读量:80 次
发布时间:2019-02-26

本文共 1658 字,大约阅读时间需要 5 分钟。

 

# -*- coding: utf-8 -*-# @Author  : matthew# @Software: PyCharmimport cv2import matplotlib.pyplot as pltimport numpy as npdef seg_kmeans_gray():    img = cv2.imread('000129.jpg', cv2.IMREAD_GRAYSCALE)    # 展平    img_flat = img.reshape((img.shape[0] * img.shape[1], 1))    img_flat = np.float32(img_flat)    # 迭代参数    criteria = (cv2.TERM_CRITERIA_EPS + cv2.TermCriteria_MAX_ITER, 20, 0.5)    flags = cv2.KMEANS_RANDOM_CENTERS    # 聚类    compactness, labels, centers = cv2.kmeans(img_flat, 2, None, criteria, 10, flags)    # 显示结果    img_output = labels.reshape((img.shape[0], img.shape[1]))    plt.subplot(121), plt.imshow(img, 'gray'), plt.title('input')    plt.subplot(122), plt.imshow(img_output, 'gray'), plt.title('kmeans')    plt.show()if __name__ == '__main__':    seg_kmeans_gray()

 

# -*- coding: utf-8 -*-# @Author  : matthew# @Software: PyCharmimport cv2import matplotlib.pyplot as pltimport numpy as npdef seg_kmeans_color():    img = cv2.imread('000129.jpg', cv2.IMREAD_COLOR)    # 变换一下图像通道bgr->rgb,否则很别扭啊    b, g, r = cv2.split(img)    img = cv2.merge([r, g, b])    # 3个通道展平    img_flat = img.reshape((img.shape[0] * img.shape[1], 3))    img_flat = np.float32(img_flat)    # 迭代参数    criteria = (cv2.TERM_CRITERIA_EPS + cv2.TermCriteria_MAX_ITER, 20, 0.5)    flags = cv2.KMEANS_RANDOM_CENTERS    # 聚类    compactness, labels, centers = cv2.kmeans(img_flat, 2, None, criteria, 10, flags)    # 显示结果    img_output = labels.reshape((img.shape[0], img.shape[1]))    plt.subplot(121), plt.imshow(img), plt.title('input')    plt.subplot(122), plt.imshow(img_output, 'gray'), plt.title('kmeans')    plt.show()if __name__ == '__main__':    seg_kmeans_color()

 

转载地址:http://wjzk.baihongyu.com/

你可能感兴趣的文章
MySQL查询语句:揭秘专家秘籍,让你秒变数据库达人!
查看>>
mysql查询超时对PHP执行的影响
查看>>
mysql查询输出到excel文件_如何保存mysql查询输出到excel或.txt文件?
查看>>
mysql查询过程
查看>>
MySQL模拟Oracle序列sequence
查看>>
Mysql模糊查询like效率,以及更高效的写法
查看>>
MySQL死锁套路:一次诡异的批量插入死锁问题分析
查看>>
Mysql死锁问题Deadlock found when trying to get lock;try restarting transaction
查看>>
mysql每个数据库的最大连接数_MySQL数据库最大连接数
查看>>
Mysql流程控制结构,if函数、case结构、if结构、循环结构
查看>>
mysql添加外网访问权限
查看>>
mysql添加用户
查看>>
MySQL添加用户、删除用户与授权
查看>>
mysql添加用户及权限
查看>>
Mysql添加用户并授予只能查询权限
查看>>
mysql添加用户权限报1064 - You have an error in your SQL syntax问题解决
查看>>
mysql添加索引
查看>>
mysql添加表注释、字段注释、查看与修改注释
查看>>
mysql清理undo线程_MySQL后台线程的清理工作
查看>>
mysql清空带外键的表
查看>>