From 88b394d85fc65d7d7cbe2e71ac5fb39c8ee1d96d Mon Sep 17 00:00:00 2001
From: Remi Cresson <remi.cresson@irstea.fr>
Date: Sat, 26 Oct 2019 15:06:25 +0200
Subject: [PATCH] ENH: new sample selection/extraction strategy (use nodata in
 selection or extraction)

---
 include/otbTensorflowSampler.hxx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/include/otbTensorflowSampler.hxx b/include/otbTensorflowSampler.hxx
index 1a830c2..8c5d989 100644
--- a/include/otbTensorflowSampler.hxx
+++ b/include/otbTensorflowSampler.hxx
@@ -187,7 +187,7 @@ TensorflowSampler<TInputImage, TVectorData>
           IndexType outIndex;
           outIndex[0] = 0;
           outIndex[1] = count * m_PatchSizes[i][1];
-          RegionType region(outIndex, m_PatchSizes);
+          RegionType region(outIndex, m_PatchSizes[i]);
 
           IteratorType it(m_OutputPatchImages[i], region);
           for (it.GoToBegin(); !it.IsAtEnd(); ++it)
-- 
GitLab