scikit-image 中用于图像分割的阈值算法

本文涉及的产品
视觉智能开放平台,图像资源包5000点
视觉智能开放平台,分割抠图1万点
视觉智能开放平台,视频资源包5000点
简介: 图像阈值技术是许多计算机视觉应用中的关键步骤。在本文中,我们将利用 scikit-image 实现阈值技术。

scikit-image 简介与安装

由于图像阈值技术是许多计算机视觉应用中的关键步骤,因此许多算法库中都包含阈值技术,其中就包括 scikit-image。scikit-image 是用于图像处理的算法包,其详细介绍可以参考官方网站,由 scikit-image 操作的图像需要先转换为 NumPy 数组。

在本文中,我们将利用 scikit-image 实现阈值技术。如果还没有安装 scikit-image 库,首先需要使用以下命令安装 scikit-image:

pip install scikit-image

使用 scikit-image 进行阈值处理

为了在 scikit-image 中使用阈值算法,我们以 Otsu 的二值化算法对测试图像进行阈值处理为例进行介绍。第一步是导入所需的包:

fromskimage.filtersimportthreshold_otsufromskimageimportimg_as_ubyte

然后使用 scikit-image 应用 Otsu 的二值化算法:

# 加载图像image=cv2.imread('example.png')
gray_image=cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
hist=cv2.calcHist([gray_image], [0], None, [256], [0, 256])
# 基于 otsu 方法的返回阈值thresh=threshold_otsu(gray_image)
# 生成布尔数组:binary=gray_image>thresh# 转换为uint8数据类型binary=img_as_ubyte(binary)

threshold_otsu(gray_image) 函数根据 Otsu 的二值化算法返回阈值。之后,使用此值构建二进制图像( dtype= bool ),最后应将其转换为 8 位无符号整数格式( dtype=uint8 )以进行可视化,使用 img_as_ubyte() 函数完成转换过程。

程序输出如下图所示:绘图158.png

接下来,介绍下如何使用 scikit-image 中的一些其它阈值技术。

scikit-image 中的其他阈值技术

接下来,将对比 Otsu、triangle、Niblack 和 Sauvola 阈值技术进行阈值处理的不同效果。Otsu 和 triangle 是全局阈值技术,而 Niblack 和 Sauvola 是局部阈值技术。当背景不均匀时,局部阈值技术则是更好的方法阈值处理方法。

同样的,第一步是导入所需的包:

fromskimage.filtersimportthreshold_otsu, threshold_triangle, threshold_niblack, threshold_sauvolafromskimageimportimg_as_ubyteimportcv2importmatplotlib.pyplotasplt

调用每个阈值方法( threshold_otsu()、threshold_niblack()、threshold_sauvola() 和 threshold_triangle() ),以使用 scikit-image 对比执行阈值操作:

# Otsuthresh_otsu=threshold_otsu(gray_image)
binary_otsu=gray_image>thresh_otsubinary_otsu=img_as_ubyte(binary_otsu)
# Niblackthresh_niblack=threshold_niblack(gray_image, window_size=25, k=0.8)
binary_niblack=gray_image>thresh_niblackbinary_niblack=img_as_ubyte(binary_niblack)
# Sauvolathresh_sauvola=threshold_sauvola(gray_image, window_size=25)
binary_sauvola=gray_image>thresh_sauvolabinary_sauvola=img_as_ubyte(binary_sauvola)
# trianglethresh_triangle=threshold_triangle(gray_image)
binary_triangle=gray_image>thresh_trianglebinary_triangle=img_as_ubyte(binary_triangle)

程序输出如下图所示:

绘图159.png

如上图所示,当图像不均匀时,局部阈值方法可以提供更好的结果。因此,可以将这些局部阈值方法应用于文本识别。

最后,我们了解一个更加有趣的阈值算法—— Multi-Otsu 阈值技术,其可用于将输入图像的像素分为多个不同的类别,每个类别根据图像内灰度的强度计算获得。

Multi-Otsu 根据所需类别的数量计算多个阈值,默认类数为 3,此时将获得三个类别,算法返回两个阈值,由直方图中的红线表示。

importmatplotlibimportnumpyasnpimportcv2fromskimageimportdatafromskimage.filtersimportthreshold_multiotsuimage=cv2.imread('8.png')
image=cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
# 使用默认值调用 threshold_multiotsu()thresholds=threshold_multiotsu(image)
regions=np.digitize(image, bins=thresholds)
defshow_img_with_matplotlib(img, title, pos, cmap):
ax=plt.subplot(1, 3, pos)
plt.imshow(img, cmap=cmap)
plt.title(title, fontsize=8)
plt.axis('off')
fig, ax=plt.subplots(nrows=1, ncols=3, figsize=(10, 3.5))
# 绘制灰度图像show_img_with_matplotlib(image, 'Original', 1, cmap='gray')
# 可视化直方图ax[1].hist(image.ravel(), bins=255)
ax[1].set_title('Histogram')
forthreshinthresholds:
ax[1].axvline(thresh, color='r')
# 可视化 Multi Otsu 结果show_img_with_matplotlib(regions, 'Multi-Otsu result', 3, cmap='jet')
plt.subplots_adjust()
plt.show()

绘图160.png

可以通过修改 threshold_multiotsu 的 classes 参数来改变类别数,以观察不同效果:

# 将类别数修改为4thresholds=threshold_multiotsu(image, classes=3)
regions=np.digitize(image, bins=thresholds)

绘图161.png

scikit-image 也提供了其他更多的阈值技术,可以参阅 API 文档,以查看所有可用方法。

小结

本文中,我们介绍了如何使用 scikit-image 中的不同阈值算法,包括两种全局阈值技术( Otsu 和 三角形二值算法)和两种局部阈值技术( Niblack 和 Sauvola 算法)。

相关文章
|
4月前
|
机器学习/深度学习 算法 数据挖掘
【Python机器学习】Mean Shift、Kmeans聚类算法在图像分割中实战(附源码和数据集)
【Python机器学习】Mean Shift、Kmeans聚类算法在图像分割中实战(附源码和数据集)
91 0
【Python机器学习】Mean Shift、Kmeans聚类算法在图像分割中实战(附源码和数据集)
|
4月前
|
算法 TensorFlow 算法框架/工具
基于直方图的图像阈值计算和分割算法FPGA实现,包含tb测试文件和MATLAB辅助验证
这是一个关于图像处理的算法实现摘要,主要包括四部分:展示了四张算法运行的效果图;提到了使用的软件版本为VIVADO 2019.2和matlab 2022a;介绍了算法理论,即基于直方图的图像阈值分割,通过灰度直方图分布选取阈值来区分图像区域;并提供了部分Verilog代码,该代码读取图像数据,进行处理,并输出结果到"result.txt"以供MATLAB显示图像分割效果。
|
4月前
|
算法 计算机视觉 索引
基于四叉树的图像分割算法matlab仿真
基于四叉树的图像分割算法matlab仿真
基于四叉树的图像分割算法matlab仿真
|
4月前
|
监控 算法 计算机视觉
基于FPGA的图像自适应阈值二值化算法实现,包括tb测试文件和MATLAB辅助验证
基于FPGA的图像自适应阈值二值化算法实现,包括tb测试文件和MATLAB辅助验证
|
4月前
|
算法 数据挖掘
[Halcon&图像] 阈值分割算法拓展
[Halcon&图像] 阈值分割算法拓展
97 1
|
4月前
|
机器学习/深度学习 算法 计算机视觉
【计算机视觉】图像分割中FCN、DeepLab、SegNet、U-Net、Mask R-CNN等算法的讲解(图文解释 超详细)
【计算机视觉】图像分割中FCN、DeepLab、SegNet、U-Net、Mask R-CNN等算法的讲解(图文解释 超详细)
135 0
|
4月前
|
算法
[Halcon&图像] 图像阈值分割算法汇总
[Halcon&图像] 图像阈值分割算法汇总
162 0
|
16天前
|
算法 BI Serverless
基于鱼群算法的散热片形状优化matlab仿真
本研究利用浴盆曲线模拟空隙外形,并通过鱼群算法(FSA)优化浴盆曲线参数,以获得最佳孔隙度值及对应的R值。FSA通过模拟鱼群的聚群、避障和觅食行为,实现高效全局搜索。具体步骤包括初始化鱼群、计算适应度值、更新位置及判断终止条件。最终确定散热片的最佳形状参数。仿真结果显示该方法能显著提高优化效率。相关代码使用MATLAB 2022a实现。
|
16天前
|
算法 数据可视化
基于SSA奇异谱分析算法的时间序列趋势线提取matlab仿真
奇异谱分析(SSA)是一种基于奇异值分解(SVD)和轨迹矩阵的非线性、非参数时间序列分析方法,适用于提取趋势、周期性和噪声成分。本项目使用MATLAB 2022a版本实现从强干扰序列中提取趋势线,并通过可视化展示了原时间序列与提取的趋势分量。代码实现了滑动窗口下的奇异值分解和分组重构,适用于非线性和非平稳时间序列分析。此方法在气候变化、金融市场和生物医学信号处理等领域有广泛应用。
|
17天前
|
资源调度 算法
基于迭代扩展卡尔曼滤波算法的倒立摆控制系统matlab仿真
本课题研究基于迭代扩展卡尔曼滤波算法的倒立摆控制系统,并对比UKF、EKF、迭代UKF和迭代EKF的控制效果。倒立摆作为典型的非线性系统,适用于评估不同滤波方法的性能。UKF采用无迹变换逼近非线性函数,避免了EKF中的截断误差;EKF则通过泰勒级数展开近似非线性函数;迭代EKF和迭代UKF通过多次迭代提高状态估计精度。系统使用MATLAB 2022a进行仿真和分析,结果显示UKF和迭代UKF在非线性强的系统中表现更佳,但计算复杂度较高;EKF和迭代EKF则更适合维数较高或计算受限的场景。