diff --git a/include/otbTensorflowMultisourceModelFilter.h b/include/otbTensorflowMultisourceModelFilter.h index 37be3aad03e374f6bc010e7c76b623768e63f32f..04a952ba2c98ec80d567972d961116ec0071db83 100644 --- a/include/otbTensorflowMultisourceModelFilter.h +++ b/include/otbTensorflowMultisourceModelFilter.h @@ -21,6 +21,10 @@ #include "otbTensorflowDataTypeBridge.h" #include "otbTensorflowCopyUtils.h" +// Tile hint +#include "itkMetaDataObject.h" +#include "otbMetaDataKey.h" + namespace otb { diff --git a/include/otbTensorflowMultisourceModelFilter.hxx b/include/otbTensorflowMultisourceModelFilter.hxx index e00be6972be307346b08ec8f737c75235a9d6a02..9aaeeda91856eb83e2532cf6eefc3fbc8a18843b 100644 --- a/include/otbTensorflowMultisourceModelFilter.hxx +++ b/include/otbTensorflowMultisourceModelFilter.hxx @@ -288,6 +288,13 @@ TensorflowMultisourceModelFilter<TInputImage, TOutputImage> m_NullPixel.SetSize(outputPtr->GetNumberOfComponentsPerPixel()); m_NullPixel.Fill(0); + ////////////////////////////////////////////////////////////////////////////////////////// + // Set the tiling layout hint in metadata + ////////////////////////////////////////////////////////////////////////////////////////// + + itk::EncapsulateMetaData(outputPtr->GetMetaDataDictionary(), MetaDataKey::TileHintX, m_OutputGridSize[0]); + itk::EncapsulateMetaData(outputPtr->GetMetaDataDictionary(), MetaDataKey::TileHintY, m_OutputGridSize[1]); + } template <class TInputImage, class TOutputImage>