ITK/Examples/Images/NormalizedCorrelationImageFilter: Difference between revisions

From KitwarePublic
< ITK‎ | Examples
Jump to navigationJump to search
No edit summary
Line 1: Line 1:
Output is all black?
Output neither the minimum nor maximum of the correlation image is the "correct" result (the pixel that we extracted the patch to correlate).


==NormalizedCorrelationImageFilter.cxx==
==NormalizedCorrelationImageFilter.cxx==
<source lang="cpp">
<source lang="cpp">
#include <itkImage.h>
#include "itkImage.h"
#include <itkImageFileReader.h>
#include "itkImageFileReader.h"
#include <itkNormalizedCorrelationImageFilter.h>
#include "itkNormalizedCorrelationImageFilter.h"
#include <itkRegionOfInterestImageFilter.h>
#include "itkRegionOfInterestImageFilter.h"
#include <itkImageKernelOperator.h>
#include "itkImageKernelOperator.h"
#include <itkCastImageFilter.h>
#include "itkRescaleIntensityImageFilter.h"
#include <itkRescaleIntensityImageFilter.h>
#include "itkImageFileWriter.h"
 
#include "itkMinimumMaximumImageCalculator.h"
#include <itkImageToVTKImageFilter.h>
#include "QuickView.h"
 
#include "vtkImageViewer.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkSmartPointer.h"
#include "vtkImageActor.h"
#include "vtkInteractorStyleImage.h"
#include "vtkRenderer.h"


#include <iostream>
#include <iostream>
#include <string>
#include <string>
typedef itk::Image<float, 2> FloatImageType;
typedef itk::Image<unsigned char, 2> UnsignedCharImageType;


int main(int argc, char *argv[])
int main(int argc, char *argv[])
Line 30: Line 26:
     return EXIT_FAILURE;
     return EXIT_FAILURE;
     }
     }
 
 
   std::string filename = argv[1];
   std::string filename = argv[1];
 
    
   typedef itk::Image<unsigned char, 2> UnsignedCharImageType;
  typedef itk::Image<float, 2> FloatImageType;
   typedef itk::ImageFileReader<FloatImageType> ReaderType;
   typedef itk::ImageFileReader<FloatImageType> ReaderType;
  typedef itk::ImageToVTKImageFilter<UnsignedCharImageType> ConnectorType;


   // Read the image
   // Read the image
   ReaderType::Pointer reader = ReaderType::New();
   ReaderType::Pointer reader = ReaderType::New();
   reader->SetFileName(filename.c_str());
   reader->SetFileName(filename.c_str());
 
   reader->Update();
   // Setup the original image actor
  typedef itk::CastImageFilter< FloatImageType,
                                UnsignedCharImageType > CastFilterType;
  CastFilterType::Pointer originalCastFilter = CastFilterType::New();
  originalCastFilter->SetInput(reader->GetOutput());
    
    
  ConnectorType::Pointer originalConnector = ConnectorType::New();
  originalConnector->SetInput(originalCastFilter->GetOutput());
  vtkSmartPointer<vtkImageActor> originalActor =
    vtkSmartPointer<vtkImageActor>::New();
  originalActor->SetInput(originalConnector->GetOutput());
   // Extract a small region
   // Extract a small region
   typedef itk::RegionOfInterestImageFilter< FloatImageType,
   typedef itk::RegionOfInterestImageFilter< FloatImageType,
Line 61: Line 42:
   ExtractFilterType::Pointer extractFilter = ExtractFilterType::New();
   ExtractFilterType::Pointer extractFilter = ExtractFilterType::New();


   UnsignedCharImageType::IndexType start;
   FloatImageType::IndexType start;
   start[0] = 50;
   start.Fill(50);
  start[1] = 50;


   UnsignedCharImageType::SizeType size;
   FloatImageType::SizeType size;
   size[0] = 100;
   size.Fill(100);
  size[1] = 100;


   UnsignedCharImageType::RegionType desiredRegion;
   FloatImageType::RegionType desiredRegion(start,size);
  desiredRegion.SetSize(size);
  desiredRegion.SetIndex(start);


   extractFilter->SetRegionOfInterest(desiredRegion);
   extractFilter->SetRegionOfInterest(desiredRegion);
   extractFilter->SetInput(reader->GetOutput());
   extractFilter->SetInput(reader->GetOutput());
 
   extractFilter->Update();
   // Display extracted region (kernel)
  CastFilterType::Pointer kernelCastFilter = CastFilterType::New();
  kernelCastFilter->SetInput(extractFilter->GetOutput());
    
    
  ConnectorType::Pointer extractedConnector = ConnectorType::New();
  extractedConnector->SetInput(kernelCastFilter->GetOutput());
  vtkSmartPointer<vtkImageActor> extractedActor =
    vtkSmartPointer<vtkImageActor>::New();
  extractedActor->SetInput(extractedConnector->GetOutput());
   // Perform normalized correlation
   // Perform normalized correlation
   // <input type, mask type (not used), output type>
   // <input type, mask type (not used), output type>
Line 93: Line 60:
   itk::ImageKernelOperator<float> kernelOperator;
   itk::ImageKernelOperator<float> kernelOperator;
   kernelOperator.SetImageKernel(extractFilter->GetOutput());
   kernelOperator.SetImageKernel(extractFilter->GetOutput());
  kernelOperator.CreateToRadius(extractFilter->GetOutput()->GetLargestPossibleRegion().GetSize());
    
    
   CorrelationFilterType::Pointer correlationFilter = CorrelationFilterType::New();
   CorrelationFilterType::Pointer correlationFilter = CorrelationFilterType::New();
   correlationFilter->SetInput(reader->GetOutput());
   correlationFilter->SetInput(reader->GetOutput());
   correlationFilter->SetTemplate(kernelOperator);
   correlationFilter->SetTemplate(kernelOperator);
  correlationFilter->Update();
  typedef itk::MinimumMaximumImageCalculator <FloatImageType>
          MinimumMaximumImageCalculatorType;
  MinimumMaximumImageCalculatorType::Pointer minimumMaximumImageCalculatorFilter
          = MinimumMaximumImageCalculatorType::New ();
  minimumMaximumImageCalculatorFilter->SetImage(correlationFilter->GetOutput());
  minimumMaximumImageCalculatorFilter->Compute();
 
  std::cout << "Maximum: " << minimumMaximumImageCalculatorFilter->GetIndexOfMaximum() << std::endl;
  std::cout << "Minimum: " << minimumMaximumImageCalculatorFilter->GetIndexOfMinimum() << std::endl;


  // Display the correlation image
   typedef itk::RescaleIntensityImageFilter< FloatImageType, UnsignedCharImageType > RescaleFilterType;
   typedef itk::RescaleIntensityImageFilter< FloatImageType, UnsignedCharImageType > RescaleFilterType;
  typedef itk::ImageFileWriter<UnsignedCharImageType> WriterType;
  {
   RescaleFilterType::Pointer rescaleFilter = RescaleFilterType::New();
   RescaleFilterType::Pointer rescaleFilter = RescaleFilterType::New();
   rescaleFilter->SetInput(correlationFilter->GetOutput());
   rescaleFilter->SetInput(correlationFilter->GetOutput());
   rescaleFilter->SetOutputMinimum(0);
   rescaleFilter->SetOutputMinimum(0);
   rescaleFilter->SetOutputMaximum(255);
   rescaleFilter->SetOutputMaximum(255);
  rescaleFilter->Update();
 
  WriterType::Pointer writer = WriterType::New();
  writer->SetInput(rescaleFilter->GetOutput());
  writer->SetFileName("correlation.png");
  writer->Update();
  }


   ConnectorType::Pointer correlationConnector = ConnectorType::New();
   {
   correlationConnector->SetInput(rescaleFilter->GetOutput());
  RescaleFilterType::Pointer rescaleFilter = RescaleFilterType::New();
 
   rescaleFilter->SetInput(extractFilter->GetOutput());
   vtkSmartPointer<vtkImageActor> correlationActor =
   rescaleFilter->SetOutputMinimum(0);
    vtkSmartPointer<vtkImageActor>::New();
   rescaleFilter->SetOutputMaximum(255);
   correlationActor->SetInput(correlationConnector->GetOutput());
   rescaleFilter->Update();
 
    
   // Visualize
   WriterType::Pointer writer = WriterType::New();
  vtkSmartPointer<vtkRenderWindow> renderWindow =
   writer->SetInput(rescaleFilter->GetOutput());
    vtkSmartPointer<vtkRenderWindow>::New();
   writer->SetFileName("patch.png");
   renderWindow->SetSize(900, 300);
   writer->Update();
 
   }
   vtkSmartPointer<vtkRenderWindowInteractor> interactor =
    
    vtkSmartPointer<vtkRenderWindowInteractor>::New();
   QuickView viewer;
  interactor->SetRenderWindow(renderWindow);
   viewer.AddImage(reader->GetOutput());
 
   viewer.AddImage(extractFilter->GetOutput());
  double leftViewport[4] = {0.0, 0.0, 0.33, 1.0};
   viewer.AddImage(correlationFilter->GetOutput());
  double centerViewport[4] = {0.33, 0.0, 0.66, 1.0};
   viewer.Visualize();
  double rightViewport[4] = {0.66, 0.0, 1.0, 1.0};
 
  vtkSmartPointer<vtkRenderer> leftRenderer =
    vtkSmartPointer<vtkRenderer>::New();
   renderWindow->AddRenderer(leftRenderer);
  leftRenderer->SetViewport(leftViewport);
   leftRenderer->SetBackground(.6, .5, .4);
 
   vtkSmartPointer<vtkRenderer> centerRenderer =
    vtkSmartPointer<vtkRenderer>::New();
   renderWindow->AddRenderer(centerRenderer);
   centerRenderer->SetViewport(centerViewport);
   centerRenderer->SetBackground(.5, .5, .6);
 
   vtkSmartPointer<vtkRenderer> rightRenderer =
    vtkSmartPointer<vtkRenderer>::New();
  renderWindow->AddRenderer(rightRenderer);
  rightRenderer->SetViewport(rightViewport);
   rightRenderer->SetBackground(.4, .5, .6);
 
  leftRenderer->AddActor(originalActor);
  centerRenderer->AddActor(extractedActor);
  rightRenderer->AddActor(correlationActor);
 
   leftRenderer->ResetCamera();
  rightRenderer->ResetCamera();
 
  renderWindow->Render();
 
   vtkSmartPointer<vtkInteractorStyleImage> style =
    vtkSmartPointer<vtkInteractorStyleImage>::New();
  interactor->SetInteractorStyle(style);
 
  interactor->Start();


   return EXIT_SUCCESS;
   return EXIT_SUCCESS;
}
}
</source>
</source>


Line 168: Line 123:
PROJECT(NormalizedCorrelationImageFilter)
PROJECT(NormalizedCorrelationImageFilter)


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


FIND_PACKAGE(VTK REQUIRED)
FIND_PACKAGE(VTK REQUIRED)
Line 176: Line 131:
INCLUDE(${ITK_USE_FILE})
INCLUDE(${ITK_USE_FILE})


ADD_EXECUTABLE(NormalizedCorrelationImageFilter NormalizedCorrelationImageFilter.cpp)
ADD_EXECUTABLE(NormalizedCorrelationImageFilter NormalizedCorrelationImageFilter.cpp /home/doriad/ITKWikiExamples/ItkVtkGlue/QuickView.cxx)
TARGET_LINK_LIBRARIES(NormalizedCorrelationImageFilter
TARGET_LINK_LIBRARIES(NormalizedCorrelationImageFilter
vtkHybrid
vtkHybrid

Revision as of 16:09, 3 February 2011

Output neither the minimum nor maximum of the correlation image is the "correct" result (the pixel that we extracted the patch to correlate).

NormalizedCorrelationImageFilter.cxx

<source lang="cpp">

  1. include "itkImage.h"
  2. include "itkImageFileReader.h"
  3. include "itkNormalizedCorrelationImageFilter.h"
  4. include "itkRegionOfInterestImageFilter.h"
  5. include "itkImageKernelOperator.h"
  6. include "itkRescaleIntensityImageFilter.h"
  7. include "itkImageFileWriter.h"
  8. include "itkMinimumMaximumImageCalculator.h"
  9. include "QuickView.h"
  1. include <iostream>
  2. include <string>

typedef itk::Image<float, 2> FloatImageType; typedef itk::Image<unsigned char, 2> UnsignedCharImageType;

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

 if(argc < 2)
   {
   std::cerr << "Required: filename" << std::endl;
   return EXIT_FAILURE;
   }
 
 std::string filename = argv[1];
 
 typedef itk::ImageFileReader<FloatImageType> ReaderType;
 // Read the image
 ReaderType::Pointer reader = ReaderType::New();
 reader->SetFileName(filename.c_str());
 reader->Update();
 
 // Extract a small region
 typedef itk::RegionOfInterestImageFilter< FloatImageType,
                                           FloatImageType > ExtractFilterType;
 ExtractFilterType::Pointer extractFilter = ExtractFilterType::New();
 FloatImageType::IndexType start;
 start.Fill(50);
 FloatImageType::SizeType size;
 size.Fill(100);
 FloatImageType::RegionType desiredRegion(start,size);
 extractFilter->SetRegionOfInterest(desiredRegion);
 extractFilter->SetInput(reader->GetOutput());
 extractFilter->Update();
 
 // Perform normalized correlation
 // <input type, mask type (not used), output type>
 typedef itk::NormalizedCorrelationImageFilter<FloatImageType, FloatImageType, FloatImageType> CorrelationFilterType;
 itk::ImageKernelOperator<float> kernelOperator;
 kernelOperator.SetImageKernel(extractFilter->GetOutput());
 kernelOperator.CreateToRadius(extractFilter->GetOutput()->GetLargestPossibleRegion().GetSize());
 
 CorrelationFilterType::Pointer correlationFilter = CorrelationFilterType::New();
 correlationFilter->SetInput(reader->GetOutput());
 correlationFilter->SetTemplate(kernelOperator);
 correlationFilter->Update();
 typedef itk::MinimumMaximumImageCalculator <FloatImageType>
         MinimumMaximumImageCalculatorType;
 MinimumMaximumImageCalculatorType::Pointer minimumMaximumImageCalculatorFilter
         = MinimumMaximumImageCalculatorType::New ();
 minimumMaximumImageCalculatorFilter->SetImage(correlationFilter->GetOutput());
 minimumMaximumImageCalculatorFilter->Compute();
 
 std::cout << "Maximum: " << minimumMaximumImageCalculatorFilter->GetIndexOfMaximum() << std::endl;
 std::cout << "Minimum: " << minimumMaximumImageCalculatorFilter->GetIndexOfMinimum() << std::endl;
 typedef itk::RescaleIntensityImageFilter< FloatImageType, UnsignedCharImageType > RescaleFilterType;
 typedef itk::ImageFileWriter<UnsignedCharImageType> WriterType;
 {
 RescaleFilterType::Pointer rescaleFilter = RescaleFilterType::New();
 rescaleFilter->SetInput(correlationFilter->GetOutput());
 rescaleFilter->SetOutputMinimum(0);
 rescaleFilter->SetOutputMaximum(255);
 rescaleFilter->Update();
 
 WriterType::Pointer writer = WriterType::New();
 writer->SetInput(rescaleFilter->GetOutput());
 writer->SetFileName("correlation.png");
 writer->Update();
 }
 {
 RescaleFilterType::Pointer rescaleFilter = RescaleFilterType::New();
 rescaleFilter->SetInput(extractFilter->GetOutput());
 rescaleFilter->SetOutputMinimum(0);
 rescaleFilter->SetOutputMaximum(255);
 rescaleFilter->Update();
 
 WriterType::Pointer writer = WriterType::New();
 writer->SetInput(rescaleFilter->GetOutput());
 writer->SetFileName("patch.png");
 writer->Update();
 }
 
 QuickView viewer;
 viewer.AddImage(reader->GetOutput());
 viewer.AddImage(extractFilter->GetOutput());
 viewer.AddImage(correlationFilter->GetOutput());
 viewer.Visualize();
 return EXIT_SUCCESS;

}

</source>

CMakeLists.txt

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

PROJECT(NormalizedCorrelationImageFilter)

include_directories(/home/doriad/ITKWikiExamples/ItkVtkGlue)

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

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

ADD_EXECUTABLE(NormalizedCorrelationImageFilter NormalizedCorrelationImageFilter.cpp /home/doriad/ITKWikiExamples/ItkVtkGlue/QuickView.cxx) TARGET_LINK_LIBRARIES(NormalizedCorrelationImageFilter vtkHybrid ITKBasicFilters ITKCommon ITKIO)

</source>