diff --git a/include/otbTensorflowMultisourceModelBase.hxx b/include/otbTensorflowMultisourceModelBase.hxx index b67d92234f4b57f6100e69dad95fe47e8dfea92a..d9d6d7c69db553a8439fcf3a3939a1b910a62b93 100644 --- a/include/otbTensorflowMultisourceModelBase.hxx +++ b/include/otbTensorflowMultisourceModelBase.hxx @@ -194,12 +194,12 @@ TensorflowMultisourceModelBase<TInputImage, TOutputImage>::GenerateOutputInforma // Check that no-data values size is consistent with the inputs // If no value is specified, set a vector of the same size as the inputs - if (m_InputNodataValues.size() == 0 && m_InputHasNodata.size() == 0) + if (m_InputNodataValues.size() == 0 && m_InputUseNodata.size() == 0) { - m_InputHasNodata = BoolListType(nbInputs, false); + m_InputUseNodata = BoolListType(nbInputs, false); m_InputNodataValues = ValueListType(nbInputs, 0.0); } - if (nbInputs != m_InputNodataValues.size() || nbInputs != m_InputNodataValues.size()) + if (nbInputs != m_InputNodataValues.size() || nbInputs != m_InputUseNodata.size()) { itkExceptionMacro("Number of input images is " << nbInputs << " but the number of no-data values is not consistent"); }