ITK/Examples/ImageProcessing/SigmoidImageFilter

From KitwarePublic
< ITK‎ | Examples
Revision as of 05:24, 23 December 2010 by Lorensen (talk | contribs) (Use QuickView)
Jump to navigationJump to search

This example passes every pixel in an image through a Sigmoid function. The output displays the input image and the filtered image.

SigmoidImageFilter.cxx

<source lang="cpp">

  1. include "itkImage.h"
  2. include "itkImageFileReader.h"
  3. include "itkSigmoidImageFilter.h"
  1. include "QuickView.h"

int main(int argc, char *argv[]) {

 if(argc < 2)
   {
   std::cerr << "Usage: ";
   std::cerr << argv[0] << " inputImageFile [alpha] [beta]" << std::endl;
   return EXIT_FAILURE;
   }
 double alpha = 5.0;
 double beta = 10.0;
 if (argc > 2)
   {
   alpha = atof(argv[2]);
   }
 if (argc > 3)
   {
   beta = atof(argv[3]);
   }
 typedef itk::Image<unsigned char, 2> ImageType;
 typedef itk::ImageFileReader<ImageType> ReaderType;
 ReaderType::Pointer reader = ReaderType::New();
 reader->SetFileName(argv[1]);
 typedef itk::SigmoidImageFilter <ImageType, ImageType>
   SigmoidImageFilterType;
 SigmoidImageFilterType::Pointer sigmoidFilter
   = SigmoidImageFilterType::New();
 sigmoidFilter->SetInput(reader->GetOutput());
 sigmoidFilter->SetOutputMinimum(0);
 sigmoidFilter->SetOutputMaximum(255);
 sigmoidFilter->SetAlpha(alpha);
 sigmoidFilter->SetBeta(beta);
 
 QuickView viewer;
 viewer.AddImage(
   reader->GetOutput(),true,
   itksys::SystemTools::GetFilenameName(argv[1]));  
 std::stringstream desc;
 desc << "SigmoidImageFilter\nalpha = " << alpha
      << " beta = " << beta;
 viewer.AddImage(
   sigmoidFilter->GetOutput(),
   true,
   desc.str());  
 viewer.Visualize();
 return EXIT_SUCCESS;

} </source>

CMakeLists.txt

<source lang="cmake"> cmake_minimum_required(VERSION 2.6)

PROJECT(SigmoidImageFilter)

include_directories(/home/doriad/src/ITK/Wrapping/WrapITK/ExternalProjects/ItkVtkGlue/src/)

FIND_PACKAGE(VTK REQUIRED) INCLUDE(${VTK_USE_FILE})

FIND_PACKAGE(ITK REQUIRED) INCLUDE(${ITK_USE_FILE})

ADD_EXECUTABLE(SigmoidImageFilter SigmoidImageFilter.cxx) TARGET_LINK_LIBRARIES(SigmoidImageFilter vtkHybrid ITKBasicFilters ITKCommon ITKIO)

</source>