Skip to content

GitLab

Commit 2062149c authored by remi cresson's avatar remi cresson
Browse files

DEBUG: temporary disable streaming for training

parent b5d53231
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
include/otbTensorflowMultisourceModelTrain.hxx
View file @ 2062149c
......@@ -102,7 +102,9 @@ TensorflowMultisourceModelTrain<TInputImage>
{
RegionType nullRegion;
ImageType * inputImage = static_cast<ImageType * >( Superclass::ProcessObject::GetInput(i) );
inputImage->SetRequestedRegion(nullRegion);
// TODO: streaming mode on/off
// inputImage->SetRequestedRegion(nullRegion);
inputImage->SetRequestedRegion(inputImage->GetLargestPossibleRegion());
} // next image
}
......@@ -170,7 +172,8 @@ TensorflowMultisourceModelTrain<TInputImage>
start[0] = 0;
start[1] = randPos * sz_y;
RegionType patchRegion(start, inputPatchSize);
tf::PropagateRequestedRegion<TInputImage>(inputPtr, patchRegion);
// TODO: streaming mode on/off
// tf::PropagateRequestedRegion<TInputImage>(inputPtr, patchRegion);
tf::RecopyImageRegionToTensorWithCast<TInputImage>(inputPtr, patchRegion, inputTensor, elem );
}
......
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