ITK 分水岭算法

最新推荐文章于 2024-08-31 14:53:24 发布

4月16！

最新推荐文章于 2024-08-31 14:53:24 发布

阅读量778

点赞数 1

分类专栏： ITK图像处理

本文链接：https://blog.csdn.net/qq_35007834/article/details/90446602

版权

ITK图像处理专栏收录该内容

52 篇文章 16 订阅

订阅专栏

https://vtk.org/Wiki/ITK/Examples/Segmentation/WatershedImageFilter

https://itk.org/Doxygen/html/Examples_2Segmentation_2WatershedSegmentation1_8cxx-example.html

这两个都可以参考，结果最后是转化为RGB彩图的，大致效果是这样的，里面参数threshold、level可以调试。

主要的代码是这些：

    typedef   float PixelType;
    const     unsigned int    Dim = 2;
    typedef itk::Image< PixelType, Dim >  floatImageType;
    typedef itk::RGBPixel<unsigned char>       RGBPixelType;
    typedef itk::Image<RGBPixelType, Dim>        RGBImageType;
    typedef itk::Image<itk::IdentifierType, Dim> LabeledImageType; 

    typedef itk::MedianImageFilter<floatImageType,floatImageType> MedianFilterType;
    MedianFilterType::Pointer medianFilter = MedianFilterType::New();
    medianFilter->SetRadius(3);
    medianFilter->SetInput(intensityFilter->GetOutput());
    medianFilter->Update();
    
    typedef itk::GradientMagnitudeRecursiveGaussianImageFilter<floatImageType,floatImageType >  GradientFilterType;
    GradientFilterType::Pointer  gradientMagnitude = GradientFilterType::New();
    gradientMagnitude->SetInput( medianFilter->GetOutput() );
    gradientMagnitude->SetSigma( 1 );
    gradientMagnitude->Update();
    
    // Custom parameters
    typedef itk::WatershedImageFilter<floatImageType> WatershedFilterType;
    WatershedFilterType::Pointer watershed = WatershedFilterType::New();
    watershed->SetThreshold(0.003);
    watershed->SetLevel(0.25);
    watershed->SetInput(gradientMagnitude->GetOutput());
    watershed->Update();
    
    typedef itk::ScalarToRGBColormapImageFilter<LabeledImageType, RGBImageType> RGBFilterType;
    RGBFilterType::Pointer colormapImageFilter = RGBFilterType::New();
    colormapImageFilter->SetInput(watershed->GetOutput());
    colormapImageFilter->SetColormap( RGBFilterType::Jet );
    colormapImageFilter->Update();