Difference between revisions of "ITK/Examples/WishList/Segmentation/MultiphaseChanAndVeseSparseFieldLevelSetSegmentation"

From KitwarePublic
< ITK‎ | Examples
Jump to: navigation, search
(Deprecated content that is moved to sphinx)
 
Line 1: Line 1:
==MultiphaseChanAndVeseSparseFieldLevelSetSegmentation.cxx==
+
{{warning|1=The media wiki content on this page is no longer maintained. The examples presented on the https://itk.org/Wiki/* pages likely require ITK version 4.13 or earlier releases.   In many cases, the examples on this page no longer conform to the best practices for modern ITK versions.}}
 
 
<source lang="cpp">
 
 
 
#include "itkScalarChanAndVeseSparseLevelSetImageFilter.h"
 
#include "itkScalarChanAndVeseLevelSetFunction.h"
 
#include "itkScalarChanAndVeseLevelSetFunctionData.h"
 
#include "itkConstrainedRegionBasedLevelSetFunctionSharedData.h"
 
#include "itkImageFileReader.h"
 
#include "itkImageFileWriter.h"
 
#include "itkImage.h"
 
#include "itkAtanRegularizedHeavisideStepFunction.h"
 
 
 
int main(int argc, char**argv)
 
{
 
  if( argc < 10 )
 
  {
 
    std::cerr << "Missing arguments" << std::endl;
 
    std::cerr << "Usage: " << std::endl;
 
    std::cerr << argv[0];
 
    std::cerr << " inputLevelSetImage1 inputLevelSetImage2 inputLevelSetImage3";
 
    std::cerr << " inputFeatureImage outputLevelSetImage";
 
    std::cerr << " CurvatureWeight AreaWeight LaplacianWeight";
 
    std::cerr << " VolumeWeight Volume OverlapWeight" << std::endl;
 
    return EXIT_FAILURE;
 
  }
 
 
 
  unsigned int nb_iteration = 50;
 
  double rms = 0.;
 
  double epsilon = 1.5;
 
  double curvature_weight = atof( argv[6] );
 
  double area_weight = atof( argv[7] );
 
  double volume_weight = atof( argv[8] );
 
  double volume = atof( argv[9] );
 
  double overlap_weight = atof( argv[10] );
 
  double l1 = 1.;
 
  double l2 = 1.;
 
 
 
  const unsigned int Dimension = 2;
 
  typedef float ScalarPixelType;
 
 
 
  typedef itk::Image< ScalarPixelType, Dimension > LevelSetImageType;
 
  typedef itk::Image< unsigned char, Dimension > FeatureImageType;
 
  typedef itk::Image< unsigned char, Dimension > OutputImageType;
 
 
 
  typedef itk::ScalarChanAndVeseLevelSetFunctionData< LevelSetImageType,
 
    FeatureImageType > DataHelperType;
 
 
 
  typedef itk::ConstrainedRegionBasedLevelSetFunctionSharedData<
 
    LevelSetImageType, FeatureImageType, DataHelperType > SharedDataHelperType;
 
 
 
  typedef itk::ScalarChanAndVeseLevelSetFunction< LevelSetImageType,
 
    FeatureImageType, SharedDataHelperType > LevelSetFunctionType;
 
 
 
  typedef itk::ScalarChanAndVeseSparseLevelSetImageFilter< LevelSetImageType,
 
    FeatureImageType, OutputImageType, LevelSetFunctionType,
 
    SharedDataHelperType > MultiLevelSetType;
 
 
 
  typedef itk::ImageFileReader< LevelSetImageType > LevelSetReaderType;
 
  typedef itk::ImageFileReader< FeatureImageType > FeatureReaderType;
 
  typedef itk::ImageFileWriter< OutputImageType > WriterType;
 
 
 
  typedef itk::AtanRegularizedHeavisideStepFunction< ScalarPixelType,
 
    ScalarPixelType > DomainFunctionType;
 
 
 
  DomainFunctionType::Pointer domainFunction = DomainFunctionType::New();
 
  domainFunction->SetEpsilon( epsilon );
 
 
 
  LevelSetReaderType::Pointer contourReader1 = LevelSetReaderType::New();
 
  contourReader1->SetFileName( argv[1] );
 
  contourReader1->Update();
 
 
 
  LevelSetReaderType::Pointer contourReader2 = LevelSetReaderType::New();
 
  contourReader2->SetFileName( argv[2] );
 
  contourReader2->Update();
 
 
 
  LevelSetReaderType::Pointer contourReader3 = LevelSetReaderType::New();
 
  contourReader3->SetFileName( argv[3] );
 
  contourReader3->Update();
 
 
 
  FeatureReaderType::Pointer featureReader = FeatureReaderType::New();
 
  featureReader->SetFileName( argv[4] );
 
  featureReader->Update();
 
 
 
  FeatureImageType::Pointer featureImage = featureReader->GetOutput();
 
  LevelSetImageType::Pointer contourImage1 = contourReader1->GetOutput();
 
  LevelSetImageType::Pointer contourImage2 = contourReader2->GetOutput();
 
  LevelSetImageType::Pointer contourImage3 = contourReader3->GetOutput();
 
 
 
  MultiLevelSetType::Pointer levelSetFilter = MultiLevelSetType::New();
 
  levelSetFilter->SetFunctionCount( 3 );
 
  levelSetFilter->SetFeatureImage( featureImage );
 
  levelSetFilter->SetLevelSet( 0, contourImage1 );
 
  levelSetFilter->SetLevelSet( 1, contourImage2 );
 
   levelSetFilter->SetLevelSet( 2, contourImage3 );
 
  levelSetFilter->SetNumberOfIterations( nb_iteration );
 
  levelSetFilter->SetMaximumRMSError( rms );
 
  levelSetFilter->SetUseImageSpacing( 1 );
 
  levelSetFilter->SetInPlace( false );
 
 
 
  for ( unsigned int i = 0; i < 3; i++ )
 
  {
 
    levelSetFilter->GetDifferenceFunction(i)->SetDomainFunction( domainFunction );
 
    levelSetFilter->GetDifferenceFunction(i)->SetCurvatureWeight( curvature_weight );
 
    levelSetFilter->GetDifferenceFunction(i)->SetAreaWeight( area_weight );
 
    levelSetFilter->GetDifferenceFunction(i)->SetOverlapPenaltyWeight( overlap_weight );
 
    levelSetFilter->GetDifferenceFunction(i)->SetVolumeMatchingWeight( volume_weight );
 
    levelSetFilter->GetDifferenceFunction(i)->SetVolume( volume );
 
    levelSetFilter->GetDifferenceFunction(i)->SetLambda1( l1 );
 
    levelSetFilter->GetDifferenceFunction(i)->SetLambda2( l2 );
 
  }
 
 
 
  levelSetFilter->Update();
 
 
 
  WriterType::Pointer writer = WriterType::New();
 
  writer->SetInput( levelSetFilter->GetOutput() );
 
  writer->SetFileName( argv[5] );
 
 
 
  try
 
  {
 
    writer->Update();
 
  }
 
  catch( itk::ExceptionObject & excep )
 
  {
 
    std::cerr << "Exception caught !" << std::endl;
 
    std::cerr << excep << std::endl;
 
    return -1;
 
  }
 
 
 
  return EXIT_SUCCESS;
 
}
 
 
 
</source>
 
 
 
{{ITKCMakeLists|{{SUBPAGENAME}}}}
 

Latest revision as of 17:58, 7 June 2019

Warning: The media wiki content on this page is no longer maintained. The examples presented on the https://itk.org/Wiki/* pages likely require ITK version 4.13 or earlier releases. In many cases, the examples on this page no longer conform to the best practices for modern ITK versions.