ARM-software / ML-examples

Arm Machine Learning tutorials and examples

Home Page:https://developer.arm.com/technologies/machine-learning-on-arm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.