ITK/Examples/Smoothing/MedianImageFilter
From KitwarePublic
Jump to navigationJump to search
Replace every pixel in an image with the median of its neighbors.
MedianImageFilter.cxx
<source lang="cpp">
- include "itkImage.h"
- include "itkImageFileReader.h"
- include "itkMedianImageFilter.h"
- include "itkSubtractImageFilter.h"
- include "itksys/SystemTools.hxx"
- include <sstream>
- include "QuickView.h"
int main(int argc, char * argv[]) {
// Verify command line arguments if( argc < 2 ) { std::cerr << "Usage: " << std::endl; std::cerr << argv[0] << " InputImageFile [radius]" << std::endl; return EXIT_FAILURE; } std::string inputFilename = argv[1];
// Setup types typedef itk::Image<float, 2 > ImageType; typedef itk::ImageFileReader<ImageType> ReaderType; typedef itk::MedianImageFilter<ImageType, ImageType > FilterType; typedef itk::SubtractImageFilter<ImageType> SubtractType;
// Create and setup a reader ReaderType::Pointer reader = ReaderType::New(); reader->SetFileName( inputFilename );
// Create and setup a median filter FilterType::Pointer medianFilter = FilterType::New(); FilterType::InputSizeType radius; radius.Fill(2); if (argc > 2) { radius.Fill(atoi(argv[2])); }
medianFilter->SetRadius(radius); medianFilter->SetInput( reader->GetOutput() );
SubtractType::Pointer diff = SubtractType::New(); diff->SetInput1(reader->GetOutput()); diff->SetInput2(medianFilter->GetOutput());
QuickView viewer; viewer.AddImage( reader->GetOutput(),true, itksys::SystemTools::GetFilenameName(inputFilename));
std::stringstream desc; desc << "MedianImageFilter, radius = " << radius; viewer.AddImage( medianFilter->GetOutput(), true, desc.str());
std::stringstream desc2; desc2 << "Original - Median"; viewer.AddImage( diff->GetOutput(), true, desc2.str());
viewer.Visualize();
return EXIT_SUCCESS;
} </source>
CMakeLists.txt
<source lang="cmake"> cmake_minimum_required(VERSION 2.6)
PROJECT(MedianImageFilter)
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(MedianImageFilter MedianImageFilter.cxx) TARGET_LINK_LIBRARIES(MedianImageFilter vtkHybrid ITKBasicFilters ITKIO ITKCommon)
</source>