Commit 4ecffc47 authored by Cresson Remi's avatar Cresson Remi
Browse files

ENH: fix bug with nodata

parent 32383381
......@@ -199,8 +199,6 @@ TensorflowSampler<TInputImage, TVectorData>
PixelType pix = it.Get();
for (unsigned int band; band < pix.Size(); band++)
{
std::cout << band << std::endl;
if (pix[band] == m_NodataValue)
{
hasBeenSampled = false;
......@@ -217,6 +215,20 @@ TensorflowSampler<TInputImage, TVectorData>
} // Next input
if (hasBeenSampled)
{
// TODO: delete this
IndexType outIndex;
outIndex[0] = 0;
outIndex[1] = count * m_PatchSizes[0][1];
RegionType region(outIndex, m_PatchSizes[0]);
IteratorType it(m_OutputPatchImages[0], region);
for (it.GoToBegin(); !it.IsAtEnd(); ++it)
{
PixelType pix = it.Get();
for (unsigned int band; band < pix.Size(); band++)
if (pix[band] == m_NodataValue)
std::cout << band << std::endl;
}
// Fill label
labelIndex[1] = count;
m_OutputLabelImage->SetPixel(labelIndex, labelPix);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment