From af8b72d3a97baa0c86583b03321e8a8d3f4bc2aa 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/OnnxMobileNet-Armnn/OnnxMobileNet-Armnn.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'tests/OnnxMobileNet-Armnn/OnnxMobileNet-Armnn.cpp') diff --git a/tests/OnnxMobileNet-Armnn/OnnxMobileNet-Armnn.cpp b/tests/OnnxMobileNet-Armnn/OnnxMobileNet-Armnn.cpp index 79ee49e595..8786feacf9 100644 --- a/tests/OnnxMobileNet-Armnn/OnnxMobileNet-Armnn.cpp +++ b/tests/OnnxMobileNet-Armnn/OnnxMobileNet-Armnn.cpp @@ -40,8 +40,7 @@ int main(int argc, char* argv[]) 224, 224, imageSet, - 1.0, // scale - 0, // offset + 255.0, // scale {{0.485f, 0.456f, 0.406f}}, // mean {{0.229f, 0.224f, 0.225f}}, // stddev DatabaseType::DataFormat::NCHW); // format -- cgit v1.2.1