OpenCV Otsu算法是一種用於圖像分割的自適應算法,主要用於二值化處理。該算法能夠對灰度圖像進行有效的分割,幫助我們將圖像轉換成二值圖像,以便於後續的處理。
一、算法原理
Otsu算法的基本思想是遍歷圖像中所有像素的灰度值,計算出一個最佳的閾值,將像素分為兩類:一類是高於閾值的像素,另一類是低於閾值的像素。
在計算最佳閾值的過程中,Otsu算法最大化了類間方差,最小化了類內方差,從而得到了最佳的閾值。
我們可以用以下公式表示算法:
t, 【max_variance】 = cv2.threshold(gray_img, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU)
其中,gray_img是輸入的灰度圖像數據,THRESH_BINARY+THRESH_OTSU是閾值處理方法。
二、算法過程
Otsu算法的過程分為以下幾步:
- 讀入灰度圖像
- 計算灰度直方圖
- 對每一個灰度級計算出它的權重和平均值
- 對每一個灰度級計算出類內和類間方差
- 最大化類間方差,得出最佳閾值
- 用最佳閾值將圖像進行二值化
這個過程可以用以下代碼來實現:
import cv2 # Read image in grayscale img = cv2.imread("img.jpg", cv2.IMREAD_GRAYSCALE) # Calculate histogram hist = cv2.calcHist([img],[0],None,[256],[0,256]) # Total number of pixels pixels = img.shape[0] * img.shape[1] # Initialize variables weight_background = 0 weight_foreground = 0 mean_background = 0 mean_foreground = 0 variance = 0 max_variance = 0 threshold = 0 # Loop over all possible threshold values for i in range(0, 256): # Compute weights weight_background += hist[i][0] weight_foreground = pixels - weight_background # Compute means mean_background += i * hist[i][0] mean_foreground = (img.sum() - mean_background) / weight_foreground # Compute variances variance = weight_background * weight_foreground * ((mean_background / weight_background) - (mean_foreground / weight_foreground)) ** 2 # Update threshold if variance > max_variance: max_variance = variance threshold = i # Threshold image img_thresholded = cv2.threshold(img, threshold, 255, cv2.THRESH_BINARY)[1]
三、應用場景
OpenCV Otsu算法可以應用於各種場景中,例如:
- 自動化工業中的零件識別
- 醫學圖像處理中的癌細胞檢測
- 計算機視覺中的圖像分割、物體檢測等
以下是一個使用OpenCV Otsu算法進行圖像二值化的示例代碼:
import cv2 # Read image img = cv2.imread("img.jpg", cv2.IMREAD_GRAYSCALE) # Threshold image threshold_value, threshold_img = cv2.threshold(img, 0, 255, cv2.THRESH_BINARY+cv2.THRESH_OTSU) # Show images cv2.imshow("Original Image", img) cv2.imshow("Thresholded Image", threshold_img) # Wait for key press and close windows cv2.waitKey() cv2.destroyAllWindows()
四、算法優缺點
OpenCV Otsu算法的主要優點是自適應性強,可以有效地對不同亮度的圖像進行處理,並且原理簡單、易於實現。
缺點是可能會出現過度分割的情況,因為該算法只考慮了像素灰度值的分布情況,沒有考慮像素之間的空間關係。此外,該算法對圖像中的噪聲也非常敏感,因此前期需要對圖像進行預處理。
原創文章,作者:小藍,如若轉載,請註明出處:https://www.506064.com/zh-hant/n/247410.html