Skip to content
GitLab
Select Git revision
Find file BlameHistoryPermalink
otbTensorflowModelServe.cxx 14.05 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 
// Tensorflow stuff
20 
#include "tensorflow/core/public/session.h"
21 
#include "tensorflow/core/platform/env.h"
22
23 
// Tensorflow model filter
24 
#include "otbTensorflowMultisourceModelFilter.h"
25
26 
// Tensorflow graph load
27 
#include "otbTensorflowGraphOperations.h"
28
29 
// Layerstack
30 
#include "otbTensorflowSource.h"
31
32 
// Streaming
33 
#include "otbTensorflowStreamerFilter.h"
34
35 
namespace otb
36 
{
37
38 
namespace Wrapper
39 
{
40
41 
class TensorflowModelServe : public Application
42 
{
43 
public:
44 
  /** Standard class typedefs. */
45 
  typedef TensorflowModelServe                       Self;
46 
  typedef Application                                Superclass;
47 
  typedef itk::SmartPointer<Self>                    Pointer;
48 
  typedef itk::SmartPointer<const Self>              ConstPointer;
49
50 
  /** Standard macro */
51 
  itkNewMacro(Self);
52 
  itkTypeMacro(TensorflowModelServe, Application);
53
54 
  /** Typedefs for tensorflow */
55 
  typedef otb::TensorflowMultisourceModelFilter<FloatVectorImageType, FloatVectorImageType> TFModelFilterType;
56 
  typedef otb::TensorflowSource<FloatVectorImageType> InputImageSource;
57
58 
  /** Typedef for streaming */
59 
  typedef otb::ImageRegionSquareTileSplitter<FloatVectorImageType::ImageDimension> TileSplitterType;
60 
  typedef otb::TensorflowStreamerFilter<FloatVectorImageType, FloatVectorImageType> StreamingFilterType;
61
62 
  /** Typedefs for images */
63 
  typedef FloatVectorImageType::SizeType SizeType;
64
65 
  void DoUpdateParameters()
66 
  {
67 
  }
68
69 
  //
70 
  // Store stuff related to one source
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
// struct ProcessObjectsBundle { InputImageSource m_ImageSource; SizeType m_PatchSize; std::string m_Placeholder; // Parameters keys std::string m_KeyIn; // Key of input image list std::string m_KeyPszX; // Key for samples sizes X std::string m_KeyPszY; // Key for samples sizes Y std::string m_KeyPHName; // Key for placeholder name in the tensorflow model }; // // Add an input source, which includes: // -an input image list // -an input patchsize (dimensions of samples) // void AddAnInputImage() { // Number of source unsigned int inputNumber = m_Bundles.size() + 1; // Create keys and descriptions std::stringstream ss_key_group, ss_desc_group, ss_key_in, ss_desc_in, ss_key_dims_x, ss_desc_dims_x, ss_key_dims_y, ss_desc_dims_y, ss_key_ph, ss_desc_ph; // Parameter group key/description ss_key_group << "source" << inputNumber; ss_desc_group << "Parameters for source #" << inputNumber; // Parameter group keys ss_key_in << ss_key_group.str() << ".il"; ss_key_dims_x << ss_key_group.str() << ".rfieldx"; ss_key_dims_y << ss_key_group.str() << ".rfieldy"; ss_key_ph << ss_key_group.str() << ".placeholder"; // Parameter group descriptions ss_desc_in << "Input image (or list to stack) for source #" << inputNumber; ss_desc_dims_x << "Input receptive field (width) for source #" << inputNumber; ss_desc_dims_y << "Input receptive field (height) for source #" << inputNumber; ss_desc_ph << "Name of the input placeholder for source #" << inputNumber; // Populate group AddParameter(ParameterType_Group, ss_key_group.str(), ss_desc_group.str()); AddParameter(ParameterType_InputImageList, ss_key_in.str(), ss_desc_in.str() ); AddParameter(ParameterType_Int, ss_key_dims_x.str(), ss_desc_dims_x.str()); SetMinimumParameterIntValue (ss_key_dims_x.str(), 1); AddParameter(ParameterType_Int, ss_key_dims_y.str(), ss_desc_dims_y.str()); SetMinimumParameterIntValue (ss_key_dims_y.str(), 1); AddParameter(ParameterType_String, ss_key_ph.str(), ss_desc_ph.str()); // Add a new bundle ProcessObjectsBundle bundle; bundle.m_KeyIn = ss_key_in.str(); bundle.m_KeyPszX = ss_key_dims_x.str(); bundle.m_KeyPszY = ss_key_dims_y.str(); bundle.m_KeyPHName = ss_key_ph.str(); m_Bundles.push_back(bundle); } void DoInit() {