From a723ec5d2ac35948efb5dfd0c121a1a89cb977b7 Mon Sep 17 00:00:00 2001 From: FinnWilliamsArm Date: Wed, 22 May 2019 14:50:55 +0100 Subject: IVGCVSW-3129 Image pre-processing fix for TFLite * Resized images for quantized models are now statically cast to uint8 instead of quantized * Removed optional quantization parameters from ImagePreprocessor constructor * Changed mean and scale for TFLite models Signed-off-by: FinnWilliamsArm Change-Id: Id5ffdf77f3614d10c417e769bd8ffc4a4c07308b --- tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp') diff --git a/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp b/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp index 0ba1e5dc00..2084d2d656 100644 --- a/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp +++ b/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp @@ -42,14 +42,12 @@ int main(int argc, char* argv[]) [&imageSet](const char* dataDir, const ModelType & model) { // we need to get the input quantization parameters from // the parsed model - auto inputBinding = model.GetInputBindingInfo(); return DatabaseType( dataDir, 224, 224, imageSet, - inputBinding.second.GetQuantizationScale(), - inputBinding.second.GetQuantizationOffset(), + 1, {{0, 0, 0}}, {{1, 1, 1}}, DatabaseType::DataFormat::NCHW, -- cgit v1.2.1