From 874f6679f0e9b44746657372eb2a93fa266eb3ba Mon Sep 17 00:00:00 2001
From: remi cresson <remi.cresson@teledetection.fr>
Date: Mon, 30 Jul 2018 11:48:41 +0000
Subject: [PATCH] FIX: last dimension of input tensor can be 1..N
---
include/otbTensorflowMultisourceModelFilter.hxx | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/include/otbTensorflowMultisourceModelFilter.hxx b/include/otbTensorflowMultisourceModelFilter.hxx
index 2422873..1ce1a6e 100644
--- a/include/otbTensorflowMultisourceModelFilter.hxx
+++ b/include/otbTensorflowMultisourceModelFilter.hxx
@@ -261,11 +261,11 @@ TensorflowMultisourceModelFilter<TInputImage, TOutputImage>
// The number of components per pixel is the last dimension of the tensor
int dim_size = protoShape.dim_size();
unsigned int nComponents = 1;
- if (0 < dim_size && dim_size <= 4)
+ if (1 < dim_size && dim_size <= 4)
{
nComponents = protoShape.dim(dim_size-1).size();
}
- else
+ else if (dim_size > 4)
{
itkExceptionMacro("Dim_size=" << dim_size << " currently not supported.");
}
--
GitLab