前言
区域生长算法已被证明是一种有效的图像分割方法。区域生长算法的基本思想是从一个或多个种子区域(该种子区域被认为在要分割的对象内部)开始，评估与该区域相邻的像素，以确定它们是否也应被视为对象的一部分。如果邻域像素满足要求，则它们被添加到该区域，并且只要有新像素被添加到该区域，区域生长过程就继续。区域增长算法根据用于决定像素是否应包含在区域中的标准、用于确定邻居的连通性类型以及用于访问相邻像素的策略而变化。
开发环境
windows10、VS2017 、VTK8.2、ITK5.0.1
运行结果
使用方法
按住ctrl键，用鼠标左键拾取种子点。区域生长默认的范围是[226,3071]，若鼠标点选的像素值在该范围之外，则不做任何处理；若像素值在该范围内，则进行区域生长，并将分割后的图像进行彩色映射和原始图像叠加。
代码
//有些头文件用不到
#pragma once
#include "pch.h"
#include <iostream>
#include <vtkAutoInit.h>
VTK_MODULE_INIT(vtkRenderingOpenGL2);
VTK_MODULE_INIT(vtkRenderingVolumeOpenGL2);
VTK_MODULE_INIT(vtkInteractionStyle);
VTK_MODULE_INIT(vtkRenderingFreeType);
#include <vtkImageData.h>
#include <vtkProperty.h>
#include <vtkDataSetMapper.h>
#include <vtkRendererCollection.h>
#include <itkImageToVTKImageFilter.h>
#include <vtkPolyDataMapper.h>
#include <vtkDICOMImageReader.h>
#include <vtkPolyData.h>
#include <vtkPointData.h>
#include <vtkAxesActor.h>
#include "vtkRenderWindowInteractor.h"
#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkMarchingCubes.h"
#include "vtkStripper.h"
#include "vtkActor.h"
#include "vtkSmoothPolyDataFilter.h"
#include "vtkPolyDataNormals.h"
#include "vtkImageShrink3D.h"
#include "vtkDecimatePro.h"
#include "vtkProperty.h"
#include <vtkInteractorStyleImage.h>
#include <vector>
#include <string>
#include <vtkSmartPointer.h>
#include <vtkImageActor.h>
#include <vtkRenderer.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkImageViewer2.h>
#include <vtkImageSliceMapper.h>
#include <vtkPNGReader.h>
#include <vtkSTLReader.h>
#include <vtkTexture.h>
#include <vtkTextureMapToCylinder.h>
#include <vtkImageFlip.h>
#include <vtkImageGradient.h>
#include <vtkImageGradient.h>
#include <vtkImageMagnitude.h>
#include <vtkImageShiftScale.h>
#include <vtkImageHybridMedian2D.h>
#include <vtkSphereSource.h>
#include <vtkTextProperty.h>
#include <vtkProperty2D.h>
#include <vtkSmartPointer.h>
#include <vtkPolyData.h>
#include <vtkSliderWidget.h>
#include <vtkPolyDataMapper.h>
#include <vtkActor.h>
#include <vtkRenderWindow.h>
#include <vtkRenderer.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkPolyData.h>
#include <vtkSmartPointer.h>
#include <vtkSphereSource.h>
#include <vtkCommand.h>
#include <vtkWidgetEvent.h>
#include <vtkCallbackCommand.h>
#include <vtkWidgetEventTranslator.h>
#include <vtkInteractorStyleTrackballCamera.h>
#include <vtkSliderWidget.h>
#include <vtkSliderRepresentation2D.h>
#include <vtkProperty.h>
#include "itkImageToVTKImageFilter.h"
#include "itkJPEGImageIOFactory.h"
#include "itkRGBPixel.h"
#include "itkImage.h"
#include "itkGDCMImageIO.h"
#include "itkGDCMSeriesFileNames.h"
#include "itkVTKImageToImageFilter.h"
#include "itkRescaleIntensityImageFilter.h"
#include "itkImageSeriesReader.h"
#include "itkImageSeriesWriter.h"
#include "itkCurvatureFlowImageFilter.h"
#include "itkConnectedThresholdImageFilter.h"
#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkCastImageFilter.h"
#include "itkPNGImageIOFactory.h"
#include "itksys/SystemTools.hxx"
#include "itkImageRegionIterator.h"
#include <itkImage.h>
#include <itkImageFileReader.h>
#include <itkImageFileWriter.h>
#include <itkPNGImageIOFactory.h>
#include <itkConnectedThresholdImageFilter.h>
#include <itkImageSeriesReader.h>
#include <itkGDCMImageIO.h>
#include <itkImageToVTKImageFilter.h>
#include <vtkLookupTable.h>
#include <vtkImageProperty.h>
#include <vtkImageMapToColors.h>
#include "vtkImageViewer2.h"
using namespace std;
typedef short PixelType;
const unsigned int Dimension = 2;
typedef itk::Image< PixelType, Dimension > Input2dImageType;
typedef itk::GDCMImageIO ImageIOType;//GDCMImageIO读DICOM
ImageIOType::Pointer gdcmImageIO = ImageIOType::New();
typedef itk::ImageFileReader< Input2dImageType > ReaderType2d;
ReaderType2d::Pointer reader2d = ReaderType2d::New();
vtkSmartPointer<vtkImageViewer2> viewer = vtkSmartPointer<vtkImageViewer2>::New();
vtkSmartPointer<vtkImageData> itkToVtk(Input2dImageType::Pointer image)
{
typedef itk::ImageToVTKImageFilter< Input2dImageType> itkTovtkFilterType;
itkTovtkFilterType::Pointer itkTovtkImageFilter = itkTovtkFilterType::New();
itkTovtkImageFilter->SetInput(image);//设置图像数据从ITK转向VTK
itkTovtkImageFilter->Update();
return itkTovtkImageFilter->GetOutput();
}
Input2dImageType::Pointer connectedThreshold(int range[2], Input2dImageType::Pointer itkImage, Input2dImageType::IndexType seed)
{
typedef itk::ConnectedThresholdImageFilter< Input2dImageType, Input2dImageType > ConnectedFilterType;
ConnectedFilterType::Pointer connectedThres = ConnectedFilterType::New();
connectedThres->SetInput(itkImage);
connectedThres->SetLower(range[0]);
connectedThres->SetUpper(range[1]);
connectedThres->SetReplaceValue(3071);
connectedThres->AddSeed(seed);
connectedThres->Update();
return connectedThres->GetOutput();
}
class LineInteractorStyle : public vtkInteractorStyleTrackballCamera
{
public:
static LineInteractorStyle* New();
vtkTypeMacro(LineInteractorStyle, vtkInteractorStyleTrackballCamera);
~LineInteractorStyle();
virtual void OnLeftButtonDown();
virtual void OnRightButtonDown();
Input2dImageType::Pointer itkImage = nullptr;
private:
};
vtkStandardNewMacro(LineInteractorStyle);
LineInteractorStyle::~LineInteractorStyle()
{
}
void LineInteractorStyle::OnLeftButtonDown()
{
if (this->Interactor->GetControlKey()) {
//视口坐标
int *clickPos;
clickPos = this->GetInteractor()->GetEventPosition();
double point1[3]{
0 };
point1[0] = clickPos[0];
point1[1] = clickPos[1];
//世界坐标
double worldPoint[3]{
0 };
this->Interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer()->SetDisplayPoint(point1[0], point1[1], 0);
this->Interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer()->DisplayToWorld();
worldPoint[0] = (this->Interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer()->GetWorldPoint())[0];
worldPoint[1] = (this->Interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer()->GetWorldPoint())[1];
worldPoint[2] = (this->Interactor->GetRenderWindow()->GetRenderers()->GetFirstRenderer()->GetWorldPoint())[2];
typedef itk::Point< double, 2 > PointType;
PointType point;
point[0] = worldPoint[0];
point[1] = worldPoint[1];
Input2dImageType::IndexType pixelIndex;
bool isInside = this->itkImage->TransformPhysicalPointToIndex(point, pixelIndex);
if (!isInside)
return;
int range[2] = {
226,3071 };
Input2dImageType::PixelType value = this->itkImage->GetPixel(pixelIndex);
if (value<range[0] || value>range[1])
return;
Input2dImageType::Pointer image1 = connectedThreshold(range, this->itkImage, pixelIndex);
vtkSmartPointer<vtkImageData> image2 = itkToVtk(image1);
vtkSmartPointer<vtkLookupTable> pColorTable = vtkSmartPointer<vtkLookupTable>::New();
pColorTable->SetNumberOfColors(2);
pColorTable->SetTableRange(0, 3071);
pColorTable->SetTableValue(0, 0.0, 0.0, 0.0, 0.0);
static bool flag = true;
if (flag)
pColorTable->SetTableValue(1, 0, 1, 0, 1.0);
else
pColorTable->SetTableValue(1, 0, 0, 1, 1.0);
flag = !flag;
pColorTable->Build();
vtkSmartPointer<vtkImageMapToColors> colorMap =
vtkSmartPointer<vtkImageMapToColors>::New();
colorMap->SetInputData(image2);
colorMap->SetLookupTable(pColorTable);
colorMap->Update();
vtkSmartPointer<vtkImageActor> colorActor =
vtkSmartPointer<vtkImageActor>::New();
colorActor->SetInputData(colorMap->GetOutput());
colorActor->SetInterpolate(false);
viewer->GetRenderer()->AddActor(colorActor);
viewer->Render();
}
else {
vtkInteractorStyleTrackballCamera::OnLeftButtonDown();
}
}
void LineInteractorStyle::OnRightButtonDown()
{
vtkInteractorStyleTrackballCamera::OnRightButtonDown();
}
Input2dImageType::Pointer read2dImage(const char * path)
{
reader2d->SetFileName(path);
reader2d->SetImageIO(gdcmImageIO);
reader2d->Update();
return reader2d->GetOutput();
}
void showImage(Input2dImageType::Pointer itkImage)
{
viewer->SetInputData(itkToVtk(itkImage));
viewer->SetSliceOrientationToXY();
viewer->Render();
viewer->GetRenderer()->SetBackground(0.5, 0.5, 0.5);
vtkSmartPointer<LineInteractorStyle> style = vtkSmartPointer<LineInteractorStyle>::New();
style->itkImage = itkImage;
vtkSmartPointer<vtkRenderWindowInteractor> rwi =
vtkSmartPointer<vtkRenderWindowInteractor>::New();
//设置交互属性
viewer->SetupInteractor(rwi);
viewer->GetRenderer()->GetRenderWindow()->GetInteractor()->SetInteractorStyle(style);
rwi->Start();
viewer->Render();
}
int main()
{
Input2dImageType::Pointer itkImage = read2dImage("C:\\Users\\Desktop\\segment\\IM12");
showImage(itkImage);
return 0;
}