1. 什么是阈值分割
阈值分割是根据图像的灰度特征按照设定的阈值将图像分割成不同的子区域。简单的理解就是先将图像进行灰度处理,然后根据灰度值和设定的灰度范围将图像灰度分类。比如0-128的是一类,129-255是一类。 根据不同的分类方法,阈值分割有以下7种方法:
- 固定阈值分割
- 直方图双峰法
- 迭代阈值图像分割
- 自适应阈值图像分割
- 大津法 OTSU
- 均值法
- 最佳阈值
2. 固定阈值分割
固定阈值分割是最简单的阈值分割方法,其方法就是将灰度值大于某一阈值的像素点置为255,而小于等于该阈值的点设置为0。这是最简单的图像分割方法,适用范围很窄。对于比较简单的图像有较好的效果。对应的CV2函数为cv2.threshold(src, thresh, maxval,cv2.THRESH_BINARY)。用numpy实现此功能非常简单,只需要1行代码:
return np.where(((img>thresh) & (img<maxval)),255,0)
np.where的第一个参数是条件,可以设置多条件。第二个是条件成立时,numpy数组的取值,而第三个参数是不成立的取值。 完成的比较代码如下:
import cv2
import numpy as np
import matplotlib.pyplot as plt
def fix_threshold(img, thresh, maxval=255):
return np.where(((img > thresh) & (img < maxval)), 255, 0)
img = cv2.imread('xk.png')
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
img_gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
ret, th = cv2.threshold(img_gray, 127, 255, cv2.THRESH_BINARY)
fix = fix_threshold(img_gray, 127, 255)
plt.subplot(131), plt.imshow(img_gray, cmap='gray')
plt.title('Original Image'), plt.xticks([]), plt.yticks([])
plt.subplot(132), plt.imshow(th, cmap='gray')
plt.title('CV2 Image1'), plt.xticks([]), plt.yticks([])
plt.subplot(133), plt.imshow(fix, cmap='gray')
plt.title('Fix Image2'), plt.xticks([]), plt.yticks([])
plt.show()
3.灰度直方图双峰法
灰度直方图通常双峰属性(bimodal,一个前景峰值,另一个为背景峰值)。两个峰值之间的最小值可以认为是最优二值化的分界点。 算法如下: (1)查找直方图最大值firstPeak (2)通过下面的公式结算第二峰:
s
e
c
o
n
d
P
e
a
k
=
max
?
x
[
(
x
?
f
r
i
s
t
P
e
a
k
)
2
?
h
i
s
t
(
x
)
]
secondPeak = \max\limits_{x}[(x-fristPeak)^2*hist(x)]
secondPeak=xmax?[(x?fristPeak)2?hist(x)] (3)双峰之间的灰度最小值为分界点(需要考虑最大峰值的灰度数值可能小于第二大峰值的灰度值) 此方法得到的阈值与CV2中threshold函数得到的阈值并不是完全一致的。本方法较为简单,对于处理简单的图像比较有效,但是对于复杂的图片,比如遥感图片效果就很差。后面的文章会增加解读遥感图片的部分。
import cv2
import numpy as np
from matplotlib import pyplot as plt
def GrayHist(img):
grayHist = np.zeros(256,dtype=np.uint64)
for v in range(256):
grayHist[v] = np.sum(img==v)
return grayHist
def threshTwoPeaks(image):
hist = GrayHist(image)
maxLoc = np.where(hist == np.max(hist))
firstPeak = maxLoc[0][0]
elementList = np.arange(256,dtype = np.uint64)
measureDists = np.power(elementList - firstPeak,2) * hist
maxLoc2 = np.where(measureDists == np.max(measureDists))
secondPeak = maxLoc2[0][0]
thresh = 0
if secondPeak > firstPeak:
firstPeak,secondPeak=secondPeak,firstPeak
temp = hist[secondPeak:firstPeak]
minloc = np.where(temp == np.min(temp))
thresh = secondPeak + minloc[0][0] + 1
threshImage_out = image.copy()
threshImage_out[threshImage_out > thresh] = 255
threshImage_out[threshImage_out <= thresh] = 0
return thresh, threshImage_out
if __name__ == "__main__":
img = cv2.imread('dt.jpg')
img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
th, img_new = threshTwoPeaks(img_gray)
th1,img_new_1 = cv2.threshold(img_gray, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_TRIANGLE)
print(th,th1)
plt.subplot(131), plt.imshow(img_gray, cmap='gray')
plt.title('Original Image'), plt.xticks([]), plt.yticks([])
plt.subplot(132), plt.imshow(img_new, cmap='gray')
plt.title('Image'), plt.xticks([]), plt.yticks([])
plt.subplot(133), plt.imshow(img_new_1, cmap='gray')
plt.title('CV2 Image1'), plt.xticks([]), plt.yticks([])
plt.show()
|