|
|
| (2 intermediate revisions by one other user not shown) |
| Line 1: |
Line 1: |
| ==ScalarConnectedComponentImageFilter.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 "itkImageFileReader.h"
| |
| #include "itkLabelToRGBImageFilter.h"
| |
| #include "itkRelabelComponentImageFilter.h"
| |
| | |
| #include "itkScalarConnectedComponentImageFilter.h"
| |
| | |
| #include "itksys/SystemTools.hxx"
| |
| #include <sstream>
| |
| | |
| #include "QuickView.h"
| |
| | |
| template <typename TImage>
| |
| static void CreateImage(TImage* const image);
| |
| | |
| int main( int argc, char *argv[])
| |
| {
| |
| const unsigned int Dimension = 2;
| |
| typedef unsigned char PixelType;
| |
| typedef itk::Image<PixelType, Dimension> ImageType;
| |
| | |
| typedef itk::RGBPixel<unsigned char> RGBPixelType;
| |
| typedef itk::Image<RGBPixelType, Dimension> RGBImageType;
| |
| | |
| typedef unsigned short LabelPixelType;
| |
| typedef itk::Image<LabelPixelType, Dimension > LabelImageType;
| |
| | |
| ImageType::Pointer image;
| |
| PixelType distanceThreshold = 4;
| |
| if( argc < 2 )
| |
| {
| |
| image = ImageType::New();
| |
| CreateImage(image.GetPointer());
| |
| }
| |
| else
| |
| {
| |
| typedef itk::ImageFileReader<ImageType> ReaderType;
| |
| ReaderType::Pointer reader = ReaderType::New();
| |
| reader->SetFileName(argv[1]);
| |
| reader->Update();
| |
| | |
| if (argc > 2)
| |
| {
| |
| distanceThreshold = static_cast<PixelType> (atoi(argv[2]));
| |
| }
| |
| image = reader->GetOutput();
| |
| }
| |
| | |
| | |
| typedef itk::ScalarConnectedComponentImageFilter <ImageType, LabelImageType >
| |
| ConnectedComponentImageFilterType;
| |
| | |
| ConnectedComponentImageFilterType::Pointer connected =
| |
| ConnectedComponentImageFilterType::New ();
| |
| connected->SetInput(image);
| |
| connected->SetDistanceThreshold(distanceThreshold);
| |
| | |
| typedef itk::RelabelComponentImageFilter <LabelImageType, LabelImageType >
| |
| RelabelFilterType;
| |
| RelabelFilterType::Pointer relabel =
| |
| RelabelFilterType::New();
| |
| RelabelFilterType::ObjectSizeType minSize = 20;
| |
| if (argc > 3)
| |
| {
| |
| minSize = atoi(argv[3]);
| |
| }
| |
| relabel->SetInput(connected->GetOutput());
| |
| relabel->SetMinimumObjectSize(minSize);
| |
| relabel->Update();
| |
| | |
| typedef itk::LabelToRGBImageFilter<LabelImageType, RGBImageType> RGBFilterType;
| |
| RGBFilterType::Pointer rgbFilter = RGBFilterType::New();
| |
| rgbFilter->SetInput( relabel->GetOutput() );
| |
| | |
| QuickView viewer;
| |
| viewer.AddImage(
| |
| image.GetPointer(),true,
| |
| argc > 1 ? itksys::SystemTools::GetFilenameName(argv[1]) : "Generated image");
| |
| | |
| std::stringstream desc;
| |
| desc << "Scalar Connected Components:\n# of Objects: "
| |
| << relabel->GetNumberOfObjects()
| |
| << " Threshold: "
| |
| << itk::NumericTraits<ConnectedComponentImageFilterType::OutputPixelType>::PrintType(connected->GetDistanceThreshold())
| |
| << " Min Size: " << relabel->GetMinimumObjectSize();
| |
| viewer.AddRGBImage(
| |
| rgbFilter->GetOutput(),
| |
| true,
| |
| desc.str());
| |
| | |
| viewer.Visualize();
| |
| | |
| return EXIT_SUCCESS;
| |
| }
| |
| | |
| template <typename TImage>
| |
| void CreateImage(TImage* const image)
| |
| {
| |
| // Create an image with 2 connected components
| |
| typename TImage::IndexType start = {{0,0}};
| |
| start[0] = 0;
| |
| start[1] = 0;
| |
| | |
| typename TImage::SizeType size;
| |
| unsigned int NumRows = 200;
| |
| unsigned int NumCols = 300;
| |
| size[0] = NumRows;
| |
| size[1] = NumCols;
| |
| | |
| typename TImage::RegionType region(start, size); | |
| | |
| image->SetRegions(region);
| |
| image->Allocate();
| |
| | |
| // Make a square
| |
| for(typename TImage::IndexValueType r = 20; r < 80; r++)
| |
| {
| |
| for(typename TImage::IndexValueType c = 30; c < 100; c++)
| |
| {
| |
| typename TImage::IndexType pixelIndex = {{r,c}};
| |
| | |
| image->SetPixel(pixelIndex, 255);
| |
| }
| |
| }
| |
| | |
| // Make another square
| |
| for(typename TImage::IndexValueType r = 100; r < 130; r++)
| |
| {
| |
| for(typename TImage::IndexValueType c = 115; c < 160; c++)
| |
| {
| |
| typename TImage::IndexType pixelIndex = {{r,c}};
| |
| | |
| image->SetPixel(pixelIndex, 255);
| |
| }
| |
| }
| |
| }
| |
| </source>
| |
| | |
| {{ITKCMakeLists|{{SUBPAGENAME}}}}
| |
