Opencv(二)—图像分割之分水岭算法

简介: 做图像处理时,我们可能会遇到一个问题:我们只需要图片的一部分区域,如何把图片中的某部分区域提取出来 或者 图像想要的区域用某种颜色(与其它区域颜色不一致)标记起来 ,以上描述的问题在像处理领域称为 图像分割

图像处理时,我们可能会遇到一个问题:我们只需要图片的一部分区域,如何把图片中的某部分区域提取出来 或者 图像想要的区域用某种颜色(与其它区域颜色不一致)标记起来 ,以上描述的问题在像处理领域称为 图像分割


说了这么多,可能还是有读者不知所云,这里我在下面放置了一张图,就是图像分割的一个应用,图像的前后对比,可以看到 人物 通过算法被很清晰地分割了出来,方便后续物体的识别跟踪。

这篇文章给大家介绍的就是 Opencv 中的一种图像分割方法 — 分水岭算法,先介绍一下这个算法所用到的基本原理:


把图像比喻成一个平面,图像灰度值高的区域被看作山峰,灰度值低的地方被看作山谷,不同区域的山谷我们可以用不同颜色来标记,但是随着标记区域不断扩大,会出现一种现象:不同山谷的交汇处区域会出现颜色错乱现象,为了防止这一现象的出现,要做的就是把高峰变得更高(改变灰度值),然后再用颜色标记,如此反复最后完成所所有山谷的颜色分割,以上就是 涉及分水岭算法基本原理。


说的简单一点,就是根据图像相邻的像素插值,分成不同区域;分水岭算法就是将不同区域染成不同颜色;分水岭分割算法最大的特征就是 区域的封闭性;


Opencv 包中的分水岭算法利用的不是原算法,而是在原算法基础上又改进了一下,加了一步预处理(因为原算法经常会造成图像过度分割):在分割之前先要设置那些山谷会出现汇合,那些不会;如果我i们能够确定该点代表的是要分割的对象,那么就用某个颜色或者灰度值标签标记它,如果不是就利用另一种颜色去标记它。随后的过程就是分水岭算法。当所有山谷区域都分割完毕之后,得到的边界对象值设置为 -1;


本次算法用到的图片实例

分水岭算法的具体使用步骤:

首先利用 cv2 读取图片,把 图片转变为灰质图,然后做阈值筛选,之前提到过的Otsu ’ s 二值化 ,变成下面的二值图片(两色图),至于如何二值化的,可以参照之前写的这篇文章:;代码如下:

import cv2
import numpy as np
from matplotlib import pyplot as plt
import cv2
import numpy as np
from matplotlib import pyplot as plt
img1 = cv2.imread('F:/opencv_test/1.jpg')
gray2 = cv2.cvtColor(img1,cv2.COLOR_BGR2RGB)
gray = cv2.cvtColor(gray2,cv2.COLOR_RGB2GRAY)
# Otsu ' s 二值化;
ret,thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV +cv2.THRESH_OTSU)
plt.subplot(1,3,1),
plt.imshow(gray2),
plt.title('Original')
plt.subplot(1,3,3),
plt.imshow(thresh),
plt.title("Otsu's")

现在我们的目标就转化为提取图像中上面的白色区域:这里需要记住的是 这里硬币区域的是前景,也就是里硬币中心区域越近前景的可能性越大,离硬币区域中心越远的为背景的可能性越大 ,模糊区域就是前景与背景之间的边缘区域 ;


提取硬币区域,目前比较有效的有两种操作:腐蚀 操作和 膨胀 操作,腐蚀是用来去除边缘像素,而膨胀操作是将硬币边缘区域向外延申,对于操作之后的区域我们就能确定哪个区域一定是前景( Foreground ),哪个区域一定是背景( Background ).


代码部分:

#nosing removoal迭代两次
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)
dist_transform = cv2.distanceTransform(opening,1,5)
ret,sure_fg = cv2.threshold(dist_transform,0.7*dist_transform.max(),255,0)
sure_fg = np.uint8(sure_fg)
unknow = cv2.subtract(sure_bg,sure_fg)
plt.subplot(1,3,1),
plt.imshow(sure_bg),
plt.title('Black region\n must be background')
plt.subplot(1,3,3),
plt.imshow(unknow),
plt.title('Yellos region\n must be foregroun'),

网络异常,图片无法展示
|

确定的背景区域减去一定确定的前景区域就是我们所要的硬币区域 ;分水岭算法在这里应用就是处理前景和背景的交界处,经过区域相减、阈值化处理之后得到肯定是硬币的区域,接下来需要我们创建标签(与源图像大小相同,数据类型为 int32 );


利用 cv2.connectedComponents() 把不同区域的颜色分为不同颜色,把背景标记为0(也就是代码中的unkbow区域),最后就是进行分水岭算法,把不同的区域赋予不同的颜色

#Marker labeling
ret,makers1 = cv2.connectedComponents(sure_fg)
#Add one to all labels so that sure background is not 0 but 1;
markers = makers1 +1
#Now mark the region of unknow with zero;
markers[unknow ==255] =0
markers3 = cv2.watershed(img1,markers)

上面提到,不同区域的边界,在进行分水岭分割算法之后都被赋值为 -1;把区域为 -1 的区域用红色表示,就能非常清楚地看到分界线了:

img1[markers3 == -1] =[255,0,0]
plt.subplot(1,3,1),
plt.imshow(makers1),
plt.title('makers1')
plt.subplot(1,3,2),
plt.imshow(markers3),
plt.title('markers3')
plt.subplot(1,3,3),
plt.imshow(img1),
plt.title('img1'),

最后的分割展示效果展示如下:

相关文章
|
4天前
|
算法 计算机视觉
OpenCV高斯差分技术实现图像边缘检测
OpenCV高斯差分技术实现图像边缘检测
|
6天前
|
计算机视觉
OpenCV图像运动模糊
OpenCV图像运动模糊
9 0
|
6天前
|
计算机视觉
OpenCV图像阈值
OpenCV图像阈值
5 0
|
6天前
|
计算机视觉
OpenCV图像混合
OpenCV图像混合
8 0
|
6天前
|
计算机视觉 Python
OpenCV为图像扩边(填充)
OpenCV为图像扩边(填充)
9 0
|
6天前
|
算法 TensorFlow 算法框架/工具
基于直方图的图像阈值计算和分割算法FPGA实现,包含tb测试文件和MATLAB辅助验证
这是一个关于图像处理的算法实现摘要,主要包括四部分:展示了四张算法运行的效果图;提到了使用的软件版本为VIVADO 2019.2和matlab 2022a;介绍了算法理论,即基于直方图的图像阈值分割,通过灰度直方图分布选取阈值来区分图像区域;并提供了部分Verilog代码,该代码读取图像数据,进行处理,并输出结果到"result.txt"以供MATLAB显示图像分割效果。
|
6天前
|
计算机视觉 Python
轻松掌握opencv的8种图像变换
轻松掌握opencv的8种图像变换
|
6天前
|
算法 计算机视觉
【OpenCV】- 图像修复
【OpenCV】- 图像修复
|
6天前
|
Serverless 计算机视觉
【OpenCV】-图像的矩
【OpenCV】-图像的矩
|
6天前
|
编解码 物联网 计算机视觉
【OpenCV】—图像金子塔与图片尺寸缩放
【OpenCV】—图像金子塔与图片尺寸缩放

热门文章

最新文章