Python使用OpenCV和K-Means聚类对毕业照进行图像分割


Posted in Python onJune 11, 2021

图像分割是将图像分割成多个不同区域(或片段)的过程。目标是将图像的表示变成更容易和更有意义的图像。

在这篇博客中,我们将看到一种图像分割方法,即K-Means Clustering

K-Means 聚类是一种无监督机器学习算法,旨在将N 个观测值划分为K 个聚类,其中每个观测值都属于具有最近均值的聚类。集群是指由于某些相似性而聚合在一起的数据点的集合。对于图像分割,这里的簇是不同的图像颜色。

我们使用的环境是pip install opencv-python numpy matplotlib

选择的图片是我们学校毕业照的图片,放心这里没有我,在学校公众号找的美图。

Python使用OpenCV和K-Means聚类对毕业照进行图像分割

导入所需模块:

import cv2
import numpy as np
import matplotlib.pyplot as plt
# read the image
image = cv2.imread("Graduation.jpg")

在进行图像分割之前,让我们将图像转换为RGB格式:

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

我们将使用cv2.kmeans()函数,它将一个2D数组作为输入,并且由于我们的原始图像是3D(宽度、高度和深度为3 个 RGB值),我们需要将高度和宽度展平为单个像素向量(3 个 RGB值):

# 将图像重塑为像素和3个颜色值(RGB)的2D数组
print(image.shape) #(853, 1280, 3)
pixel_values = image.reshape((-1, 3))
# 转换为numpy的float32
pixel_values = np.float32(pixel_values)
print(pixel_values.shape) #(1091840, 3)

关于opencv下的kmean算法,函数为cv2.kmeans()
函数的格式为:kmeans(data, K, bestLabels, criteria, attempts, flags)

data: 分类数据,最好是np.float32的数据,每个特征放一列。之所以是np.float32原因是这种数据类型运算速度快,如果是uint型数据将会很慢。

K: 分类数,opencv2的kmeans分类是需要已知分类数的。

bestLabels:预设的分类标签:没有的话 None

criteria:迭代停止的模式选择,这是一个含有三个元素的元组型数。格式为(type,max_iter,epsilon)max_iter迭代次数,epsilon结果的精确性

其中,type又有三种选择:

  • cv2.TERM_CRITERIA_EPS :精确度(误差)满足epsilon停止。
  • cv2.TERM_CRITERIA_MAX_ITER:迭代次数超过max_iter停止
  • cv2.TERM_CRITERIA_EPS+cv2.TERM_CRITERIA_MAX_ITER,两者合体,任意一个满足结束。
  • - attempts:重复试验kmeans算法次数,将会返回最好的一次结果

flags:初始类中心选择,两种方法

cv2.KMEANS_PP_CENTERS 算法kmeans++的center; cv2.KMEANS_RANDOM_CENTERS随机初始化

在这里,我们需要设置criteria确定停止标准。我们将在超过某些迭代次数(例如500)时停止,或者如果集群移动小于某个 epsilon 值(让我们在这里选择0.1),下面的代码在OpenCV 中定义了这个停止标准:

# 确定停止标准
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 500, 0.1)

上面图像,会发现五种主要颜色(分别是天空、草地、树、人的上身白,人的下身黑)

因此,我们将为这张图片使用K=5:

k = 5
_, labels, (centers) = cv2.kmeans(pixel_values, k, None, criteria, 10, cv2.KMEANS_RANDOM_CENTERS)

cv2.KMEANS_RANDOM_CENTERS只是指示OpenCV最初随机分配集群的值。

我们将扁平化的图像像素值转换为浮点数32类型,是因为cv2.kmeans() 浮点数32类型,然后,让我们将浮点数转换回8 位像素值np.uint8(centers)

# 转换回np.uint8
centers = np.uint8(centers)

# 展平标签阵列
labels = labels.flatten()

segmented_image = centers[labels.flatten()]

转换回原始图像形状并显示:

#重塑回原始图像尺寸
segmented_image = segmented_image.reshape(image.shape)
plt.imshow(segmented_image)
plt.show()

Python使用OpenCV和K-Means聚类对毕业照进行图像分割

当然,我们还可以禁用图像中的一些K-Means 聚类集群。例如,让我们禁用集群编号1并显示图像:

# 禁用2号群集(将像素变为黑色)
masked_image = np.copy(segmented_image)
# 转换为像素值向量的形状
masked_image = masked_image.reshape((-1, 3))
cluster1 = 1
masked_image[labels == cluster1] = [0, 0, 0]
# 转换回原始形状
masked_image = masked_image.reshape(image.shape)
plt.imshow(masked_image)
plt.show()

Python使用OpenCV和K-Means聚类对毕业照进行图像分割

原来K-Means 聚类2 号集群 是树。

请注意,还有其他分割技术,例如霍夫变换、轮廓检测和当前最先进的语义分割。

到此这篇关于Python使用OpenCV和K-Means聚类对毕业照进行图像分割的文章就介绍到这了,更多相关OpenCV和K-Means图像分割内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
Python连接数据库学习之DB-API详解
Feb 07 Python
开源软件包和环境管理系统Anaconda的安装使用
Sep 04 Python
使用Python+Splinter自动刷新抢12306火车票
Jan 03 Python
pandas 快速处理 date_time 日期格式方法
Nov 12 Python
基于python历史天气采集的分析
Feb 14 Python
Python Numpy 实现交换两行和两列的方法
Jun 26 Python
python多进程下实现日志记录按时间分割
Jul 22 Python
pytorch下大型数据集(大型图片)的导入方式
Jan 08 Python
tensorflow 查看梯度方式
Feb 04 Python
python pyecharts 实现一个文件绘制多张图
May 13 Python
C++和python实现阿姆斯特朗数字查找实例代码
Dec 07 Python
Python使用plt.boxplot()函数绘制箱图、常用方法以及含义详解
Aug 14 Python
Python3中PyQt5简单实现文件打开及保存
Jun 10 #Python
Python selenium的这三种等待方式一定要会!
python实现股票历史数据可视化分析案例
Python如何识别银行卡卡号?
使用python+pygame开发消消乐游戏附完整源码
Python数据可视化之基于pyecharts实现的地理图表的绘制
python使用PySimpleGUI设置进度条及控件使用
You might like
php中拷贝构造函数、赋值运算符重载
2012/07/25 PHP
基于PHP+Ajax实现表单验证的详解
2013/06/25 PHP
PHP可变变量学习小结
2015/11/29 PHP
PHP连接MSSQL方法汇总
2016/02/05 PHP
Zend Framework入门教程之Zend_Db数据库操作详解
2016/12/08 PHP
PHP实现基于栈的后缀表达式求值功能
2017/11/10 PHP
PHP chunk_split()函数讲解
2019/02/12 PHP
PHP实现读取文件夹及批量重命名文件操作示例
2019/04/15 PHP
JavaScript-世界上误解最深的语言分析
2007/08/12 Javascript
JQuery 确定css方框模型(盒模型Box Model)
2010/01/22 Javascript
理解Javascript_12_执行模型浅析
2010/10/18 Javascript
javascript中onmouse事件在div中失效问题的解决方法
2012/01/09 Javascript
jQuery Mobile的loading对话框显示/隐藏方法分享
2013/11/26 Javascript
javaScript对文字按照拼音排序实现代码
2013/12/27 Javascript
jquery操作下拉列表、文本框、复选框、单选框集合(收藏)
2014/01/08 Javascript
js实现checkbox全选和反选示例
2014/05/01 Javascript
排序算法的javascript实现与讲解(99js手记)
2014/09/28 Javascript
AngularJS入门教程之AngularJS表达式
2016/04/18 Javascript
Javascript 事件冒泡机制详细介绍
2016/10/10 Javascript
TypeScript学习之强制类型的转换
2016/12/27 Javascript
JS设计模式之命令模式概念与用法分析
2018/02/06 Javascript
深入解析vue 源码目录及构建过程分析
2019/04/24 Javascript
jquery操作select常见方法大全【7种情况】
2019/05/28 jQuery
解决vue bus.$emit触发第一次$on监听不到问题
2020/07/28 Javascript
python list转dict示例分享
2014/01/28 Python
python 字典(dict)遍历的四种方法性能测试报告
2014/06/25 Python
Python模块搜索路径代码详解
2018/01/29 Python
Python 实现的 Google 批量翻译功能
2019/08/26 Python
python清空命令行方式
2020/01/13 Python
TensorFlow Saver:保存和读取模型参数.ckpt实例
2020/02/10 Python
HTML5之多线程(Web Worker)
2019/01/02 HTML / CSS
实例讲解利用HTML5 Canvas API操作图形旋转的方法
2016/03/22 HTML / CSS
医院检讨书范文
2014/02/01 职场文书
教育实习指导教师评语
2014/12/31 职场文书
毕业典礼邀请函
2015/01/31 职场文书
带你学习MySQL执行计划
2021/05/31 MySQL