From af8b72d3a97baa0c86583b03321e8a8d3f4bc2aa Mon Sep 17 00:00:00 2001
From: FinnWilliamsArm <Finn.Williams@arm.com>
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 <Finn.Williams@arm.com>
Change-Id: Id5ffdf77f3614d10c417e769bd8ffc4a4c07308b
---
 tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp')

diff --git a/tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp b/tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp
index 36fc72cdf4..1e2ffbf568 100644
--- a/tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp
+++ b/tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp
@@ -28,7 +28,6 @@ int main(int argc, char* argv[])
         using DatabaseType = ImagePreprocessor<DataType>;
         using ParserType = armnnTfLiteParser::ITfLiteParser;
         using ModelType = InferenceModel<ParserType, DataType>;
-
         // Coverity fix: ClassifierInferenceTestMain() may throw uncaught exceptions.
         retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType,
                                                           ParserType>(
@@ -43,7 +42,9 @@ int main(int argc, char* argv[])
                              dataDir,
                              299,
                              299,
-                             imageSet);
+                             imageSet,
+                             127.5f,
+                             {0.5f,0.5f,0.5f});
                      },
                      &inputTensorShape);
     }
-- 
cgit v1.2.1