python opencv之分水岭算法示例


Posted in Python onFebruary 24, 2018

本文介绍了python opencv之分水岭算法示例,分享给大家,具体如下:

目标

  1. 使用分水岭算法对基于标记的图像进行分割
  2. 使用函数cv2.watershed()

原理:

灰度图像可以被看成拓扑平面,灰度值高的区域可以看出山峰,灰度值低的区域可以看成是山谷。向每一个山谷当中灌不同颜色的水。水位升高,不同山谷的水会汇合,为防止不同山谷的水汇合,小在汇合处建立起堤坝。然后继续灌水,然后再建立堤坝,直到山峰都掩模。构建好的堤坝就是图像的分割。

此方法通常会得到过渡分割的结果,因为图像中的噪声以及其他因素。为了减少此影响,opencv使用基于标记的分水岭算法,此算法要设置哪些山谷中的汇合点,哪些不是。这是一种交互式的图像分割算法那。我们要给已知对象打上不同表情。如果某个区域肯定是前景或对象,就使用某个颜色或灰度值标签标记它。如果是背景那么使用其他颜色进行标记,其余不能确定的部分用0标记。然后使用分水岭算法,每次灌水,标签会被更新,当两个不同颜色的标签相遇就会构建堤坝,知道所有山峰掩模,最后得到的边界对象值是-1。

代码:

对挨在一起的对象进行分割。

python opencv之分水岭算法示例

使用Otsu's 二值化后的结果为

python opencv之分水岭算法示例

要出去图像中的白噪声。可以使用形态学运算,使用闭运算去除对象中的空洞。

靠近对象中心的区域是前景,离对象远的区域是背景,不确定的区域是边界。

首先提取硬币区域,使用腐蚀操作去掉边缘,剩下的就是硬币。但硬币没有接触时,此方法有效,但是由于硬币相互接触,就要使用另外一种有效的方法:距离变换加上合适的阈值。

之后,要寻找不确定是否是硬币的区域。这里需要膨胀操作。膨胀操作会将对象边界延伸到背景当中。由于边界区域被去除,现在就能知道哪些区域是前景,哪些是背景。

余下的区域不知道如何区分,那么使用分水岭算法。这些区域通常是前景与背景的交界处。从能否确认是否是背景的区域中减去确定是前景的区域就得到了边界。

(前景和背景)

python opencv之分水岭算法示例

(上面的图是直接使用作者的代码后生产的结果,提取到了前景,为了演示一下不确定的区域,调了一下计算前景的距离变换的参数,使得中间出现不确定的区域)

python opencv之分水岭算法示例

这里面使用个cv2.distanceTransform函数

该函数用于计算2值图象中所有像素离其最近的值为0像素的近似距离。

参数为

cv2.distanceTransform(src, distanceType, maskSize[, dst]) → dst

#src为输入的二值图像。distanceType为计算距离的方式,可以是如下值
DIST_USER = ⑴, //!< User defined distance
DIST_L1  = 1, //!< distance = |x1-x2| + |y1-y2|
DIST_L2  = 2, //!< the simple euclidean distance
DIST_C  = 3, //!< distance = max(|x1-x2|,|y1-y2|)
DIST_L12  = 4, //!< L1-L2 metric: distance = 2(sqrt(1+x*x/2) - 1))
DIST_FAIR = 5, //!< distance = c^2(|x|/c-log(1+|x|/c)), c = 1.3998
DIST_WELSCH = 6, //!< distance = c^2/2(1-exp(-(x/c)^2)), c = 2.9846
DIST_HUBER = 7 //!< distance = |x|<c ? x^2/2 : c(|x|-c/2), c=1.345


#maskSize是蒙板尺寸,只有0,3,5
DIST_MASK_3  = 3, //!< mask=3
DIST_MASK_5  = 5, //!< mask=5
DIST_MASK_PRECISE = 0 //!< mask=0
import numpy as np
import cv2
from matplotlib import pyplot as plt
img = cv2.imread('21.jpg')
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)

kernel = np.ones((3,3),np.uint8)
opening = cv2.morphologyEx(thresh,cv2.MORPH_OPEN,kernel, iterations = 2)
# sure background area
sure_bg = cv2.dilate(opening,kernel,iterations=3)#膨胀
# Finding sure foreground area

dist_transform = cv2.distanceTransform(opening,1,5)
ret, sure_fg = cv2.threshold(dist_transform,0.2*dist_transform.max(),255,0)#参数改小了,出现不确定区域
# Finding unknown region
sure_fg = np.uint8(sure_fg)
unknown = cv2.subtract(sure_bg,sure_fg)#减去前景

cv2.imshow('p',sure_fg)
cv2.waitKey(0)

现在知道了那些背景是硬币,可以创建标签。(与原图像大小相同,数据类型为int32的数组)。

对于已经确定分类的区域,也就是背景和前景,使用整数标记,不确定的区域是用0标记。可以使用cv2.connectedComponents()函数来实现此功能。它会将背景标记为0,其他标记为位从1开始的正整数。

但是,如果背景标记为0,那么分水岭算法会将其当成位置区域,所以使用不同的整数进行标记,对于不确定的区域,函数标记为0.

结果使用JET颜色地图表示。深蓝色未知区域,硬币区域使用不同颜色。其余部分用浅蓝色。

使用分水岭算法

效果不错

python opencv之分水岭算法示例

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python处理文本文件实现生成指定格式文件的方法
Jul 31 Python
python sort、sorted高级排序技巧
Nov 21 Python
Python通过正则表达式选取callback的方法
Jul 18 Python
利用numpy实现一、二维数组的拼接简单代码示例
Dec 15 Python
对numpy中的数组条件筛选功能详解
Jul 02 Python
python实现微信机器人: 登录微信、消息接收、自动回复功能
Apr 29 Python
java判断三位数的实例讲解
Jun 10 Python
python调用webservice接口的实现
Jul 12 Python
PyQt5基本控件使用之消息弹出、用户输入、文件对话框的使用方法
Aug 06 Python
Python程序控制语句用法实例分析
Jan 14 Python
python 计算方位角实例(根据两点的坐标计算)
Jan 17 Python
Tensorflow 使用pb文件保存(恢复)模型计算图和参数实例详解
Feb 11 Python
python3爬取各类天气信息
Feb 24 #Python
python opencv之SIFT算法示例
Feb 24 #Python
python3 破解 geetest(极验)的滑块验证码功能
Feb 24 #Python
python opencv之SURF算法示例
Feb 24 #Python
几种实用的pythonic语法实例代码
Feb 24 #Python
使用Python爬取最好大学网大学排名
Feb 24 #Python
python opencv 直方图反向投影的方法
Feb 24 #Python
You might like
php实现的获取网站备案信息查询代码(360)
2013/09/23 PHP
thinkphp3.2实现在线留言提交验证码功能
2017/07/19 PHP
jquery 图片 上一张 下一张 链接效果(续篇)
2010/04/20 Javascript
基于PHP+Jquery制作的可编辑的表格的代码
2011/04/10 Javascript
JS按位非(~)运算符与~~运算符的理解分析
2011/07/31 Javascript
基于jQuery实现模拟页面加载进度条
2013/04/01 Javascript
Jquery通过JSON字符串创建JSON对象
2014/08/24 Javascript
jQuery焦点控制图层展示延迟隐藏的方法
2015/03/09 Javascript
jquery实现简单的无缝滚动
2015/04/15 Javascript
JS实现很实用的对联广告代码(可自适应高度)
2015/09/18 Javascript
基于MVC方式实现三级联动(JavaScript)
2017/01/23 Javascript
jQuery实现的页面弹幕效果【测试可用】
2018/08/17 jQuery
Vue CLI3中使用compass normalize的方法
2019/05/30 Javascript
layui问题之渲染数据表格时,仅出现10条数据的解决方法
2019/09/12 Javascript
关于在LayUI中使用AJAX提交巨坑记录
2019/10/25 Javascript
python实现批量转换文件编码(批转换编码示例)
2014/01/23 Python
Python的消息队列包SnakeMQ使用初探
2016/06/29 Python
Python之py2exe打包工具详解
2017/06/14 Python
pandas 使用均值填充缺失值列的小技巧分享
2019/07/04 Python
如何为Python终端提供持久性历史记录
2019/09/03 Python
修改 CentOS 6.x 上默认Python的方法
2019/09/06 Python
Python使用微信接入图灵机器人过程解析
2019/11/04 Python
Python中xml和dict格式转换的示例代码
2019/11/07 Python
python可视化 matplotlib画图使用colorbar工具自定义颜色
2020/12/07 Python
最便宜促销价格订机票:Airpaz(总部设在印尼,支持中文)
2018/11/13 全球购物
举例说明类变量和实例变量的区别
2016/06/30 面试题
《她是我的朋友》教学反思
2014/04/26 职场文书
岗位标兵事迹材料
2014/05/17 职场文书
员工薪酬激励方案
2014/06/13 职场文书
2014年派出所工作总结
2014/11/21 职场文书
单位租房协议书范本
2014/12/04 职场文书
2015年师德师风自我评价范文
2015/03/05 职场文书
大学生个人简历自我评价
2015/03/11 职场文书
休假证明书
2015/06/24 职场文书
js中Map和Set的用法及区别实例详解
2022/02/15 Javascript
SQL Server使用T-SQL语句批处理
2022/05/20 SQL Server