Skip to content
GitLab
Select Git revision
  • zonalstats_histogram default
Find file BlameHistoryPermalink
ImageDimensionalityReductionFilter.txx 7.60 KiB
1 
/*=========================================================================
2
3 
  Program:   ORFEO Toolbox
4 
  Language:  C++
5 
  Date:      $Date$
6 
  Version:   $Revision$
7
8
9 
  Copyright (c) Centre National d'Etudes Spatiales. All rights reserved.
10 
  See OTBCopyright.txt for details.
11
12
13 
     This software is distributed WITHOUT ANY WARRANTY; without even
14 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
15 
     PURPOSE.  See the above copyright notices for more information.
16
17 
=========================================================================*/
18 
#ifndef otbImageClassificationFilter_txx
19 
#define otbImageClassificationFilter_txx
20
21 
#include "ImageDimensionalityReductionFilter.h"
22 
#include "itkImageRegionIterator.h"
23 
#include "itkProgressReporter.h"
24
25 
namespace otb
26 
{
27 
/**
28 
 * Constructor
29 
 */
30 
template <class TInputImage, class TOutputImage, class TMaskImage>
31 
ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage>
32 
::ImageDimensionalityReductionFilter()
33 
{
34 
  this->SetNumberOfIndexedInputs(2);
35 
  this->SetNumberOfRequiredInputs(1);
36
37 
  //m_DefaultLabel = itk::NumericTraits<LabelType>::ZeroValue();
38
39 
  this->SetNumberOfRequiredOutputs(2);
40 
  this->SetNthOutput(0,TOutputImage::New());
41 
  this->SetNthOutput(1,ConfidenceImageType::New());
42 
  m_UseConfidenceMap = false;
43 
  m_BatchMode = true;
44 
}
45
46 
template <class TInputImage, class TOutputImage, class TMaskImage>
47 
void
48 
ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage>
49 
::SetInputMask(const MaskImageType * mask)
50 
{
51 
  this->itk::ProcessObject::SetNthInput(1, const_cast<MaskImageType *>(mask));
52 
}
53
54 
template <class TInputImage, class TOutputImage, class TMaskImage>
55 
const typename ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage>
56 
::MaskImageType *
57 
ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage>
58 
::GetInputMask()
59 
{
60 
  if (this->GetNumberOfInputs() < 2)
61 
    {
62 
    return ITK_NULLPTR;
63 
    }
64 
  return static_cast<const MaskImageType *>(this->itk::ProcessObject::GetInput(1));
65 
}
66
67 
template <class TInputImage, class TOutputImage, class TMaskImage>
68 
typename ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage>
69 
::ConfidenceImageType *
70 
ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage>
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
::GetOutputConfidence() { if (this->GetNumberOfOutputs() < 2) { return ITK_NULLPTR; } return static_cast<ConfidenceImageType *>(this->itk::ProcessObject::GetOutput(1)); } template <class TInputImage, class TOutputImage, class TMaskImage> void ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage> ::BeforeThreadedGenerateData() { if (!m_Model) { itkGenericExceptionMacro(<< "No model for classification"); } if(m_BatchMode) { #ifdef _OPENMP // OpenMP will take care of threading this->SetNumberOfThreads(1); #endif } } template <class TInputImage, class TOutputImage, class TMaskImage> void ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage> ::ClassicThreadedGenerateData(const OutputImageRegionType& outputRegionForThread, itk::ThreadIdType threadId) { // Get the input pointers InputImageConstPointerType inputPtr = this->GetInput(); MaskImageConstPointerType inputMaskPtr = this->GetInputMask(); OutputImagePointerType outputPtr = this->GetOutput(); ConfidenceImagePointerType confidencePtr = this->GetOutputConfidence(); // Progress reporting itk::ProgressReporter progress(this, threadId, outputRegionForThread.GetNumberOfPixels()); // Define iterators typedef itk::ImageRegionConstIterator<InputImageType> InputIteratorType; typedef itk::ImageRegionConstIterator<MaskImageType> MaskIteratorType; typedef itk::ImageRegionIterator<OutputImageType> OutputIteratorType; typedef itk::ImageRegionIterator<ConfidenceImageType> ConfidenceMapIteratorType; InputIteratorType inIt(inputPtr, outputRegionForThread); OutputIteratorType outIt(outputPtr, outputRegionForThread); // Walk the part of the image for (inIt.GoToBegin(), outIt.GoToBegin(); !inIt.IsAtEnd() && !outIt.IsAtEnd(); ++inIt, ++outIt) { // Classifify outIt.Set(m_Model->Predict(inIt.Get())); progress.CompletedPixel(); } } template <class TInputImage, class TOutputImage, class TMaskImage> void ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage>::GenerateOutputInformation() { Superclass::GenerateOutputInformation(); this->GetOutput()->SetNumberOfComponentsPerPixel( m_Model->GetDimension() ); std::cout << m_Model->GetDimension() << std::endl; }
141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210
template <class TInputImage, class TOutputImage, class TMaskImage> void ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage> ::BatchThreadedGenerateData(const OutputImageRegionType& outputRegionForThread, itk::ThreadIdType threadId) { // Get the input pointers InputImageConstPointerType inputPtr = this->GetInput(); MaskImageConstPointerType inputMaskPtr = this->GetInputMask(); OutputImagePointerType outputPtr = this->GetOutput(); ConfidenceImagePointerType confidencePtr = this->GetOutputConfidence(); // Progress reporting itk::ProgressReporter progress(this, threadId, outputRegionForThread.GetNumberOfPixels()); // Define iterators typedef itk::ImageRegionConstIterator<InputImageType> InputIteratorType; typedef itk::ImageRegionConstIterator<MaskImageType> MaskIteratorType; typedef itk::ImageRegionIterator<OutputImageType> OutputIteratorType; typedef itk::ImageRegionIterator<ConfidenceImageType> ConfidenceMapIteratorType; InputIteratorType inIt(inputPtr, outputRegionForThread); OutputIteratorType outIt(outputPtr, outputRegionForThread); typedef typename ModelType::InputSampleType InputSampleType; typedef typename ModelType::InputListSampleType InputListSampleType; typedef typename ModelType::TargetValueType TargetValueType; typedef typename ModelType::TargetListSampleType TargetListSampleType; typename InputListSampleType::Pointer samples = InputListSampleType::New(); unsigned int num_features = inputPtr->GetNumberOfComponentsPerPixel(); samples->SetMeasurementVectorSize(num_features); InputSampleType sample(num_features); // Fill the samples for (inIt.GoToBegin(); !inIt.IsAtEnd(); ++inIt) { typename InputImageType::PixelType pix = inIt.Get(); for(size_t feat=0; feat<num_features; ++feat) { sample[feat]=pix[feat]; } samples->PushBack(sample); } //Make the batch prediction typename TargetListSampleType::Pointer labels; // This call is threadsafe labels = m_Model->PredictBatch(samples); // Set the output values typename TargetListSampleType::ConstIterator labIt = labels->Begin(); for (outIt.GoToBegin(); !outIt.IsAtEnd(); ++outIt) { itk::VariableLengthVector<TargetValueType> labelValue; labelValue = labIt.GetMeasurementVector(); ++labIt; outIt.Set(labelValue); progress.CompletedPixel(); } } template <class TInputImage, class TOutputImage, class TMaskImage> void
211212213214215216217218219220221222223224225226227228229230231232233234235236
ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage> ::ThreadedGenerateData(const OutputImageRegionType& outputRegionForThread, itk::ThreadIdType threadId) { if(m_BatchMode) { this->BatchThreadedGenerateData(outputRegionForThread, threadId); } else { this->ClassicThreadedGenerateData(outputRegionForThread, threadId); } } /** * PrintSelf Method */ template <class TInputImage, class TOutputImage, class TMaskImage> void ImageDimensionalityReductionFilter<TInputImage, TOutputImage, TMaskImage> ::PrintSelf(std::ostream& os, itk::Indent indent) const { Superclass::PrintSelf(os, indent); } } // End namespace otb #endif