OTSU (大津法)阈值选择算法

本文最后更新于:2022年7月4日 上午

大津法(OTSU)是一种确定图像二值化分割阈值的算法,由日本学者大津于1979年提出。从大津法的原理上来讲,该方法又称作最大类间方差法,因为按照大津法求得的阈值进行图像二值化分割后,前景与背景图像的类间方差最大。 本文记录相关内容。

简介

大津法(OTSU)是图像分割中阈值选取的最佳算法,计算简单,不受图像亮度和对比度的影响,因此在数字图像处理上得到了广泛的应用。它是按图像的灰度特性,将图像分成背景和前景两部分。因方差是灰度分布均匀性的一种度量,背景和前景之间的类间方差越大,说明构成图像的两部分的差别越大,当部分前景错分为背景或部分背景错分为前景都会导致两部分差别变小。因此,使类间方差最大的分割意味着错分概率最小。

  • 应用:是求图像全局阈值的最佳方法,应用不言而喻,适用于大部分需要求图像全局阈值的场合。

  • 优点:计算简单快速,不受图像亮度和对比度的影响。

  • 缺点:对图像噪声敏感;只能针对单一目标分割;当目标和背景大小比例悬殊、类间方差函数可能呈现双峰或者多峰,这个时候效果不好。

  • Opencv 官方文档

Otsu 原理

  • 核心思想:最大化被阈值分隔的像素类间方差
  • 假设存在图像$I$,色阶为 $0-L$ ,阈值 $TH=k$ 将图像所有像素分为两类 C1(小于TH) 和 C2(大于TH) ,则这两类像素各自的均值就为$m_1、m_2$,图像全局均值为$m_G$。同时像素被分为C1和C2类的概率分别为$p_1、p_2$。因此就有:
$$ p_1m_1+p_2m_2=m_G \tag{1} \label{1} $$ $$ p_1+p_2=1 \tag{2} \label{2} $$
  • 其中:
$$ \begin{array}{c} p_ 1=\sum_{i=0}^{k} p_{i}^‘ \\ m_ 1=1 / p_ 1 \sum_{i=0}^{k} i p_{i}^‘ \\ m _2=1 / p_ 2 \sum_{i=k+1}^{L-1} i p_{i}^‘ \end{array} \tag{3} \label{333} $$
  • 类间方差表达式为:

$$
\sigma^{2}=p_1(m_1-m_ G)^{2}+p_ 2(m_ 2-m _G)^{2} \tag{4} \label{3}
$$

  • 根据($\ref{1},\ref{2},\ref{3}$)式,有:

$$
\sigma^{2}=p_ 1 p_ 2(m _1-m _2)^{2} \tag{5} \label{5}
$$

  • 此时 $\sigma^{2}$ 即为阈值划分后的像素类间方差,遍历所有像素求得最大方差的阈值

事实上到此为止用着就很舒服了,但原文还做了进一步推导

  • 灰度级K的累加均值m图像全局均值mG分别为:
$$ \left. \begin{array} { l } { m = \sum _ { i = 0 } ^ { k } i p _ { i }^‘ } \\ { m_G = \sum _ { i = 0 } ^ { k - 1 } i p _ { i }^‘ } \end{array} \right. \tag{6} $$
  • 那么($\ref{333}$)式可变为:
$$ \begin{array}{c} m _1=1 / p_ 1 m \\ m_ 2=1 / p_ 2 (m_ G-m) \end{array} \tag{7} $$
  • 带入($\ref{5}$)式得:
$$ \sigma^{2} = \frac { ( m _G p_ 1 - m ) ^ { 2 } } { p_1 ( 1 - p _1 ) } $$
  • 遍历所有 $k$ 值得到最大 $\sigma^{2}$ 的阈值即可

实现

Opencv 接口

1
2
C++: double threshold(InputArray src, OutputArray dst, double thresh, double maxval, int type)
Python: cv2.threshold(src, thresh, maxval, type[, dst]) → retval, dst
  • 支持的默认方式为:

  • 使用 Otsu 时在参数 type 中填入 cv2.THRESH_OTSU 即可

示例代码

1
2
3
img = mt.cv_rgb_imread('img1.jpg')
res = cv2.threshold(img[:,:,0], 0, 255, cv2.THRESH_OTSU)
PIS(res[1])

参考资料


OTSU (大津法)阈值选择算法
https://www.zywvvd.com/notes/study/image-processing/otsu-thre/otsu-thre/
作者
Yiwei Zhang
发布于
2022年3月14日
许可协议