An error occurred while loading the file. Please try again.
-
Cresson Remi authorede3272cad
/*=========================================================================
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");