Commit 6791bc2d authored by Cresson Remi's avatar Cresson Remi
Browse files

ENH: fix bug with nodata

parent e5ddc958
...@@ -230,6 +230,7 @@ TensorflowSampler<TInputImage, TVectorData> ...@@ -230,6 +230,7 @@ TensorflowSampler<TInputImage, TVectorData>
std::cout << band << std::endl; 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