VTK 边缘检测_Canny算子

1.Canny算子

Canny算子是John Canny于20世纪80年代提出的一种多级边缘检测算法。John Canny研究了最优边缘的特性,即检测到的边缘要尽可能跟实际的边缘接近并尽可能的多,同时,要尽量降低噪声对边缘检测的干扰。其计算步骤如下
1)对源图像进行高斯平滑以消除图像中噪声
2)采用差分法近似计算图像每一个像素的梯度,并计算梯度的模值和方向
3)对梯度进行"非极大抑制":图像边缘点梯度值通常在梯度方向是极大值,因此检测边缘需要将非极大值赋值0来抑制非边缘点。检测方法就是在一个局部窗口内,如果中心像素点的梯度不比梯度方向上相邻两个像素值大,那么该中心像素点梯度值赋0。
4)双阈值法检测边缘和连接边缘。取两个梯度阈值high和low,将梯度图像中小于high的像素赋0得到边缘图像I1,该图像能够接近图像边缘但是可能会存在间断点;将梯度图像中小于low的像素赋0得到边缘图像I2,该图中受噪声影响比较大,但是边缘信息更多。在连接边缘时,以I1为基础,对非零点进行边缘跟踪,如果追踪过程中出现中断,则从I2对应像素点及其邻域来寻找可以连接的边缘,直至结束。
以上是Canny算子的计算步骤。
在VTK中没有实现一个专门的类来做Canny边缘检测。
但是我们可以根据以上步骤来实现: 
  1 #include <vtkSmartPointer.h>
  2 #include <vtkImageData.h>
  3 #include <vtkImageShiftScale.h>
  4 #include <vtkRenderWindow.h>
  5 #include <vtkRenderWindowInteractor.h>
  6 #include <vtkInteractorStyleImage.h>
  7 #include <vtkRenderer.h>
  8 #include <vtkImageActor.h>
  9 #include <vtkJPEGReader.h>
 10 #include <vtkImageCast.h>
 11 #include <vtkImageGaussianSmooth.h>
 12 #include <vtkImageGradient.h>
 13 #include <vtkImageMagnitude.h>
 14 #include <vtkImageNonMaximumSuppression.h>
 15 #include <vtkImageConstantPad.h>
 16 #include <vtkImageToStructuredPoints.h>
 17 #include <vtkLinkEdgels.h>
 18 #include <vtkThreshold.h>
 19 #include <vtkGeometryFilter.h>
 20 #include <vtkSubPixelPositionEdgels.h>
 21 #include <vtkCamera.h>
 22 #include <vtkProperty.h>
 23 #include <vtkStripper.h>
 24 #include <vtkPolyDataMapper.h>
 25  
 26 int main(int argc, char* argv[])
 27 {
 28     vtkSmartPointer<vtkJPEGReader> reader =
 29         vtkSmartPointer<vtkJPEGReader>::New();
 30     reader->SetFileName("lena.jpg");
 31     reader->Update();
 32  
 33     vtkSmartPointer<vtkImageCast> ic =
 34         vtkSmartPointer<vtkImageCast>::New();
 35     ic->SetOutputScalarTypeToFloat();
 36     ic->SetInputConnection(reader->GetOutputPort());
 37  
 38     vtkSmartPointer<vtkImageGaussianSmooth> gs =
 39         vtkSmartPointer<vtkImageGaussianSmooth>::New();
 40     gs->SetInputConnection(ic->GetOutputPort());
 41     gs->SetDimensionality(2);
 42     gs->SetRadiusFactors(1, 1, 0);
 43  
 44     vtkSmartPointer<vtkImageGradient> imgGradient =
 45         vtkSmartPointer<vtkImageGradient>::New();
 46     imgGradient->SetInputConnection(gs->GetOutputPort());
 47     imgGradient->SetDimensionality(2);
 48  
 49     vtkSmartPointer<vtkImageMagnitude> imgMagnitude =
 50         vtkSmartPointer<vtkImageMagnitude>::New();
 51     imgMagnitude->SetInputConnection(imgGradient->GetOutputPort());
 52  
 53     vtkSmartPointer<vtkImageNonMaximumSuppression> nonMax =
 54         vtkSmartPointer<vtkImageNonMaximumSuppression>::New();
 55     nonMax->SetMagnitudeInputData(imgMagnitude->GetOutput());
 56     nonMax->SetVectorInputData(imgGradient->GetOutput());
 57     nonMax->SetDimensionality(2);
 58  
 59     vtkSmartPointer<vtkImageConstantPad> pad =
 60         vtkSmartPointer<vtkImageConstantPad>::New();
 61     pad->SetInputConnection(imgGradient->GetOutputPort());
 62     pad->SetOutputNumberOfScalarComponents(3);
 63     pad->SetConstant(0);
 64  
 65     vtkSmartPointer<vtkImageToStructuredPoints> i2sp1 =
 66         vtkSmartPointer<vtkImageToStructuredPoints>::New();
 67     i2sp1->SetInputConnection(nonMax->GetOutputPort());
 68     i2sp1->SetVectorInputData(pad->GetOutput());
 69  
 70     vtkSmartPointer<vtkLinkEdgels> imgLink =
 71         vtkSmartPointer<vtkLinkEdgels>::New();
 72     imgLink->SetInputData(i2sp1->GetOutput());
 73     imgLink->SetGradientThreshold(2);
 74  
 75     vtkSmartPointer<vtkThreshold> thresholdEdgels =
 76         vtkSmartPointer<vtkThreshold>::New();
 77     thresholdEdgels->SetInputConnection(imgLink->GetOutputPort());
 78     thresholdEdgels->ThresholdByUpper(10);
 79     thresholdEdgels->AllScalarsOff();
 80  
 81     vtkSmartPointer<vtkGeometryFilter> gf =
 82         vtkSmartPointer<vtkGeometryFilter>::New();
 83     gf->SetInputConnection(thresholdEdgels->GetOutputPort());
 84  
 85     vtkSmartPointer<vtkImageToStructuredPoints> i2sp =
 86         vtkSmartPointer<vtkImageToStructuredPoints>::New();
 87     i2sp->SetInputConnection(imgMagnitude->GetOutputPort());
 88     i2sp->SetVectorInputData(pad->GetOutput());
 89  
 90     vtkSmartPointer<vtkSubPixelPositionEdgels> spe =
 91         vtkSmartPointer<vtkSubPixelPositionEdgels>::New();
 92     spe->SetInputConnection(gf->GetOutputPort());
 93     spe->SetGradMapsData(i2sp->GetStructuredPointsOutput());
 94  
 95     vtkSmartPointer<vtkStripper> strip =
 96         vtkSmartPointer<vtkStripper>::New();
 97     strip->SetInputConnection(spe->GetOutputPort());
 98  
 99     vtkSmartPointer<vtkPolyDataMapper> dsm =
100         vtkSmartPointer<vtkPolyDataMapper>::New();
101     dsm->SetInputConnection(strip->GetOutputPort());
102     dsm->ScalarVisibilityOff();
103  
104     vtkSmartPointer<vtkActor> planeActor =
105         vtkSmartPointer<vtkActor>::New();
106     planeActor->SetMapper(dsm);
107     planeActor->GetProperty()->SetAmbient(1.0);
108     planeActor->GetProperty()->SetDiffuse(0.0);
109     planeActor->GetProperty()->SetColor(1.0, 0.0, 0.0);
110  
111     vtkSmartPointer<vtkImageActor> originalActor =
112         vtkSmartPointer<vtkImageActor>::New();
113     originalActor->SetInputData(reader->GetOutput());
114  
115     double originalViewport[4] = { 0.0, 0.0, 0.5, 1.0 };
116     double gradviewport[4] = { 0.5, 0.0, 1.0, 1.0 };
117  
118     vtkSmartPointer<vtkRenderer> originalRenderer =
119         vtkSmartPointer<vtkRenderer>::New();
120     originalRenderer->SetViewport(originalViewport);
121     originalRenderer->AddActor(originalActor);
122     originalRenderer->ResetCamera();
123     originalRenderer->SetBackground(1.0, 1.0, 1.0);
124  
125     vtkSmartPointer<vtkRenderer> gradRenderer =
126         vtkSmartPointer<vtkRenderer>::New();
127     gradRenderer->SetViewport(gradviewport);
128     gradRenderer->AddActor(planeActor);
129     gradRenderer->ResetCamera();
130     gradRenderer->SetBackground(1.0, 1.0, 1.0);
131  
132     vtkSmartPointer<vtkRenderWindow> renderWindow =
133         vtkSmartPointer<vtkRenderWindow>::New();
134     renderWindow->SetSize(900, 300);
135     renderWindow->AddRenderer(originalRenderer);
136     renderWindow->AddRenderer(gradRenderer);
137     renderWindow->Render();
138     renderWindow->SetWindowName("CannyExample");
139  
140     vtkSmartPointer<vtkRenderWindowInteractor> renderWindowInteractor =
141         vtkSmartPointer<vtkRenderWindowInteractor>::New();
142     vtkSmartPointer<vtkInteractorStyleImage> style =
143         vtkSmartPointer<vtkInteractorStyleImage>::New();
144  
145     renderWindowInteractor->SetInteractorStyle(style);
146     renderWindowInteractor->SetRenderWindow(renderWindow);
147     renderWindowInteractor->Initialize();
148     renderWindowInteractor->Start();
149  
150     return EXIT_SUCCESS;
151 }
该程序比较复杂, 处理边缘时将其作为几何数据来进行处理。因此涉及了部分几何数据操作的filter,这里如果不明白可以先放一下,再几何数据处理部分会做详细介绍。
程序首先读入图像,计算图像的梯度和模值。
接下来按照Canny算子的步骤进行处理。
我们详细介绍用到的相应的filter:
vtkImageNonMaximumSuppression将图像中的非局部峰值设置为0,输入和输出类型都是vtkImageData:其中输入有两个,模值图像(magnitude)和向量图像,一个典型的应用就是输入梯度模值图像和梯度向量图像对梯度做非极大值抑制。
vtkImageConstantPad增加图像的大小,其输入和输出都为vtkImageData。其中函数SetOutputNumberOfScalarComponents(3)用于设置输出图像的像素数据组分个数,函数SetConstant(0)用于设置输出图像中扩大的区域像素值。而SetOutputWholeExtent()则用于设置输出图像的范围。这里的作用是将梯度图像像素的组分修改为3,方便下面vtkImageToStructuredPoints使用。
vtkImageToStructuredPoints将vtkImageData格式转换为规则点集。该类的输入类型是vtkImageData,另外还有一个可选的RGB三组分向量图像输入;输出类型是vtkStructuredPoints,当输入向量图像时,向量图像像素数据会转为输出图像的对应点的属性。
vtkLinkEdgels类根据点的相邻关系连接成连续的折线Polyline。其内部阈值变量GradientThreshold,可以用来排除输入点中梯度值小于该阈值的点。当使用vtkLinkEdgels进行Canny算子的双阈值边缘检测时,GradientThreshold可以用作较小的阈值。设置该阈值的函数是SetGradientThreshold(2)。
vtkThreshold用于获取输入任意类型数据的满足阈值条件的单元数据。该类的输入为VTK的任意数据类型,输出数据类型是不规则网格。阈值设置有:大于阈值,小于阈值和介于两个阈值之间。内部提供了两种属性模式AttributeMode设置,即阈值比较时是采用的点属性还是单元属性,默认下是点属性。而当属性为多元数据时,还需要设置阈值比较时使用哪个组分的数据。其中提供了三种模式选择,所有组分都满足阈值条件,任意一个满足阈值条件和用户指定的组分满足阈值条件。当使用点属性数据时,如果设置了AllScalars,那么单元满足阈值条件的前提会是其所有点的属性都满足阈值条件。这里将阈值设置为10,即Canny中双阈值的较大阈值。
vtkGeometryFilter将数据转换为几何数据,输出类型为vtkPolyData。该类从vtkThreshold的输出中提取图像边缘的几何数据。
vtkSubPixelPositionEdgels接收一系列连续曲线及其对应的梯度系信息作为输入,利用梯度信息来调整曲线位置。这里对前面提取的图像边缘再根据其梯度进行调整。
vtkStripper用来将输入的多边形、三角形或者线段生成三角形带或者折线段。输入的多边形数据必须是三角形,否则不会进行带化处理。因此处理多边形数据时,可以先用vtkTriangleFilter进行三角化后再使用本类。如果输入中存在孤立点的话,也不会进行任何处理。默认情况下,该filter处理后会丢弃掉属性数据。
原文地址:https://www.cnblogs.com/ybqjymy/p/14241430.html