博客
关于我
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/

你可能感兴趣的文章
MySQL8修改密码报错ERROR 1819 (HY000): Your password does not satisfy the current policy requirements
查看>>
MySQL8找不到my.ini配置文件以及报sql_mode=only_full_group_by解决方案
查看>>
mysql8的安装与卸载
查看>>
mysqlbinlog报错unknown variable ‘default-character-set=utf8mb4‘
查看>>
mysqldump 导出中文乱码
查看>>
mysqldump备份时忽略某些表
查看>>
mysqldump实现数据备份及灾难恢复
查看>>
mysqlreport分析工具详解
查看>>
MySQL一个表A中多个字段关联了表B的ID,如何关联查询?
查看>>
MYSQL一直显示正在启动
查看>>
MySQL一站到底!华为首发MySQL进阶宝典,基础+优化+源码+架构+实战五飞
查看>>
MySQL万字总结!超详细!
查看>>
Mysql下载以及安装(新手入门,超详细)
查看>>
MySQL中B+Tree索引原理
查看>>
mysql中cast() 和convert()的用法讲解
查看>>
mysql中datetime与timestamp类型有什么区别
查看>>
mysql中floor函数的作用是什么?
查看>>
MySQL中group by 与 order by 一起使用排序问题
查看>>
mysql中having的用法
查看>>
MySQL中interactive_timeout和wait_timeout的区别
查看>>