otbTensorflowCopyUtils.cxx

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

  Copyright (c) Remi Cresson (IRSTEA). All rights reserved.


     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 "otbTensorflowCopyUtils.h"

namespace otb {
namespace tf {

//
// Display a TensorShape
//
std::string PrintTensorShape(const tensorflow::TensorShape & shp)
{
  std::stringstream s;
  unsigned int nDims = shp.dims();
  s << "{" << shp.dim_size(0);
  for (unsigned int d = 1 ; d < nDims ; d++)
    s << ", " << shp.dim_size(d);
  s << "}" ;
  return s.str();
}

//
// Display infos about a tensor
//
std::string PrintTensorInfos(const tensorflow::Tensor & tensor)
{
  std::stringstream s;
  s << "Tensor ";
  // Show dims
  s << "shape is " << PrintTensorShape(tensor.shape());
  // Data type
  s << " data type is " << tensor.dtype();
  return s.str();
}

//
// Create a tensor with the good datatype
//
template<class TImage>
tensorflow::Tensor CreateTensor(tensorflow::TensorShape & shape)
{
  tensorflow::DataType ts_dt = GetTensorflowDataType<typename TImage::InternalPixelType>();
  tensorflow::Tensor out_tensor(ts_dt, shape);

  return out_tensor;
}

//
// Populate a tensor with the buffered region of a vector image using std::copy
// Warning: tensor datatype must be consistent with the image value type
//
template<class TImage>
void PopulateTensorFromBufferedVectorImage(const typename TImage::Pointer bufferedimagePtr, tensorflow::Tensor & out_tensor)
{
  size_t n_elem = bufferedimagePtr->GetNumberOfComponentsPerPixel() *
      bufferedimagePtr->GetBufferedRegion().GetNumberOfPixels();
  std::copy_n(bufferedimagePtr->GetBufferPointer(),
      n_elem,
      out_tensor.flat<typename TImage::InternalPixelType>().data());
}

//