ITK 计算 EuclideanDistance

1、代码

1 #include "itkImage.h" 2 #include "itkImageFileReader.h" 3 #include "itkSignedMaurerDistanceMapImageFilter.h" 4 5 #include "itksys/SystemTools.hxx" 6 #include <sstream> 7 //#define ENABLE_QUICKVIEW 8 #ifdef ENABLE_QUICKVIEW 9 #include "QuickVieh" 10 #endif 11 12 using UnsignedCharImageType = itk::Image<unsigned char, 2>; 13 using FloatImageType = itk::Image<float, 2>; 14 15 static void 16 CreateImage(UnsignedCharImageType::Pointer image); 17 18 int 19 main(int argc, char * argv[]) 20 26 else 27 34 35 using SignedMaurerDistanceMapImageFilterType = 36 itk::SignedMaurerDistanceMapImageFilter<UnsignedCharImageType, FloatImageType>; 37 SignedMaurerDistanceMapImageFilterType::Pointer distanceMapImageFilter = 38 SignedMaurerDistanceMapImageFilterType::New(); 39 distanceMapImageFilter>SetInput(image); 40 distanceMapImageFilter>Update(); 41 auto outputImage = distanceMapImageFilter>GetOutput(); 42 43 for (unsigned int i = 40; i < 50; ++i) 44 ; 48 //pixel.Fill(i); 49 std::cout << outputImage>GetPixel(pixel) << "\t"; 50 } 51 std::cout << std::endl; 52 } 53 54 #ifdef ENABLE_QUICKVIEW 55 QuickView viewer; 56 viewer.AddImage( 57 image.GetPointer(), true, argc > 1 ? itksys::SystemTools::GetFilenameName(argv[1]) : "Generated image"); 58 59 std::stringstream desc; 60 desc << "Signed Maurer Distance"; 61 viewer.AddImage(distanceMapImageFilter>GetOutput(), true, desc.str()); 62 viewer.Visualize(); 64 #endif 65 66 return EXIT_SUCCESS; 67 } 68 69 70 void 71 CreateImage(UnsignedCharImageType::Pointer image) 72 92 }

2、CMakeLists.txt

关于 CMakeLists.txt 可以修改只保留itk，下面是因为我测试过itk和vtk一起使用才写成下面的样子的。

1 cmake_minimum_required(VERSION 2.8 FATAL_ERROR) 2 project(testfilter) 3 4 set(VTK_DIR D:/ProgramFiles/vtk9.0/lib/cmake/vtk9.0) 5 find_package(VTK CPONENTS 6 vtkCmonCore 7 vtkCmonDataModel 8 vtkFiltersSources 9 vtkIOImage 10 vtkImagingStencil 11 vtkFiltersCore 12 vtkRenderingCore 13 vtkFiltersHybrid 14 vtkInteractionStyle 15 vtkRenderingFreeType 16 vtkRenderingOpenGL2 17 vtkRenderingVolumeOpenGL2 18 vtkImagingColor 19 QUIET 20 ) 21 22 if (NOT VTK_FOUND) 23 message("Skipping testfilter: $") 24 return() 25 endif() 26 message (STATUS "VTK_VERSION: $") 27 28 if (VTK_VERSION VERSION_LESS "8.90.0") 29 # old system 30 include($) 31 endif() 32 33 set(ITK_DIR D:/ProgramFiles/ITK5.2/lib/cmake/ITK5.2) 34 35 FIND_PACKAGE(ITK REQUIRED) 36 INCLUDE($) 37 38 file(GLOB_RECURSE mains $ *pp) 39 message(STATUS $) 40 foreach(mainfile IN LISTS mains) 41 # Get file name without directory 42 get_filename_cponent(mainname $ NAME_WE) 43 add_executable($ $) 44 target_link_libraries ($ $ $) 45 # vtk_module_autoinit is needed 46 if (NOT VTK_VERSION VERSION_LESS "8.90.0") 47 # vtk_module_autoinit is needed 48 vtk_module_autoinit( 49 TARGETS $ 50 MODULES $ 51 ) 52 endif() 53 54 endforeach()

3、测试输出

下面的输出是我们想要的结果

1 0 1 1.41421 2.23607 2.82843 3.60555 4.24264 5 5.65685 6.40312 2 1 0 1 1.41421 2.23607 2.82843 3.60555 4.24264 5 5.65685 3 1.41421 1 0 1 1.41421 2.23607 2.82843 3.60555 4.24264 5 4 2.23607 1.41421 1 0 1 1.41421 2.23607 2.82843 3.60555 4.24264 5 2.82843 2.23607 1.41421 1 0 1 1.41421 2.23607 2.82843 3.60555 6 3.60555 2.82843 2.23607 1.41421 1 0 1 1.41421 2.23607 2.82843 7 4.24264 3.60555 2.82843 2.23607 1.41421 1 0 1 1.41421 2.23607 8 5 4.24264 3.60555 2.82843 2.23607 1.41421 1 0 1 1.41421 9 5.65685 5 4.24264 3.60555 2.82843 2.23607 1.41421 1 0 1 10 6.40312 5.65685 5 4.24264 3.60555 2.82843 2.23607 1.41421 1 0

该结果与scipy中ndimage中的distance_transform_edt函数作用相同

ndimage.distance_transform_edt(seed == 0)

上一篇：NRRD批量转换成NIFTI

下一篇：SimpleITK 图像对齐

---

ITK