VTK学习之边缘检测（梯度算子）

最新推荐文章于 2024-08-12 23:09:11 发布

Littlehero_121

最新推荐文章于 2024-08-12 23:09:11 发布

阅读量218

点赞数

文章标签：学习数学建模算法

本文链接：https://blog.csdn.net/Littlehero_121/article/details/131284787

版权

该代码示例展示了如何利用VTK库对JPEG图像进行边缘检测。首先，它将图像转换为灰度图，然后计算图像的梯度并确定其大小。接着，通过`vtkImageGradient`和`vtkImageMagnitude`过滤器来检测边缘，最后通过`vtkImageShiftScale`调整灰度范围并将结果显示在渲染窗口中。

摘要由CSDN通过智能技术生成

参考博客：VTK修炼之道32：边缘检测_梯度算子_基于梯度的边缘检测算子_沈子恒的博客-CSDN博客

直接上源码：

#include <vtkAutoInit.h>
#include <vtkSmartPointer.h>
#include <vtkJPEGReader.h>
#include <vtkImageGradient.h>
#include <vtkImageMagnitude.h>
#include <vtkImageData.h>
#include <vtkImageShiftScale.h>
#include <vtkImageActor.h>
#include <vtkRenderer.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkInteractorStyleImage.h>

VTK_MODULE_INIT(vtkRenderingOpenGL2);
VTK_MODULE_INIT(vtkInteractionStyle);

void main()
{
	//输入原始图像
	vtkSmartPointer<vtkJPEGReader> reader = vtkSmartPointer<vtkJPEGReader>::New();
	reader->SetFileName("234.jpg");
	reader->Update();

	//先转换为灰度图
	vtkSmartPointer<vtkImageLuminance> luminanceFilter = vtkSmartPointer<vtkImageLuminance>::New();
	luminanceFilter->SetInputConnection(reader->GetOutputPort());
	luminanceFilter->Update();

	//开始加入边缘检测，计算图像梯度，图像梯度是一个向量，具有方向和大小
	vtkSmartPointer<vtkImageGradient> imgGradient = vtkSmartPointer<vtkImageGradient>::New();
	imgGradient->SetInputConnection(luminanceFilter->GetOutputPort());
	imgGradient->SetDimensionality(2);//设置计算图像的维数，默认是二维

	vtkSmartPointer<vtkImageMagnitude> imgMagnitude = vtkSmartPointer<vtkImageMagnitude>::New();
	imgMagnitude->SetInputConnection(imgGradient->GetOutputPort());
	imgMagnitude->Update();

	//设置灰度范围
	double Range[2];
	vtkSmartPointer<vtkImageData> getRange = vtkSmartPointer<vtkImageData>::New();
	imgMagnitude->GetOutput()->GetScalarRange(Range);//图像灰度范围最小值、最大值


	vtkSmartPointer<vtkImageShiftScale> imgShift = vtkSmartPointer<vtkImageShiftScale>::New();
	imgShift->SetOutputScalarTypeToUnsignedChar(); //强制类型转换 0~255
	imgShift->SetScale(255 / Range[1]); //灰度映射间距
	imgShift->SetInputConnection(imgMagnitude->GetOutputPort());
	imgShift->Update();

	vtkSmartPointer<vtkImageActor> origActor = vtkSmartPointer<vtkImageActor>::New();
	origActor->SetInputData(reader->GetOutput());

	vtkSmartPointer<vtkImageActor> GradientActor = vtkSmartPointer<vtkImageActor>::New();
	GradientActor->SetInputData(imgShift->GetOutput());

	double origView[4] = { 0, 0, 0.5, 1 };
	double gradientView[4] = { 0.5, 0, 1, 1 };
	vtkSmartPointer<vtkRenderer> origRender = vtkSmartPointer<vtkRenderer>::New();
	origRender->SetViewport(origView);
	origRender->AddActor(origActor);
	origRender->ResetCamera();
	origRender->SetBackground(1.0, 0, 0);

	vtkSmartPointer<vtkRenderer> gradientRender = vtkSmartPointer<vtkRenderer>::New();
	gradientRender->SetViewport(gradientView);
	gradientRender->AddActor(GradientActor);
	gradientRender->ResetCamera();
	gradientRender->SetBackground(1, 1, 1);

	vtkSmartPointer<vtkRenderWindow> renderWindow = vtkSmartPointer<vtkRenderWindow>::New();
	renderWindow->AddRenderer(origRender);
	renderWindow->AddRenderer(gradientRender);
	renderWindow->SetSize(640, 320);
	renderWindow->Render();
	renderWindow->SetWindowName("Gradient Show");

	vtkSmartPointer<vtkRenderWindowInteractor> interactor = vtkSmartPointer<vtkRenderWindowInteractor>::New();
	vtkSmartPointer<vtkInteractorStyleImage> style = vtkSmartPointer<vtkInteractorStyleImage>::New();
	interactor->SetRenderWindow(renderWindow);
	interactor->SetInteractorStyle(style);
	interactor->Initialize();
	interactor->Start();
}

最终效果：