otbTensorflowModelTrain.cxx

/*=========================================================================

     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"
#include "otbWrapperApplicationFactory.h"

// Application engine
#include "otbStandardFilterWatcher.h"
#include "itkFixedArray.h"

// Tensorflow SavedModel
#include "tensorflow/cc/saved_model/loader.h"

// Tensorflow model train
#include "otbTensorflowMultisourceModelTrain.h"
#include "otbTensorflowMultisourceModelValidate.h"

// Tensorflow graph load
#include "otbTensorflowGraphOperations.h"

// Layerstack
#include "otbTensorflowSource.h"

// Metrics
#include "otbConfusionMatrixMeasurements.h"

namespace otb
{

namespace Wrapper
{

class TensorflowModelTrain : public Application
{
public:

  /** Standard class typedefs. */
  typedef TensorflowModelTrain                       Self;
  typedef Application                                Superclass;
  typedef itk::SmartPointer<Self>                    Pointer;
  typedef itk::SmartPointer<const Self>              ConstPointer;

  /** Standard macro */
  itkNewMacro(Self);
  itkTypeMacro(TensorflowModelTrain, Application);

  /** Typedefs for TensorFlow */
  typedef otb::TensorflowMultisourceModelTrain<FloatVectorImageType>    TrainModelFilterType;
  typedef otb::TensorflowMultisourceModelValidate<FloatVectorImageType> ValidateModelFilterType;
  typedef otb::TensorflowSource<FloatVectorImageType>                   TFSource;

  /* Typedefs for evaluation metrics */
  typedef ValidateModelFilterType::ConfMatType                          ConfMatType;
  typedef ValidateModelFilterType::MapOfClassesType                     MapOfClassesType;
  typedef ValidateModelFilterType::LabelValueType                       LabelValueType;
  typedef otb::ConfusionMatrixMeasurements<ConfMatType, LabelValueType> ConfusionMatrixCalculatorType;

  //
  // Store stuff related to one source
  //
  struct ProcessObjectsBundle