博客
关于我
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学习总结(29)——MySQL中CHAR和VARCHAR
查看>>
Mysql学习总结(2)——Mysql超详细Window安装教程
查看>>
Mysql学习总结(30)——MySQL 索引详解大全
查看>>
Mysql学习总结(31)——MySql使用建议,尽量避免这些问题
查看>>
Mysql学习总结(33)——阿里云centos配置MySQL主从复制
查看>>
Mysql学习总结(35)——Mysql两千万数据优化及迁移
查看>>
Mysql学习总结(36)——Mysql查询优化
查看>>
Mysql学习总结(37)——Mysql Limit 分页查询优化
查看>>
Mysql学习总结(38)——21条MySql性能优化经验
查看>>
Mysql学习总结(39)——49条MySql语句优化技巧
查看>>
Mysql学习总结(3)——MySql语句大全:创建、授权、查询、修改等
查看>>
Mysql学习总结(40)——MySql之Select用法汇总
查看>>
Mysql学习总结(41)——MySql数据库基本语句再体会
查看>>
Mysql学习总结(42)——MySql常用脚本大全
查看>>
Mysql学习总结(43)——MySQL主从复制详细配置
查看>>
Mysql学习总结(44)——Linux下如何实现mysql数据库每天自动备份定时备份
查看>>
Mysql学习总结(45)——Mysql视图和事务
查看>>
Mysql学习总结(46)——8种常被忽视的SQL错误用法
查看>>
Mysql学习总结(48)——MySql的日志与备份还原
查看>>
Mysql学习总结(49)——从开发规范、选型、拆分到减压
查看>>