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 GAE、Django导出Excel的方法
Nov 24 Python
Python ZipFile模块详解
Nov 01 Python
python获取当前时间对应unix时间戳的方法
May 15 Python
浅谈python中列表、字符串、字典的常用操作
Sep 19 Python
python如何实现int函数的方法示例
Feb 19 Python
Python实现的微信好友数据分析功能示例
Jun 21 Python
浅谈python 导入模块和解决文件句柄找不到问题
Dec 15 Python
pyqt实现.ui文件批量转换为对应.py文件脚本
Jun 19 Python
Pytorch中的variable, tensor与numpy相互转化的方法
Oct 10 Python
Python3+Requests+Excel完整接口自动化测试框架的实现
Oct 11 Python
Python自动化完成tb喵币任务的操作方法
Oct 30 Python
完美解决pyinstaller打包报错找不到依赖pypiwin32或pywin32-ctypes的错误
Apr 01 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
使用dump函数,给php加断点测试
2013/06/25 PHP
Linux下安装oracle客户端并配置php5.3
2014/10/12 PHP
composer.lock文件的作用
2016/02/03 PHP
window.location和document.location的区别分析
2008/12/23 Javascript
jQuery 操作XML入门
2008/12/25 Javascript
javascript parseInt 大改造
2009/09/27 Javascript
一个简单的JavaScript数据缓存系统实现代码
2010/10/24 Javascript
javascript获取URL参数与参数值的示例代码
2013/12/20 Javascript
jQuery如何取id有.的值一般的方法是取不到的
2014/04/18 Javascript
Javascript学习笔记之 对象篇(三) : hasOwnProperty
2014/06/24 Javascript
nodejs中实现路由功能
2014/12/29 NodeJs
在Ubuntu系统上安装Node.JS的教程
2015/10/15 Javascript
AngularJS中监视Scope变量以及外部调用Scope方法
2016/01/23 Javascript
微信小程序 location API接口详解及实例代码
2016/10/12 Javascript
Vue.js Ajax动态参数与列表显示实现方法
2016/10/20 Javascript
详解用vue.js和laravel实现微信授权登陆
2017/06/23 Javascript
Angular 2父子组件数据传递之@Input和@Output详解(下)
2017/07/05 Javascript
vue-router命名路由和编程式路由传参讲解
2019/01/19 Javascript
ES6基础之展开语法(Spread syntax)
2019/02/21 Javascript
Python中的MongoDB基本操作:连接、查询实例
2015/02/13 Python
python 地图经纬度转换、纠偏的实例代码
2018/08/06 Python
Python模拟浏览器上传文件脚本的方法(Multipart/form-data格式)
2018/10/22 Python
selenium python 实现基本自动化测试的示例代码
2019/02/25 Python
python语言基本语句用法总结
2019/06/11 Python
python 实现提取log文件中的关键句子,并进行统计分析
2019/12/24 Python
关于Pytorch的MLP模块实现方式
2020/01/07 Python
django 读取图片到页面实例
2020/03/27 Python
解决在keras中使用model.save()函数保存模型失败的问题
2020/05/21 Python
HTML5 Canvas自定义圆角矩形与虚线示例代码
2013/08/02 HTML / CSS
日本一家专门经营各种箱包的大型网站:Traveler Store
2016/08/03 全球购物
Exception类的常用方法
2012/06/16 面试题
考核评语大全
2014/04/29 职场文书
小学生五年级大队长竞选发言稿
2014/09/12 职场文书
医药公司采购员岗位职责
2015/04/03 职场文书
MySQL CHAR和VARCHAR该如何选择
2021/05/31 MySQL