分割是圖像分析中的一項基本技術(shù)，它允許我們根據(jù)對象、形狀或顏色將圖像劃分為有意義的部分。它在物體檢測、計算機視覺甚至藝術(shù)圖像處理等應(yīng)用中發(fā)揮著關(guān)鍵作用。但如何才能有效實現(xiàn)細(xì)分呢？幸運的是，OpenCV (cv2) 提供了幾種用戶友好且強大的分割方法。

在本教程中，我們將探討三種流行的分割技術(shù)：

• Canny 邊緣檢測 – 非常適合勾畫物體輪廓。
• 分水嶺算法 – 非常適合分離重疊區(qū)域。
• K-Means 顏色分割 – 非常適合對圖像中的相似顏色進行聚類。

為了使本教程引人入勝且實用，我們將使用來自日本大阪的衛(wèi)星和航空圖像，重點關(guān)注古代古墳。您可以從教程的 GitHub 頁面下載這些圖像和相應(yīng)的示例筆記本。

Canny 邊緣檢測到輪廓分割

Canny 邊緣檢測是一種簡單而強大的方法來識別圖像中的邊緣。它的工作原理是檢測強度快速變化的區(qū)域，這些區(qū)域通常是物體的邊界。該技術(shù)通過應(yīng)用強度閾值生成“薄邊緣”輪廓。讓我們深入了解它使用 OpenCV 的實現(xiàn)。

示例：檢測衛(wèi)星圖像中的邊緣
在這里，我們使用大阪的衛(wèi)星圖像，特別是古墳，作為測試用例。

import cv2 
import numpy as np
import matplotlib.pyplot as plt
files = sorted(glob("SAT*.png")) #Get png files 
print(len(files))
img=cv2.imread(files[0])
use_image= img[0:600,700:1300]
gray = cv2.cvtColor(use_image, cv2.COLOR_BGR2GRAY)

#Stadard values 
min_val = 100
max_val = 200
# Apply Canny Edge Detection
edges = cv2.Canny(gray, min_val, max_val)
#edges = cv2.Canny(gray, min_val, max_val,apertureSize=5,L2gradient = True )
False
# Show the result
plt.figure(figsize=(15, 5))
plt.subplot(131), plt.imshow(cv2.cvtColor(use_image, cv2.COLOR_BGR2RGB))
plt.title('Original Image'), plt.axis('off')
plt.subplot(132), plt.imshow(gray, cmap='gray')
plt.title('Grayscale Image'), plt.axis('off')
plt.subplot(133), plt.imshow(edges, cmap='gray')
plt.title('Canny Edges'), plt.axis('off')
plt.show()

輸出邊緣清晰地勾勒出古墳和其他感興趣區(qū)域的部分輪廓。然而，由于閾值過大，一些區(qū)域被遺漏了。結(jié)果在很大程度上取決于 min_val 和 max_val 的選擇以及圖像質(zhì)量。

為了增強邊緣檢測，我們可以對圖像進行預(yù)處理以分散像素強度并減少噪聲。這可以使用直方圖均衡 (cv2.equalizeHist()) 和高斯模糊 (cv2.GaussianBlur()) 來實現(xiàn)。

use_image= img[0:600,700:1300]
gray = cv2.cvtColor(use_image, cv2.COLOR_BGR2GRAY)
gray_og = gray.copy()
gray = cv2.equalizeHist(gray)
gray = cv2.GaussianBlur(gray, (9, 9),1)

plt.figure(figsize=(15, 5))
plt.subplot(121), plt.imshow(gray, cmap='gray')
plt.title('Grayscale Image')
plt.subplot(122)
_= plt.hist(gray.ravel(), 256, [0,256],label="Equalized") 
_ = plt.hist(gray_og.ravel(), 256, [0,256],label="Original",histtype='step')
plt.legend()
plt.title('Grayscale Histogram')

這種預(yù)處理可以均勻強度分布并平滑圖像，這有助于 Canny 邊緣檢測算法捕獲更有意義的邊緣。

邊緣很有用，但它們僅表示邊界。為了分割封閉區(qū)域，我們將邊緣轉(zhuǎn)換為輪廓并可視化它們。

# Edges to contours 
contours, hierarchy = cv2.findContours(edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# Calculate contour areas
areas = [cv2.contourArea(contour) for contour in contours]

# Normalize areas for the colormap
normalized_areas = np.array(areas)
if normalized_areas.max() > 0:
    normalized_areas = normalized_areas / normalized_areas.max()

# Create a colormap
cmap = plt.cm.jet

# Plot the contours with the color map
plt.figure(figsize=(10, 10))
plt.subplot(1,2,1)
plt.imshow(gray, cmap='gray', alpha=0.5)  # Display the grayscale image in the background
mask = np.zeros_like(use_image)
for contour, norm_area in zip(contours, normalized_areas):
    color = cmap(norm_area)  # Map the normalized area to a color
    color = [int(c*255) for c in color[:3]]
    cv2.drawContours(mask, [contour], -1, color,-1 )  # Draw contours on the image

plt.subplot(1,2,2)

上述方法用代表其相對區(qū)域的顏色突出顯示檢測到的輪廓。這種可視化有助于驗證輪廓是否形成封閉體或僅形成線條。然而，在此示例中，許多輪廓仍然是未閉合的多邊形。進一步的預(yù)處理或參數(shù)調(diào)整可以解決這些限制。

通過結(jié)合預(yù)處理和輪廓分析，Canny 邊緣檢測成為識別圖像中對象邊界的強大工具。然而，當(dāng)對象定義明確且噪聲最小時，它的效果最佳。接下來，我們將探索 K 均值聚類以按顏色分割圖像，從而為相同數(shù)據(jù)提供不同的視角。

K均值聚類

K-Means 聚類是數(shù)據(jù)科學(xué)中一種流行的方法，用于將相似的項目分組為聚類，并且它對于基于顏色相似性的圖像分割特別有效。 OpenCV 的 cv2.kmeans 函數(shù)簡化了此過程，使其可以執(zhí)行對象分割、背景去除或視覺分析等任務(wù)。

在本節(jié)中，我們將使用 K 均值聚類將古墳?zāi)箞D像分割為相似顏色的區(qū)域。

首先，我們對圖像的 RGB 值應(yīng)用 K 均值聚類，將每個像素視為一個數(shù)據(jù)點。

import cv2 
import numpy as np
import matplotlib.pyplot as plt
files = sorted(glob("SAT*.png")) #Get png files 
print(len(files))
img=cv2.imread(files[0])
use_image= img[0:600,700:1300]
gray = cv2.cvtColor(use_image, cv2.COLOR_BGR2GRAY)

#Stadard values 
min_val = 100
max_val = 200
# Apply Canny Edge Detection
edges = cv2.Canny(gray, min_val, max_val)
#edges = cv2.Canny(gray, min_val, max_val,apertureSize=5,L2gradient = True )
False
# Show the result
plt.figure(figsize=(15, 5))
plt.subplot(131), plt.imshow(cv2.cvtColor(use_image, cv2.COLOR_BGR2RGB))
plt.title('Original Image'), plt.axis('off')
plt.subplot(132), plt.imshow(gray, cmap='gray')
plt.title('Grayscale Image'), plt.axis('off')
plt.subplot(133), plt.imshow(edges, cmap='gray')
plt.title('Canny Edges'), plt.axis('off')
plt.show()

在分割圖像中，古墳和周圍區(qū)域聚集成不同的顏色。然而，噪聲和顏色的微小變化會導(dǎo)致簇分散，這會給解釋帶來挑戰(zhàn)。

為了減少噪音并創(chuàng)建更平滑的聚類，我們可以在運行 K-Means 之前應(yīng)用中值模糊。

use_image= img[0:600,700:1300]
gray = cv2.cvtColor(use_image, cv2.COLOR_BGR2GRAY)
gray_og = gray.copy()
gray = cv2.equalizeHist(gray)
gray = cv2.GaussianBlur(gray, (9, 9),1)

plt.figure(figsize=(15, 5))
plt.subplot(121), plt.imshow(gray, cmap='gray')
plt.title('Grayscale Image')
plt.subplot(122)
_= plt.hist(gray.ravel(), 256, [0,256],label="Equalized") 
_ = plt.hist(gray_og.ravel(), 256, [0,256],label="Original",histtype='step')
plt.legend()
plt.title('Grayscale Histogram')

模糊的圖像會產(chǎn)生更平滑的簇，減少噪音并使分割區(qū)域在視覺上更具凝聚力。

為了更好地理解分割結(jié)果，我們可以使用 matplotlib plt.fill_ Between;
創(chuàng)建獨特簇顏色的顏色圖

# Edges to contours 
contours, hierarchy = cv2.findContours(edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# Calculate contour areas
areas = [cv2.contourArea(contour) for contour in contours]

# Normalize areas for the colormap
normalized_areas = np.array(areas)
if normalized_areas.max() > 0:
    normalized_areas = normalized_areas / normalized_areas.max()

# Create a colormap
cmap = plt.cm.jet

# Plot the contours with the color map
plt.figure(figsize=(10, 10))
plt.subplot(1,2,1)
plt.imshow(gray, cmap='gray', alpha=0.5)  # Display the grayscale image in the background
mask = np.zeros_like(use_image)
for contour, norm_area in zip(contours, normalized_areas):
    color = cmap(norm_area)  # Map the normalized area to a color
    color = [int(c*255) for c in color[:3]]
    cv2.drawContours(mask, [contour], -1, color,-1 )  # Draw contours on the image

plt.subplot(1,2,2)

這種可視化可以深入了解圖像中的主色及其相應(yīng)的 RGB 值，這對于進一步分析非常有用。因為我們現(xiàn)在可以屏蔽并選擇我的顏色代碼區(qū)域。

簇的數(shù)量 (K) 顯著影響結(jié)果。增加 K 會創(chuàng)建更詳細(xì)的細(xì)分，而較低的值會產(chǎn)生更廣泛的分組。為了進行實驗，我們可以迭代多個 K 值。

import cv2 
import numpy as np
import matplotlib.pyplot as plt
files = sorted(glob("SAT*.png")) #Get png files 
print(len(files))
img=cv2.imread(files[0])
use_image= img[0:600,700:1300]
gray = cv2.cvtColor(use_image, cv2.COLOR_BGR2GRAY)

#Stadard values 
min_val = 100
max_val = 200
# Apply Canny Edge Detection
edges = cv2.Canny(gray, min_val, max_val)
#edges = cv2.Canny(gray, min_val, max_val,apertureSize=5,L2gradient = True )
False
# Show the result
plt.figure(figsize=(15, 5))
plt.subplot(131), plt.imshow(cv2.cvtColor(use_image, cv2.COLOR_BGR2RGB))
plt.title('Original Image'), plt.axis('off')
plt.subplot(132), plt.imshow(gray, cmap='gray')
plt.title('Grayscale Image'), plt.axis('off')
plt.subplot(133), plt.imshow(edges, cmap='gray')
plt.title('Canny Edges'), plt.axis('off')
plt.show()

不同 K 值的聚類結(jié)果揭示了細(xì)節(jié)和簡單性之間的權(quán)衡：

?較低的 K 值（例如 2-3）：聚類范圍廣泛，區(qū)分清晰，適合高級分割。
?更高的 K 值（例如 12-15）：更詳細(xì)的分割，但代價是增加復(fù)雜性和潛在的過度分割。

K-Means 聚類是一種基于顏色相似性分割圖像的強大技術(shù)。通過正確的預(yù)處理步驟，它可以生成清晰且有意義的區(qū)域。然而，其性能取決于 K 的選擇、輸入圖像的質(zhì)量以及所應(yīng)用的預(yù)處理。接下來，我們將探索分水嶺算法，該算法利用地形特征來實現(xiàn)對象和區(qū)域的精確分割。

分水嶺分割

分水嶺算法的靈感來自地形圖，其中分水嶺劃分流域。此方法將灰度強度值視為高程，有效地創(chuàng)建“峰”和“谷”。通過識別感興趣區(qū)域，該算法可以分割具有精確邊界的對象。它對于分離重疊對象特別有用，使其成為細(xì)胞分割、對象檢測和區(qū)分密集特征等復(fù)雜場景的絕佳選擇。

第一步是預(yù)處理圖像以增強特征，然后應(yīng)用分水嶺算法。

use_image= img[0:600,700:1300]
gray = cv2.cvtColor(use_image, cv2.COLOR_BGR2GRAY)
gray_og = gray.copy()
gray = cv2.equalizeHist(gray)
gray = cv2.GaussianBlur(gray, (9, 9),1)

plt.figure(figsize=(15, 5))
plt.subplot(121), plt.imshow(gray, cmap='gray')
plt.title('Grayscale Image')
plt.subplot(122)
_= plt.hist(gray.ravel(), 256, [0,256],label="Equalized") 
_ = plt.hist(gray_og.ravel(), 256, [0,256],label="Original",histtype='step')
plt.legend()
plt.title('Grayscale Histogram')

分段區(qū)域和邊界可以與中間處理步驟一起可視化。

# Edges to contours 
contours, hierarchy = cv2.findContours(edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# Calculate contour areas
areas = [cv2.contourArea(contour) for contour in contours]

# Normalize areas for the colormap
normalized_areas = np.array(areas)
if normalized_areas.max() > 0:
    normalized_areas = normalized_areas / normalized_areas.max()

# Create a colormap
cmap = plt.cm.jet

# Plot the contours with the color map
plt.figure(figsize=(10, 10))
plt.subplot(1,2,1)
plt.imshow(gray, cmap='gray', alpha=0.5)  # Display the grayscale image in the background
mask = np.zeros_like(use_image)
for contour, norm_area in zip(contours, normalized_areas):
    color = cmap(norm_area)  # Map the normalized area to a color
    color = [int(c*255) for c in color[:3]]
    cv2.drawContours(mask, [contour], -1, color,-1 )  # Draw contours on the image

plt.subplot(1,2,2)

該算法成功識別不同的區(qū)域并在對象周圍繪制清晰的邊界。在本例中，古墳被精確分割。然而，該算法的性能在很大程度上取決于閾值處理、噪聲去除和形態(tài)學(xué)操作等預(yù)處理步驟。

添加高級預(yù)處理，例如直方圖均衡或自適應(yīng)模糊，可以進一步增強結(jié)果。例如：

# Kmean color segmentation
use_image= img[0:600,700:1300]
#use_image = cv2.medianBlur(use_image, 15)


 # Reshape image for k-means
pixel_values = use_image.reshape((-1, 3)) if len(use_image.shape) == 3 else use_image.reshape((-1, 1))
pixel_values = np.float32(pixel_values)

criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10, 1.0)
K = 3
attempts=10
ret,label,center=cv2.kmeans(pixel_values,K,None,criteria,attempts,cv2.KMEANS_PP_CENTERS)

centers = np.uint8(center)
segmented_data = centers[label.flatten()]
segmented_image = segmented_data.reshape(use_image.shape)

plt.figure(figsize=(10, 6))
plt.subplot(1,2,1),plt.imshow(use_image[:,:,::-1])
plt.title("RGB View")
plt.subplot(1,2,2),plt.imshow(segmented_image[:,:,[2,1,0]])
plt.title(f"Kmean Segmented Image K={K}")

通過這些調(diào)整，可以準(zhǔn)確分割更多區(qū)域，并最大限度地減少噪聲偽影。

分水嶺算法在需要精確邊界劃分和重疊對象分離的場景中表現(xiàn)出色。通過利用預(yù)處理技術(shù)，它甚至可以有效地處理像古墳古墳區(qū)域這樣的復(fù)雜圖像。然而，它的成功取決于仔細(xì)的參數(shù)調(diào)整和預(yù)處理。

結(jié)論

分割是圖像分析中的重要工具，提供了隔離和理解圖像中不同元素的途徑。本教程演示了三種強大的分割技術(shù)：Canny 邊緣檢測、K 均值聚類和分水嶺算法，每種技術(shù)都是針對特定應(yīng)用程序量身定制的。從勾勒出大阪古代古墳的輪廓，到聚類城市景觀和劃分不同區(qū)域，這些方法凸顯了 OpenCV 在應(yīng)對現(xiàn)實世界挑戰(zhàn)方面的多功能性。

現(xiàn)在去將其中的一些方法應(yīng)用到您選擇的應(yīng)用程序中，并發(fā)表評論并分享結(jié)果。另外，如果您知道任何其他簡單的分割方法，也請分享

以上是[Python-CV圖像分割：Canny邊緣、分水嶺和K-Means方法的詳細(xì)內(nèi)容。更多信息請關(guān)注PHP中文網(wǎng)其他相關(guān)文章！