|
|
| Line 1: |
Line 1: |
| ==LabelShapeKeepNObjectsImageFilter.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 "itkImage.h"
| |
| #include "itkImageFileWriter.h"
| |
| #include "itkImageRegionIterator.h"
| |
| #include "itkBinaryImageToLabelMapFilter.h"
| |
| #include "itkLabelShapeKeepNObjectsImageFilter.h"
| |
| #include "itkLabelMapToLabelImageFilter.h"
| |
| #include "itkScalarToRGBColormapImageFilter.h"
| |
| | |
| typedef itk::Image<unsigned char, 2> ImageType;
| |
| typedef itk::Image<unsigned char, 2> LabelImageType;
| |
| static void CreateImage(ImageType::Pointer image);
| |
| | |
| int main(int, char *[])
| |
| {
| |
| ImageType::Pointer image = ImageType::New();
| |
| CreateImage(image);
| |
| | |
| typedef itk::BinaryImageToLabelMapFilter<ImageType> BinaryImageToLabelMapFilterType;
| |
| BinaryImageToLabelMapFilterType::Pointer binaryImageToLabelMapFilter = BinaryImageToLabelMapFilterType::New();
| |
| binaryImageToLabelMapFilter->SetInput(image);
| |
| binaryImageToLabelMapFilter->Update();
| |
|
| |
| typedef itk::LabelMapToLabelImageFilter<BinaryImageToLabelMapFilterType::OutputImageType, LabelImageType> LabelMapToLabelImageFilterType;
| |
| LabelMapToLabelImageFilterType::Pointer labelMapToLabelImageFilter = LabelMapToLabelImageFilterType::New();
| |
| labelMapToLabelImageFilter->SetInput(binaryImageToLabelMapFilter->GetOutput());
| |
| labelMapToLabelImageFilter->Update();
| |
| | |
| typedef itk::LabelShapeKeepNObjectsImageFilter< LabelImageType > LabelShapeKeepNObjectsImageFilterType;
| |
| LabelShapeKeepNObjectsImageFilterType::Pointer labelShapeKeepNObjectsImageFilter = LabelShapeKeepNObjectsImageFilterType::New();
| |
| labelShapeKeepNObjectsImageFilter->SetInput( labelMapToLabelImageFilter->GetOutput() );
| |
| labelShapeKeepNObjectsImageFilter->SetBackgroundValue( 0 );
| |
| labelShapeKeepNObjectsImageFilter->SetNumberOfObjects( 2 );
| |
| //KeepNObjects->ReverseOrderingOn();
| |
| //labelShapeKeepNObjectsImageFilter->SetAttribute( LabelObjectType::PERIMETER );
| |
| labelShapeKeepNObjectsImageFilter->SetAttribute( LabelShapeKeepNObjectsImageFilterType::LabelObjectType::PERIMETER);
| |
| labelShapeKeepNObjectsImageFilter->Update();
| |
|
| |
| typedef itk::RGBPixel<unsigned char> RGBPixelType;
| |
| typedef itk::Image<RGBPixelType, 2> RGBImageType;
| |
| | |
| // Color each label/object a different color
| |
| typedef itk::ScalarToRGBColormapImageFilter<LabelImageType, RGBImageType> RGBFilterType;
| |
| RGBFilterType::Pointer colormapImageFilter = RGBFilterType::New();
| |
| colormapImageFilter->SetInput(labelShapeKeepNObjectsImageFilter->GetOutput());
| |
| colormapImageFilter->SetColormap( RGBFilterType::Jet );
| |
| colormapImageFilter->Update();
| |
|
| |
| typedef itk::ImageFileWriter< RGBImageType > WriterType;
| |
| WriterType::Pointer writer = WriterType::New();
| |
| writer->SetInput( colormapImageFilter->GetOutput() );
| |
| writer->SetFileName("output.png");
| |
| writer->Update();
| |
| | |
| return EXIT_SUCCESS;
| |
| }
| |
| | |
| void CreateImage(ImageType::Pointer image)
| |
| {
| |
| // Create a black image with three white squares
| |
| ImageType::IndexType start;
| |
| start.Fill(0);
| |
| | |
| ImageType::SizeType size;
| |
| size.Fill(200);
| |
| | |
| ImageType::RegionType region;
| |
| region.SetSize(size);
| |
| region.SetIndex(start);
| |
| image->SetRegions(region);
| |
| image->Allocate();
| |
| | |
| itk::ImageRegionIterator<ImageType> imageIterator(image,image->GetLargestPossibleRegion()); | |
| | |
| // Make a square
| |
| while(!imageIterator.IsAtEnd())
| |
| {
| |
| if(((imageIterator.GetIndex()[0] > 5 && imageIterator.GetIndex()[0] < 20) &&
| |
| (imageIterator.GetIndex()[1] > 5 && imageIterator.GetIndex()[1] < 20)) ||
| |
| ((imageIterator.GetIndex()[0] > 50 && imageIterator.GetIndex()[0] < 60) &&
| |
| (imageIterator.GetIndex()[1] > 50 && imageIterator.GetIndex()[1] < 60)) ||
| |
| ((imageIterator.GetIndex()[0] > 100 && imageIterator.GetIndex()[0] < 130) &&
| |
| (imageIterator.GetIndex()[1] > 100 && imageIterator.GetIndex()[1] < 130)) )
| |
| {
| |
| imageIterator.Set(255);
| |
| }
| |
| else
| |
| {
| |
| imageIterator.Set(0);
| |
| }
| |
| | |
| ++imageIterator;
| |
| }
| |
| | |
| typedef itk::ImageFileWriter< ImageType > WriterType;
| |
| WriterType::Pointer writer = WriterType::New();
| |
| writer->SetFileName("input.png");
| |
| writer->SetInput(image);
| |
| writer->Update();
| |
|
| |
| }
| |
| | |
| </source>
| |
| | |
| {{ITKCMakeLists|{{SUBPAGENAME}}}}
| |
