Skip to content
GitLab
Select Git revision
  • master default
Find file BlameHistoryPermalink
otbClassificationMapMosaic.cxx 5.34 KiB
1 
/*=========================================================================
2
3 
  Copyright (c) Remi Cresson (IRSTEA). All rights reserved.
4
5
6 
     This software is distributed WITHOUT ANY WARRANTY; without even
7 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
8 
     PURPOSE.  See the above copyright notices for more information.
9
10 
=========================================================================*/
11 
#include "itkFixedArray.h"
12 
#include "itkObjectFactory.h"
13 
#include "otbWrapperApplicationFactory.h"
14
15 
// Application engine
16 
#include "otbStandardFilterWatcher.h"
17 
#include "itkFixedArray.h"
18
19 
// Filter
20 
#include "otbClassificationMosaicArgmaxFilter.h"
21
22 
namespace otb
23 
{
24
25 
namespace Wrapper
26 
{
27
28 
class ClassificationMapMosaic : public Application
29 
{
30 
public:
31 
  /** Standard class typedefs. */
32 
  typedef ClassificationMapMosaic                   Self;
33 
  typedef Application                         Superclass;
34 
  typedef itk::SmartPointer<Self>             Pointer;
35 
  typedef itk::SmartPointer<const Self>       ConstPointer;
36
37 
  /** Standard macro */
38 
  itkNewMacro(Self);
39 
  itkTypeMacro(ClassificationMapMosaic, Application);
40
41 
  /** Filters */
42 
  typedef otb::ClassificationMosaicArgmaxFilter<FloatVectorImageType, FloatVectorImageType, double> ArgmaxFilterType;
43
44 
  /** Interpolators typedefs */
45 
  typedef itk::LinearInterpolateImageFunction<FloatVectorImageType, double>            LinearInterpolationType;
46 
  typedef itk::NearestNeighborInterpolateImageFunction<FloatVectorImageType, double>   NearestNeighborInterpolationType;
47 
  typedef otb::BCOInterpolateImageFunction<FloatVectorImageType>                       BCOInterpolationType;
48
49 
  void DoUpdateParameters()
50 
  {
51 
  }
52
53 
  void DoInit()
54 
  {
55
56 
    // Documentation
57 
    SetName("ClassificationMapMosaic");
58 
    SetDescription("Mosaic multiple classification maps");
59 
    SetDocLongDescription("The application perform the mosaicing of multiple classification maps ");
60
61 
    SetDocAuthors("Remi Cresson");
62
63 
    AddDocTag(Tags::Learning);
64
65 
    // Input
66 
    AddParameter(ParameterType_InputImageList, "il", "Input classification maps");
67
68 
    // Mode
69 
    AddParameter(ParameterType_Choice, "mode", "Fusion mode");
70 
    AddChoice("mode.argmax", "Argmax of the averages values");
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
// Interpolators AddParameter(ParameterType_Choice, "interpolator", "Interpolation"); SetParameterDescription ("interpolator", "This group of parameters allows to define how the input image will be interpolated during resampling."); MandatoryOff ("interpolator"); AddChoice ("interpolator.linear", "Linear interpolation"); SetParameterDescription("interpolator.linear", "Linear interpolation leads to average image quality but is quite fast"); AddChoice ("interpolator.nn", "Nearest Neighbor interpolation"); SetParameterDescription("interpolator.nn", "Nearest neighbor interpolation leads to poor image quality, but it is fast."); AddChoice ("interpolator.bco", "Bicubic interpolation"); SetParameterDescription("interpolator.bco", "Bicubic interpolation leads to great image quality, but it is slow."); AddParameter(ParameterType_Radius, "interpolator.bco.radius", "Radius for bicubic interpolation"); SetParameterDescription ("interpolator.bco.radius", "Size of the bicubic interpolation filter. If the target pixel size " "is higher than the input pixel size, increasing this parameter will reduce aliasing artefacts."); SetDefaultParameterInt ("interpolator.bco.radius", 2); // Output AddParameter(ParameterType_OutputImage, "out", "output classification map"); SetDefaultOutputPixelType("out", ImagePixelType_uint8); } template <class TMosaicFilter> void SelectInterpolator(typename TMosaicFilter::Pointer filter) { if ( GetParameterString("interpolator").compare("linear") == 0 ) { LinearInterpolationType::Pointer interpolator = LinearInterpolationType::New(); filter->SetInterpolator(interpolator); } else if ( GetParameterString("interpolator").compare("nn") == 0 ) { NearestNeighborInterpolationType::Pointer interpolator = NearestNeighborInterpolationType::New(); filter->SetInterpolator(interpolator); } else if ( GetParameterString("interpolator").compare("bco") == 0 ) { BCOInterpolationType::Pointer interpolator = BCOInterpolationType::New(); interpolator->SetRadius(GetParameterInt("interpolator.bco.radius") ); filter->SetInterpolator(interpolator); } else { otbAppLogFATAL("Unknown interpolator: " << GetParameterString("interpolator")) } } void DoExecute() { // Get the input image list FloatVectorImageListType::Pointer inputArray = this->GetParameterImageList("il"); if (GetParameterAsString("mode") == "argmax") { otbAppLogINFO("Fusion mode is argmax"); m_ArgmaxMosaicFilter = ArgmaxFilterType::New(); for (unsigned int i = 0 ; i < inputArray->Size() ; i++) { m_ArgmaxMosaicFilter->PushBackInput(inputArray->GetNthElement(i) ); } SelectInterpolator<ArgmaxFilterType>(m_ArgmaxMosaicFilter); } SetParameterOutputImage("out", m_ArgmaxMosaicFilter->GetOutput()); }
141142143144145146147148149150
private: ArgmaxFilterType::Pointer m_ArgmaxMosaicFilter; }; // end of class } // end namespace wrapper } // end namespace otb OTB_APPLICATION_EXPORT( otb::Wrapper::ClassificationMapMosaic )