詳解Python中圖像邊緣檢測算法的實現(xiàn)

更新時間：2022年05月31日 09:18:29 作者：Mr.Winter`

這篇文章主要為大家詳細介紹了python中圖像邊緣檢測算法的原理及實現(xiàn)，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下

寫在前面

從本節(jié)開始，計算機視覺教程進入第三章節(jié)——圖像特征提取。在本章，你會見到一張簡簡單單的圖片中蘊含著這么多你沒注意到的細節(jié)特征，而這些特征將會在今后更高級的應用中發(fā)揮著極其重要的作用。本文講解基礎(chǔ)特征之一——圖像邊緣。

本文采用面向?qū)ο笤O(shè)計，定義了一個邊緣檢測類EdgeDetect，使圖像邊緣檢測算法的應用更簡潔，例如

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

Detector = EdgeDetect('1.jpg')
Prewitt = Detector.prewitt()
plt.imshow(Prewitt , 'gray')
plt.show()

這個類的構(gòu)造函數(shù)為

class EdgeDetect:
    def __init__(self, img) -> None:
        self.src = cv2.imread(img)
        self.gray = cv2.cvtColor(self.src, cv2.COLOR_BGR2GRAY)

讀取的是圖像的基本信息。

1.一階微分算子

圖像邊緣是數(shù)字圖像的高頻成分，對應圖像梯度的極值。在二維離散數(shù)字圖像上，某個方向上圖像強度函數(shù)微分使用有限差分法來近似，即：

因此圖像邊緣檢測即是對圖像的差分運算。

1.1 Prewitt算子

Prewitt算子本質(zhì)上就是x或y方向上相鄰像素的差分。

那我們常說的圖像梯度是什么意思呢？

其實就是用x與y方向上相鄰像素的差分為方向的向量

在編程實現(xiàn)上，就是構(gòu)造上圖的兩個方向的濾波算子，然后將x xx、y yy兩個方向的邊緣合成就是整張圖各方向的邊緣檢測結(jié)果

def prewitt(self):
    # Prewitt 算子
    kernelX = np.array([[1,1,1],[0,0,0],[-1,-1,-1]], dtype=int)
    kernelY = np.array([[-1,0,1],[-1,0,1],[-1,0,1]], dtype=int)
    # 對圖像濾波
    x = cv2.filter2D(self.gray, cv2.CV_16S, kernelX)
    y = cv2.filter2D(self.gray, cv2.CV_16S, kernelY)
    # 轉(zhuǎn) uint8 ,圖像融合
    absX = cv2.convertScaleAbs(x)
    absY = cv2.convertScaleAbs(y)
    return cv2.addWeighted(absX, 0.5, absY, 0.5, 0)

1.2 Sobel算子

對高斯核函數(shù)x、y方向求導，并將其模板化即得Sobel算子。Sobel算子相比于Prewitt算子有更強的抗噪能力，因為其結(jié)合了高斯濾波的效果。

在編程實現(xiàn)上，就是構(gòu)造上圖的兩個方向的濾波算子，然后將x、y兩個方向的邊緣合成就是整張圖各方向的邊緣檢測結(jié)果

def sobel(self):
    # Sobel 算子
    kernelX = np.array([[1, 2, 1],[0, 0, 0],[-1, -2, -1]],dtype=int)
    kernelY = np.array([[-1, -2, -1],[0, 0, 0],[1, 2, 1]],dtype=int)
    # 對圖像濾波
    x = cv2.filter2D(self.gray, cv2.CV_16S, kernelX)
    y = cv2.filter2D(self.gray, cv2.CV_16S, kernelY)
    # 轉(zhuǎn) uint8 ,圖像融合
    absX = cv2.convertScaleAbs(x)
    absY = cv2.convertScaleAbs(y)
    return cv2.addWeighted(absX, 0.5, absY, 0.5, 0)

2.二階微分算子

2.1 Laplace算子

將Laplace算子

寫成差分方程形式為

將差分方程進一步寫成卷積核形式如圖(a)，可將其擴展為圖(b)使之具有各向同性。微分算子屬于高通濾波，在銳化邊緣的同時也增強了噪點，因此Laplace算子抗噪能力弱，且不能檢測邊緣方向。

在編程實現(xiàn)上，就是構(gòu)造上圖的濾波算子

# Laplace 算子
def laplace(self):
    kernel = np.array([[0, -1, 0], [-1, 4, -1], [0, -1, 0]], dtype=int)
    img = cv2.filter2D(self.gray, cv2.CV_16S, kernel)
    return cv2.convertScaleAbs(img)

2.2 LoG算子

為克服Laplace算子抗噪能力弱這一問題，引入高斯-拉普拉斯算子(LoG, Laplace of Gaussian)，即先低通濾除噪聲，再高通強化邊緣，LoG算子本質(zhì)上是帶通濾波器。

在編程實現(xiàn)上，就是構(gòu)造上圖的濾波算子

# LoG算子
def LoG(self):
    kernel = np.array([[0, 0, 1, 0, 0], [0, 1, 2, 1, 0], [1, 2, -16, 2, 1], [0, 1, 2, 1, 0], [0, 0, 1, 0, 0]], dtype=int)
    img = cv2.filter2D(self.gray, cv2.CV_16S, kernel)
    return cv2.convertScaleAbs(img)