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> ...@@ -199,8 +199,6 @@ TensorflowSampler<TInputImage, TVectorData>
PixelType pix = it.Get(); PixelType pix = it.Get();
for (unsigned int band; band < pix.Size(); band++) for (unsigned int band; band < pix.Size(); band++)
{ {
std::cout << band << std::endl;
if (pix[band] == m_NodataValue) if (pix[band] == m_NodataValue)
{ {
hasBeenSampled = false; hasBeenSampled = false;
...@@ -217,6 +215,20 @@ TensorflowSampler<TInputImage, TVectorData> ...@@ -217,6 +215,20 @@ TensorflowSampler<TInputImage, TVectorData>
} // Next input } // Next input
if (hasBeenSampled) 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 // Fill label
labelIndex[1] = count; labelIndex[1] = count;
m_OutputLabelImage->SetPixel(labelIndex, labelPix); 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