ITK/Examples/ImageProcessing/ScalarConnectedComponentImageFilter: Difference between revisions

From KitwarePublic
< ITK‎ | Examples
Jump to navigationJump to search
(Added statistics summary)
(Deprecated content that is moved to sphinx)
 
Line 1: Line 1:
<div class="floatcenter">[[File:ITK_Examples_Baseline_ImageProcessing_TestScalarConnectedComponentImageFilter.png]]</div>
{{warning|1=The media wiki content on this page is no longer maintainedThe 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.}}
==ScalarConnectedComponentImageFilter.cxx==
<source lang="cpp">
#include "itkImage.h"
#include "itkImageFileReader.h"
#include "itkLabelToRGBImageFilter.h"
#include "itkRelabelComponentImageFilter.h"
#include "itkLabelStatisticsImageFilter.h"
 
#include "itkScalarConnectedComponentImageFilter.h"
 
#include "itksys/SystemTools.hxx"
#include <sstream>
 
#include "QuickView.h"
 
template <typename TImage>
static void CreateImage(TImage* const image);
 
template<typename TImage, typename TLabelImage>
static void SummarizeLabelStatistics (TImage* image,
                                      TLabelImage* labelImage);
 
int main( int argc, char *argv[])
{
  const unsigned int Dimension = 2;
  typedef short                                  PixelType;
  typedef itk::Image<PixelType, Dimension>      ImageType;
 
  typedef itk::RGBPixel<unsigned char>          RGBPixelType;
  typedef itk::Image<RGBPixelType, Dimension>    RGBImageType;
 
  typedef unsigned int                          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();
 
  SummarizeLabelStatistics (image.GetPointer(), relabel->GetOutput());
 
  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);
      }
    }
}
 
template<typename TImage, typename TLabelImage>
void SummarizeLabelStatistics (TImage* image,
                              TLabelImage* labelImage)
{
   typedef itk::LabelStatisticsImageFilter< TImage, TLabelImage >
    LabelStatisticsImageFilterType;
  typename LabelStatisticsImageFilterType::Pointer labelStatisticsImageFilter =
    LabelStatisticsImageFilterType::New();
  labelStatisticsImageFilter->SetLabelInput( labelImage );
  labelStatisticsImageFilter->SetInput(image);
  labelStatisticsImageFilter->UseHistogramsOn(); // needed to compute median
  labelStatisticsImageFilter->Update();
 
  std::cout << "Number of labels: "
            << labelStatisticsImageFilter->GetNumberOfLabels() << std::endl;
  std::cout << std::endl;
 
  typedef typename LabelStatisticsImageFilterType::ValidLabelValuesContainerType ValidLabelValuesType;
  typedef typename LabelStatisticsImageFilterType::LabelPixelType LabelPixelType;
 
  for(typename ValidLabelValuesType::const_iterator vIt = labelStatisticsImageFilter->GetValidLabelValues().begin();
      vIt != labelStatisticsImageFilter->GetValidLabelValues().end();
      ++vIt)
    {
    if ( labelStatisticsImageFilter->HasLabel(*vIt) )
      {
      LabelPixelType labelValue = *vIt;
      std::cout << "Label: " << *vIt << std::endl;
      std::cout << "\tmin: "
                << labelStatisticsImageFilter->GetMinimum( labelValue )
                << std::endl;
      std::cout << "\tmax: "
                << labelStatisticsImageFilter->GetMaximum( labelValue )
                << std::endl;
      std::cout << "\tmedian: "
                << labelStatisticsImageFilter->GetMedian( labelValue )
                << std::endl;
      std::cout << "\tmean: "
                << labelStatisticsImageFilter->GetMean( labelValue )
                << std::endl;
      std::cout << "\tsigma: "
                << labelStatisticsImageFilter->GetSigma( labelValue )
                << std::endl;
      std::cout << "\tvariance: "
                << labelStatisticsImageFilter->GetVariance( labelValue )
                << std::endl;
      std::cout << "\tsum: "
                << labelStatisticsImageFilter->GetSum( labelValue )
                << std::endl;
      std::cout << "\tcount: "
                << labelStatisticsImageFilter->GetCount( labelValue )
                << std::endl;
      std::cout << "\tregion: "
                << labelStatisticsImageFilter->GetRegion( labelValue )
                << std::endl;
      }
    }
 
  }
</source>
 
{{ITKCMakeLists|{{SUBPAGENAME}}}}

Latest revision as of 15:46, 7 June 2019

Warning: 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.