Skip to content

GitLab

Commit 86e331b2 authored by Cresson Remi's avatar Cresson Remi
Browse files

ENH: fix bug with nodata

parent 4ecffc47
Hide whitespace changes
Inline Side-by-side
Showing with 13 additions and 13 deletions
include/otbTensorflowSampler.hxx
View file @ 86e331b2
......@@ -215,19 +215,19 @@ 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;
}
// // 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;
......
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