aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteMnasNet-Armnn
diff options
context:
space:
mode:
authorFinnWilliamsArm <Finn.Williams@arm.com>2019-05-22 14:50:55 +0100
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-05-27 16:27:36 +0000
commitaf8b72d3a97baa0c86583b03321e8a8d3f4bc2aa (patch)
tree3b4d86842399842b852e593c3bcf131d86d3ae20 /tests/TfLiteMnasNet-Armnn
parent43aec5886449c1b024b740fd6f4500e827bde221 (diff)
downloadarmnn-af8b72d3a97baa0c86583b03321e8a8d3f4bc2aa.tar.gz
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
Diffstat (limited to 'tests/TfLiteMnasNet-Armnn')
-rw-r--r--tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp4
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp b/tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp
index c676cd7355..4cf16d78cc 100644
--- a/tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp
+++ b/tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp
@@ -42,7 +42,9 @@ int main(int argc, char* argv[])
dataDir,
224,
224,
- imageSet);
+ imageSet,
+ 127.5f,
+ {0.5f,0.5f,0.5f});
},
&inputTensorShape);
}