原理
Canny 边缘检测是一个很流行的边缘检测算法。由John F.Canny在1986年开发。这是一个多步骤的算法。
1.降噪
由于边缘检测易受图片的噪点影响,所以第一步我们用一个5x5的高斯滤波器去除图片上的噪点。这个在之前的章节已经见过了。
2.找到图片中的亮度梯度
然后用索贝尔核在水平和垂直方向过滤第一步被平滑过的图片,这会得到水平方向一阶导数Gx和垂直方向一阶导数Gy。从这两个图像我们可以找到边缘梯度和每个像素的方向。
梯度方向始终和边缘垂直。
3.非最大值抑制
在得到了梯度幅值和方向之后,对图像做一次全扫描去除掉不构成边的不想要的像素。对每一个像素,检查它是否在周围的点里在梯度方向是局部极大值。
点A在边上(在竖直方向),梯度方向垂直于边。点B和点C在梯度方向,所以点A和点B点C一起被检查看是否构成了局部极大值。如果是,就进入下一阶段,如果不是,就会被抑制(设置为0)。
简单来说,你得到的图像是一个“瘦边”的图像。
4.滞后阈值
这一步决定哪些边是真边缘哪些不是。为此,我们需要两个阈值,minVal和maxVal。任何强度梯度比maxVal 大的肯定是边,比minVal小的肯定不是边,所以会被丢弃。那些在这两个阈值范围内的是边或者根据连通性分类为非边。如果他们和“确定为边”的像素联通,他们就会被认为是边的一部分,否则也会被丢弃。见下图:
边A比maxVal 大,所以是“确定为边”,虽然边C比maxVal小,但是它和A相连,所以也被认为是有效的边,这样我们就得到了整个曲线,但对于边B,虽然它比minVal大,而且和C在同一区域,但是它没有和任何“确定为边”的边相连,会被丢弃。所以我们选择相应的minVal和MaxVal来得到正确的结果就很重要。
在这一步里还会去掉一些噪点,因为我们假设所有的边都是长线。
我们最后得到的就是图像里最突出的边界。
OpenCV里的Canny边缘检测
OpenCV把所有这些放在一个函数里,cv2.Canny()。我们来看看怎么用它,第一个参数是输入图片,第二个和第三个参数是我们的minVal 和maxVal。aperture_size参数是索贝尔核的大小,用来找图片的梯度。默认是3,最后一个参数是L2gradient,用来指定寻找梯度幅值的公式。如果为True,会使用上面提到的更准确的公式,否则会用下面这个函数,默认情况下为False:
import cv2
import numpy as np
from matplotlib import pyplot as plt
img = cv2.imread('messi5.jpg',0)
edges = cv2.Canny(img,100,200)
plt.subplot(121), plt.imshow(img,cmap='gray')
plt.title('Original Image'), plt.xticks([]),plt.yticks([])
plt.subplot(122), plt.imshow(edges,cmap='gray')
plt.title('Edge Image'),plt.xticks([]),plt.yticks([])
plt.show()
结果如下:
更多资源:
1. Canny edge detector at Wikipedia
2. Canny Edge Detection Tutorial by Bill Green, 2002.