隨著當(dāng)今世界的發(fā)展，計(jì)算機(jī)視覺(jué)技術(shù)的應(yīng)用越來(lái)越廣泛。伴隨著硬件設(shè)備的不斷升級(jí)，構(gòu)造復(fù)雜的計(jì)算機(jī)視覺(jué)應(yīng)用變得越來(lái)越容易了。OpenCV像是一個(gè)黑盒，讓我們專(zhuān)注于視覺(jué)應(yīng)用的開(kāi)發(fā)，而不必過(guò)多的關(guān)注基礎(chǔ)圖象處理的具體細(xì)節(jié)。

圖像分割

了解分水嶺算法之前，我們需要了解什么是圖像的分割。

在圖像的處理過(guò)程中，經(jīng)常需要從圖像中將前景對(duì)象作為目標(biāo)圖像分割或者提取出來(lái)。例如，在視頻監(jiān)控中，觀(guān)測(cè)到的是固定背景下的視頻內(nèi)容，而我們對(duì)背景本身并無(wú)興趣，感興趣的是背景中出現(xiàn)的車(chē)輛，行人或者其他對(duì)象。我們希望將這些對(duì)象從視頻中提取出來(lái)，而忽略那些沒(méi)有對(duì)象進(jìn)入背景的視頻內(nèi)容。

分水嶺算法

圖像分割是圖像處理過(guò)程中一種非常重要的操作。分水嶺算法將圖像形象地比喻為地理學(xué)上的地形表面，實(shí)現(xiàn)圖像分割，該算法非常有用。

下面，博主對(duì)分水嶺算法的相關(guān)內(nèi)容做簡(jiǎn)單的介紹。（詳細(xì)可以參考岡薩雷斯的《數(shù)字圖像處理》一書(shū)）

任何一副灰度圖像，都可以被看作是地理學(xué)上的地形表面，灰度值越高的區(qū)域可以被看成是山峰，灰度值越低的區(qū)域可以被看成是山谷。

如果我們向每個(gè)山谷中灌注不同顏色的水。那么隨著水位的不斷升高，不同山谷的水就匯聚到一起。在這個(gè)過(guò)程中，為了防止不同山谷的水交匯，我們需要在水流可能匯合的地方構(gòu)建堤壩。該過(guò)程將圖像分為兩個(gè)不同的集合：集水盆地和分水嶺線(xiàn)。我們構(gòu)建的堤壩就是分水嶺線(xiàn)，也即對(duì)原始圖像的分割。這就是分水嶺算法的原理。

不過(guò)，一般的圖像都存在著噪聲，采用分水嶺算法時(shí)，會(huì)經(jīng)常得到過(guò)度分割的結(jié)果。為了改善圖像分割的效果，人們提出了基于掩摸的改進(jìn)的分水嶺算法。改進(jìn)的分水嶺算法允許用戶(hù)將它認(rèn)為是同一個(gè)分割區(qū)域的部分標(biāo)注出來(lái)。這樣，分水嶺算法在處理時(shí)，就會(huì)將標(biāo)注的部分處理為同一個(gè)分割區(qū)域。

如果對(duì)于該理論不怎么了解，可以使用軟件PowerPoint中的“刪除背景”功能進(jìn)行觀(guān)察配合理解。

waterShed函數(shù)

在OpenCV中，可以使用函數(shù)cv2.watershed()函數(shù)實(shí)現(xiàn)分水嶺算法。不過(guò)，具體實(shí)現(xiàn)的過(guò)程，還需要借助形態(tài)學(xué)函數(shù)，距離變換函數(shù)cv2.distanceTransform()，cv2.connectedComponents()來(lái)完成圖像分割。

形態(tài)學(xué)分割

在使用分水嶺算法之前，我們需要對(duì)圖像進(jìn)行簡(jiǎn)單的形態(tài)學(xué)處理。一般情況下，我們都是使用形態(tài)學(xué)中的開(kāi)運(yùn)算，因?yàn)殚_(kāi)運(yùn)算是先腐蝕后膨脹的操作，能夠去除圖像內(nèi)的噪聲。

import cv2
import numpy as np
import matplotlib.pyplot as plt

img = cv2.imread("36.jpg")
k=np.ones((5,5),dtype=np.uint8)
e=cv2.erode(img,k)
result=cv2.subtract(img,e)

plt.subplot(131)
plt.imshow(img, cmap="gray")
plt.axis('off')

plt.subplot(132)
plt.imshow(e, cmap="gray")
plt.axis('off')

plt.subplot(133)
plt.imshow(result, cmap="gray")
plt.axis('off')
plt.show()

回顧一下，我們前面的開(kāi)運(yùn)算函數(shù)為cv2.erode()，這里我們首先經(jīng)過(guò)開(kāi)運(yùn)算去除噪聲。然后減法運(yùn)算cv2.subtract()獲取圖像邊界。運(yùn)行之后，效果如下：

distanceTransform函數(shù)

當(dāng)圖像內(nèi)的各個(gè)子圖沒(méi)有連接時(shí)，可以直接使用形態(tài)學(xué)的腐蝕操作確定前景對(duì)象，但是如果圖像內(nèi)的子圖連接在一起時(shí)，就很難確定前景對(duì)象了。這個(gè)時(shí)候，就需要借助變換函數(shù)cv2.distanceTransform()方便地將前景對(duì)象提取出來(lái)。

cv2.distanceTransform()反應(yīng)了各個(gè)像素點(diǎn)與背景（值為0的像素點(diǎn)）的距離關(guān)系。通常情況下：

• 如果前景對(duì)象的中心距離值為0的像素點(diǎn)距離較遠(yuǎn)，會(huì)得到一個(gè)較大的值。
• 如果前景對(duì)象的邊緣距離值為0的像素點(diǎn)較近，會(huì)得到一個(gè)較小的值。

下面，我們來(lái)使用該函數(shù)確定一副圖像的前景，并觀(guān)察效果。

import cv2
import numpy as np
import matplotlib.pyplot as plt

img = cv2.imread("36.jpg")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
k = np.ones((5, 5), dtype=np.uint8)
opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, k, iterations=2)
distTransform = cv2.distanceTransform(opening, cv2.DIST_L2, 5)
ret, fore = cv2.threshold(distTransform, 0.7 * distTransform.max(), 255, 0)

plt.subplot(131)
plt.imshow(img, cmap="gray")
plt.axis('off')

plt.subplot(132)
plt.imshow(distTransform, cmap="gray")
plt.axis('off')

plt.subplot(133)
plt.imshow(fore, cmap="gray")
plt.axis('off')
plt.show()

這里，我們使用cv2.morphologyEx函數(shù)進(jìn)行開(kāi)運(yùn)算，同時(shí)使用cv2.distanceTransform得到距離圖像，最后在通過(guò)cv2.threshold對(duì)距離圖像進(jìn)行閾值處理，確定前景。運(yùn)行之后，效果如下：

確定未知區(qū)域

通過(guò)距離函數(shù)，我們獲取到了圖像的“中心”，也就是“確定前景”。為了方便后續(xù)的講解，我們將確定前景稱(chēng)為F。

圖像中有了確定前景F和確定背景B，剩下的區(qū)域就是未知區(qū)域UN了。這部分區(qū)域正是分水嶺算法要進(jìn)一步明確的區(qū)域。

針對(duì)一副圖像0，通過(guò)以下關(guān)系能夠得到未知區(qū)域UN：

未知區(qū)域UN=圖像0-確定背景B-確定前景F

由上述公式變換得到：

未知區(qū)域UN=（圖像0-確定背景B）-確定前景F

其中（圖像0-確定背景B）就是我們開(kāi)始的減法操作，通過(guò)形態(tài)學(xué)膨脹得到。也只需要將上面的代碼添加4行并更改顯示的代碼內(nèi)容：

bg=cv2.dilate(opening,k,iterations=3)
fore=np.uint8(fore)
un=cv2.subtract(bg,fore)

plt.subplot(221)
plt.imshow(img, cmap="gray")
plt.axis('off')

plt.subplot(222)
plt.imshow(bg, cmap="gray")
plt.axis('off')

plt.subplot(223)
plt.imshow(fore, cmap="gray")
plt.axis('off')

plt.subplot(224)
plt.imshow(un, cmap="gray")
plt.axis('off')
plt.show()

運(yùn)行之后，效果如下：

左上為原圖

右上為原圖膨脹后得到的圖像bg，其背景圖像是確定背景B。前景圖像是“原始圖像0-確定背景B”

左下為確定前景圖像fore

右下為未知區(qū)域圖像UN

ConnectedComponents函數(shù)

明確了確定前景后，就可以對(duì)確定前景進(jìn)行標(biāo)注了。在OpenCV中，它提供了cv2.ConnectedComponents()函數(shù)進(jìn)行標(biāo)注。

該函數(shù)會(huì)將背景標(biāo)注為0，將其他的對(duì)象使用從1開(kāi)始的正整數(shù)標(biāo)注。它只有一個(gè)參數(shù)8位單通道的待標(biāo)注圖像。

返回值有兩個(gè)：retval為返回的標(biāo)注數(shù)量，labels為標(biāo)注的結(jié)果圖像。

下面，我們來(lái)使用該函數(shù)進(jìn)行標(biāo)注。代碼如下（同樣更改上面bg下面代碼就行）：

bg = cv2.dilate(opening, k, iterations=3)
fore = np.uint8(fore)
ret, markets = cv2.connectedComponents(fore)
unknown=cv2.subtract(bg,fore)
markets=markets+1
markets[unknown==255]=0

plt.subplot(131)
plt.imshow(img, cmap="gray")
plt.axis('off')

plt.subplot(132)
plt.imshow(fore, cmap="gray")
plt.axis('off')

plt.subplot(133)
plt.imshow(markets, cmap="gray")
plt.axis('off')
plt.show()

修改上面fore = np.uint8(fore)的代碼，并修改輸出內(nèi)容。運(yùn)行之后，我們會(huì)得到原圖，前景圖像的中心點(diǎn)圖像fore以及標(biāo)注后的結(jié)果圖像markets。效果如下：

實(shí)戰(zhàn)分水嶺算法

經(jīng)過(guò)前文的介紹，我們了解了使用分水嶺算法進(jìn)行圖像分割的基本步驟：

• 通過(guò)形態(tài)學(xué)開(kāi)運(yùn)算對(duì)原始圖像0進(jìn)行去噪
• 通過(guò)腐蝕操作獲取“確定背景B”。需要注意，這里得到“原始圖像-確定背景”即可
• 利用距離變換函數(shù)對(duì)原始圖像進(jìn)行運(yùn)算，并對(duì)其進(jìn)行閾值處理，得到“確定前景F”
• 計(jì)算未知區(qū)域UN（UN=0-B-F）
• 利用函數(shù)cv2.connectedComponents()對(duì)原始圖像0進(jìn)行標(biāo)注
• 對(duì)函數(shù)cv2.connectedComponents()的標(biāo)注結(jié)果進(jìn)行修正
• 使用分水嶺函數(shù)完成圖像分割
  

完整代碼如下：

import cv2
import numpy as np
import matplotlib.pyplot as plt

img = cv2.imread("36.jpg")
plt.subplot(121)
plt.imshow(img, cmap="gray")
plt.axis('off')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)
k = np.ones((5, 5), dtype=np.uint8)
opening = cv2.morphologyEx(thresh, cv2.MORPH_OPEN, k, iterations=2)
distTransform = cv2.distanceTransform(opening, cv2.DIST_L2, 5)
ret, fore = cv2.threshold(distTransform, 0.2 * distTransform.max(), 255, 0)
bg = cv2.dilate(opening, k, iterations=3)
fore = np.uint8(fore)
ret, markets = cv2.connectedComponents(fore)
unknown = cv2.subtract(bg, fore)
markets = markets + 1
markets[unknown == 255] = 0
markets = cv2.watershed(img, markets)
img[markets == -1] = [255, 0, 0]

plt.subplot(122)
plt.imshow(img, cmap="gray")
plt.axis('off')
plt.show()

運(yùn)行之后，我們就可以得到分割的圖像：

當(dāng)然，參數(shù)可以調(diào)整，可以看到大致的硬幣被完整的分割出來(lái)了。

到此這篇關(guān)于OpenCV-Python使用分水嶺算法實(shí)現(xiàn)圖像的分割與提取的文章就介紹到這了,更多相關(guān)OpenCV圖像分割與提取內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！