1 // =========================================================================
2 // @author Leonardo Florez Valencia
3 // @email florez-l@javeriana.edu.co
4 // =========================================================================
7 #include <tclap/CmdLine.h>
8 #include <itkBinaryThresholdImageFilter.h>
10 #include <fpa/Filters/Image/RandomWalker.h>
11 #include <fpa/Functors/Dijkstra/Image/Gaussian.h>
12 #include "Functions.h"
14 // -------------------------------------------------------------------------
15 const unsigned int Dim = 3;
17 typedef unsigned char TLabel;
18 typedef itk::NumericTraits< TPixel >::RealType TScalar;
19 typedef itk::Image< TPixel, Dim > TImage;
20 typedef itk::Image< TLabel, Dim > TLabels;
22 // -------------------------------------------------------------------------
23 int main( int argc, char* argv[] )
25 typedef TCLAP::ValueArg< std::string > _TStringArg;
26 typedef TCLAP::ValueArg< TScalar > _TRealArg;
29 _TStringArg in( "i", "input", "Input image", true, "", "file" );
30 _TStringArg labels( "l", "labels", "Input labels", true, "", "file" );
31 _TStringArg out( "o", "output", "Output image", true, "", "file" );
32 _TRealArg beta( "b", "beta", "Beta", false, 2.5, "value (2.5)" );
33 _TRealArg eps( "e", "epsilon", "Epsilon", false, 1e-5, "value (1e-5)" );
36 TCLAP::CmdLine cmd( "FastRandomWalker", ' ', "1.0.0" );
42 cmd.parse( argc, argv );
44 catch( TCLAP::ArgException& err )
47 << "===============================" << std::endl
48 << "Error caught: " << std::endl
49 << err.error( ) << " " << err.argId( ) << std::endl
50 << "===============================" << std::endl
59 TImage::Pointer input_image;
60 CTBronchi::ReadImage( input_image, in.getValue( ) );
63 TLabels::Pointer input_labels;
64 CTBronchi::ReadImage( input_labels, labels.getValue( ) );
66 // Prepare weight functor
67 typedef fpa::Functors::Dijkstra::Image::Gaussian< TImage, TScalar > TWeight;
68 TWeight::Pointer weight = TWeight::New( );
69 weight->SetBeta( beta.getValue( ) );
70 weight->SetEpsilon( eps.getValue( ) );
73 typedef fpa::Filters::Image::RandomWalker< TImage, TLabels, TScalar > TFilter;
74 TFilter::Pointer filter = TFilter::New( );
75 filter->SetInputImage( input_image );
76 filter->SetInputLabels( input_labels );
77 filter->SetWeightFunction( weight );
78 double t = CTBronchi::MeasureTime( filter );
79 std::cout << "FastRandomWalk executed in " << t << " s" << std::endl;
82 typedef itk::BinaryThresholdImageFilter< TLabels, TLabels > _TExtract;
83 typename _TExtract::Pointer extract = _TExtract::New( );
84 extract->SetInput( filter->GetOutputLabels( ) );
85 extract->SetInsideValue( 1 );
86 extract->SetOutsideValue( 0 );
87 extract->SetLowerThreshold( 1 );
88 extract->SetUpperThreshold( 1 );
89 t = CTBronchi::MeasureTime( extract );
90 std::cout << "Extract labels executed in " << t << " s" << std::endl;
93 CTBronchi::WriteImage( extract->GetOutput( ), out.getValue( ) );
95 catch( std::exception& err )
98 << "===============================" << std::endl
99 << "Error caught: " << std::endl
100 << err.what( ) << std::endl
101 << "===============================" << std::endl