Skip to content

GitLab

Commit 5ede6f90 authored by Cresson Remi's avatar Cresson Remi
Browse files

ENH: fix bug with nodata

parent b0c3172a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
include/otbTensorflowSampler.hxx
View file @ 5ede6f90
......@@ -164,6 +164,7 @@ TensorflowSampler<TInputImage, TVectorData>
PixelType labelPix;
labelPix.SetSize(1);
std::cout << "REJECT is " << m_RejectPatchesWithNodata << std::endl;
std::cout << "NODATAVAL is " << m_NodataValue << std::endl;
while (!itVector.IsAtEnd())
{
if (!itVector.Get()->IsRoot() && !itVector.Get()->IsDocument() && !itVector.Get()->IsFolder())
......@@ -198,6 +199,7 @@ TensorflowSampler<TInputImage, TVectorData>
for (it.GoToBegin(); !it.IsAtEnd(); ++it)
{
PixelType pix = it.Get();
std::cout << pix << std::endl;
for (int i; i<pix.Size(); i++)
if (pix[i] == m_NodataValue)
{
......
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