Commit b0c3172a authored by Cresson Remi's avatar Cresson Remi
Browse files

ENH: fix bug with nodata

parent e67ee1d3
......@@ -163,6 +163,7 @@ TensorflowSampler<TInputImage, TVectorData>
labelIndex[0] = 0;
PixelType labelPix;
labelPix.SetSize(1);
std::cout << "REJECT is " << m_RejectPatchesWithNodata << std::endl;
while (!itVector.IsAtEnd())
{
if (!itVector.Get()->IsRoot() && !itVector.Get()->IsDocument() && !itVector.Get()->IsFolder())
......@@ -201,7 +202,6 @@ TensorflowSampler<TInputImage, TVectorData>
if (pix[i] == m_NodataValue)
{
hasBeenSampled = false;
std::cout << "REJECT!" << std::endl;
break;
}
if (!hasBeenSampled)
......
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