otbImageClassifierFromDeepFeatures.cxx 4.12 KiB
/*=========================================================================
     Copyright (c) 2018-2019 IRSTEA
     Copyright (c) 2020-2021 INRAE
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.
=========================================================================*/
#include "itkFixedArray.h"
#include "itkObjectFactory.h"
// Elevation handler
#include "otbWrapperElevationParametersHandler.h"
#include "otbWrapperApplicationFactory.h"
#include "otbWrapperCompositeApplication.h"
// Application engine
#include "otbStandardFilterWatcher.h"
#include "itkFixedArray.h"
// TF (used to get the environment variable for the number of inputs)
#include "otbTensorflowCommon.h"
namespace otb
namespace Wrapper
class ImageClassifierFromDeepFeatures : public CompositeApplication
public:
  /** Standard class typedefs. */
  typedef ImageClassifierFromDeepFeatures              Self;
  typedef Application                         Superclass;
  typedef itk::SmartPointer<Self>             Pointer;
  typedef itk::SmartPointer<const Self>       ConstPointer;
  /** Standard macro */
  itkNewMacro(Self);
  itkTypeMacro(ImageClassifierFromDeepFeatures, otb::Wrapper::CompositeApplication);
private:
  // Add an input source, which includes:
  // -an input image list
  // -an input patchsize (dimensions of samples)
  void AddAnInputImage(int inputNumber = 0)
    inputNumber++;
    // Create keys and descriptions
    std::stringstream ss_key_group, ss_desc_group;
    ss_key_group << "source" << inputNumber;
    ss_desc_group << "Parameters for source " << inputNumber;
    // Populate group
    ShareParameter(ss_key_group.str(), "tfmodel." + ss_key_group.str(), ss_desc_group.str());
  void DoInit()
    SetName("ImageClassifierFromDeepFeatures");