1 // =========================================================================
2 // @author Leonardo Florez Valencia
3 // @email florez-l@javeriana.edu.co
4 // =========================================================================
7 #include <itkImageFileReader.h>
8 #include <itkImageFileWriter.h>
9 #include <fpa/Common/OriginalRandomWalker.h>
10 #include <fpa/Functors/Dijkstra/Image/Gaussian.h>
12 // -------------------------------------------------------------------------
13 const unsigned int Dim = 2;
14 typedef unsigned char TPixel;
15 typedef unsigned char TLabel;
16 typedef double TScalar;
18 typedef itk::Image< TPixel, Dim > TImage;
19 typedef itk::Image< TLabel, Dim > TLabels;
21 // -------------------------------------------------------------------------
22 int main( int argc, char* argv[] )
24 // Get input arguments
28 << "Usage: " << argv[ 0 ]
29 << " input_image input_labels output_labels output_probabilities"
30 << " beta eps normalize"
35 std::string input_image_filename, input_labels_filename;
36 std::string output_labels_filename, output_probabilities_filename;
39 input_image_filename = argv[ 1 ];
40 input_labels_filename = argv[ 2 ];
41 output_labels_filename = argv[ 3 ];
42 output_probabilities_filename = argv[ 4 ];
43 beta = std::atof( argv[ 5 ] );
44 eps = std::atof( argv[ 6 ] );
45 normalize = ( argv[ 7 ][ 0 ] == '1' );
48 typedef itk::ImageFileReader< TImage > TImageReader;
49 TImageReader::Pointer input_image_reader = TImageReader::New( );
50 input_image_reader->SetFileName( input_image_filename );
53 typedef itk::ImageFileReader< TLabels > TLabelsReader;
54 TLabelsReader::Pointer input_labels_reader = TLabelsReader::New( );
55 input_labels_reader->SetFileName( input_labels_filename );
57 // Random walker function
58 typedef fpa::Functors::Dijkstra::Image::Gaussian< TImage, TScalar > TFunction;
59 TFunction::Pointer edge = TFunction::New( );
60 edge->SetBeta( beta );
61 edge->SetEpsilon( eps );
62 edge->TreatAsWeightOff( );
65 typedef fpa::Common::OriginalRandomWalker< TImage, TLabels, TScalar > TFilter;
66 TFilter::Pointer rw = TFilter::New( );
67 rw->SetInput( input_image_reader->GetOutput( ) );
68 rw->SetInputLabels( input_labels_reader->GetOutput( ) );
69 rw->SetEdgeFunction( edge );
72 typedef itk::ImageFileWriter< TFilter::TLabels > TLabelsWriter;
73 TLabelsWriter::Pointer output_labels_writer = TLabelsWriter::New( );
74 output_labels_writer->SetFileName( output_labels_filename );
75 output_labels_writer->SetInput( rw->GetOutput( ) );
78 typedef itk::ImageFileWriter< TFilter::TScalarImage > TProbabilitesWriter;
79 TProbabilitesWriter::Pointer output_probabilities_writer =
80 TProbabilitesWriter::New( );
81 output_probabilities_writer->SetFileName( output_probabilities_filename );
82 output_probabilities_writer->SetInput( rw->GetOutputProbabilities( ) );
86 output_labels_writer->Update( );
87 output_probabilities_writer->Update( );
89 catch( std::exception& err )
91 std::cerr << "Error caught: " << err.what( ) << std::endl;