Skip to content
GitLab
Select Git revision
  • master
  • develop default
  • 49-zarr-geoparquet
  • wip_model_output_autonaming
  • docfix-2
  • move_python_packages_usr
  • tensorrt_support
  • test_graph_transform
  • 9-cannot-mix-otbtf-application-and-tensorflow-in-python
  • r4.3.0
  • r4.2.2
  • r4.2.1
  • r4.2.0
  • r4.1.0
  • r4.0.0
  • r3.4.0
  • r3.3.0
  • r3.2.1
  • r3.1
  • r3.0-rc2
  • r3.0-rc1
  • r2.5
  • r2.4
  • r2.3
  • r2.2
  • r2.1
  • r1.8
  • r1.7
  • r1.6
Find file BlameHistoryPermalink
otbTensorflowMultisourceModelBase.hxx 4.92 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 
#ifndef otbTensorflowMultisourceModelBase_txx
12 
#define otbTensorflowMultisourceModelBase_txx
13
14 
#include "otbTensorflowMultisourceModelBase.h"
15
16 
namespace otb
17 
{
18
19 
template <class TInputImage, class TOutputImage>
20 
TensorflowMultisourceModelBase<TInputImage, TOutputImage>
21 
::TensorflowMultisourceModelBase()
22 
 {
23 
  m_Session = nullptr;
24 
  Superclass::SetCoordinateTolerance(itk::NumericTraits<double>::max() );
25 
  Superclass::SetDirectionTolerance(itk::NumericTraits<double>::max() );
26 
 }
27
28 
template <class TInputImage, class TOutputImage>
29 
void
30 
TensorflowMultisourceModelBase<TInputImage, TOutputImage>
31 
::PushBackInputTensorBundle(std::string placeholder, SizeType receptiveField, ImagePointerType image)
32 
 {
33 
  Superclass::PushBackInput(image);
34 
  m_InputReceptiveFields.push_back(receptiveField);
35 
  m_InputPlaceholders.push_back(placeholder);
36 
 }
37
38 
template <class TInputImage, class TOutputImage>
39 
std::stringstream
40 
TensorflowMultisourceModelBase<TInputImage, TOutputImage>
41 
::GenerateDebugReport(DictType & inputs)
42 
 {
43 
  // Create a debug report
44 
  std::stringstream debugReport;
45
46 
  // Describe the output buffered region
47 
  ImagePointerType outputPtr = this->GetOutput();
48 
  const RegionType outputReqRegion = outputPtr->GetRequestedRegion();
49 
  debugReport << "Output image buffered region: " << outputReqRegion << "\n";
50
51 
  // Describe inputs
52 
  for (unsigned int i = 0 ; i < this->GetNumberOfInputs() ; i++)
53 
    {
54 
    const ImagePointerType inputPtr = const_cast<TInputImage*>(this->GetInput(i));
55 
    const RegionType reqRegion = inputPtr->GetRequestedRegion();
56 
    debugReport << "Input #" << i << ":\n";
57 
    debugReport << "Requested region: " << reqRegion << "\n";
58 
    debugReport << "Tensor shape (\"" << inputs[i].first << "\": " << tf::PrintTensorShape(inputs[i].second.shape()) << "\n";
59 
    }
60
61 
  // Show user placeholders
62 
  debugReport << "User placeholders:\n" ;
63 
  for (auto& dict: this->GetUserPlaceholders())
64 
    {
65 
    debugReport << dict.first << " " << tf::PrintTensorInfos(dict.second) << "\n" << std::endl;
66 
    }
67
68 
  return debugReport;
69 
 }
70
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
template <class TInputImage, class TOutputImage> void TensorflowMultisourceModelBase<TInputImage, TOutputImage> ::RunSession(DictType & inputs, TensorListType & outputs) { // Add the user's placeholders for (auto& dict: this->GetUserPlaceholders()) { inputs.push_back(dict); } // Run the TF session here // The session will initialize the outputs // Run the session, evaluating our output tensors from the graph auto status = this->GetSession()->Run(inputs, m_OutputTensors, m_TargetNodesNames, &outputs); if (!status.ok()) { // Create a debug report std::stringstream debugReport = GenerateDebugReport(inputs); // Throw an exception with the report itkExceptionMacro("Can't run the tensorflow session !\n" << "Tensorflow error message:\n" << status.ToString() << "\n" "OTB Filter debug message:\n" << debugReport.str() ); } } template <class TInputImage, class TOutputImage> void TensorflowMultisourceModelBase<TInputImage, TOutputImage> ::GenerateOutputInformation() { // Check that the number of the following is the same // - input placeholders names // - input receptive fields // - input images const unsigned int nbInputs = this->GetNumberOfInputs(); if (nbInputs != m_InputReceptiveFields.size() || nbInputs != m_InputPlaceholders.size()) { itkExceptionMacro("Number of input images is " << nbInputs << " but the number of input patches size is " << m_InputReceptiveFields.size() << " and the number of input tensors names is " << m_InputPlaceholders.size()); } // Check that the number of the following is the same // - output tensors names // - output expression fields if (m_OutputExpressionFields.size() != m_OutputTensors.size()) { itkExceptionMacro("Number of output tensors names is " << m_OutputTensors.size() << " but the number of output fields of expression is " << m_OutputExpressionFields.size()); } ////////////////////////////////////////////////////////////////////////////////////////// // Get tensors information ////////////////////////////////////////////////////////////////////////////////////////// // Get input and output tensors datatypes and shapes tf::GetTensorAttributes(m_Graph, m_InputPlaceholders, m_InputTensorsShapes, m_InputTensorsDataTypes); tf::GetTensorAttributes(m_Graph, m_OutputTensors, m_OutputTensorsShapes, m_OutputTensorsDataTypes); }
141142143144145
} // end namespace otb #endif