How to use Batch processing
gangooteli opened this issue · comments
Below snippet in the ML-examples/armnn-mobilenet-quant/mobilenetv1_quant_tflite.cpp code is used to load and preprocess the input.
// Load and preprocess input image
const std::vector<TContainer> inputDataContainers =
{ PrepareImageTensor<uint8_t>(programOptions.imagePath,
inputTensorWidth, inputTensorHeight,
normParams,
inputTensorBatchSize,
inputTensorDataLayout) };
programOptions.imagePath takes a path of single image.
There is an option of batch processing using argument "inputTensorBatchSize" in the "PrepareImageTensor".
I am able to run it successfully for single image.
How to utilise this above code for batch processing with multiple images. Please help.
Thanks in advance.