aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteInceptionV4Quantized-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/TfLiteInceptionV4Quantized-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/TfLiteInceptionV4Quantized-Armnn')
-rw-r--r--tests/TfLiteInceptionV4Quantized-Armnn/TfLiteInceptionV4Quantized-Armnn.cpp4
1 files changed, 1 insertions, 3 deletions
diff --git a/tests/TfLiteInceptionV4Quantized-Armnn/TfLiteInceptionV4Quantized-Armnn.cpp b/tests/TfLiteInceptionV4Quantized-Armnn/TfLiteInceptionV4Quantized-Armnn.cpp
index bcb9db8a62..b0af830ba7 100644
--- a/tests/TfLiteInceptionV4Quantized-Armnn/TfLiteInceptionV4Quantized-Armnn.cpp
+++ b/tests/TfLiteInceptionV4Quantized-Armnn/TfLiteInceptionV4Quantized-Armnn.cpp
@@ -40,14 +40,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,
299,
299,
imageSet,
- inputBinding.second.GetQuantizationScale(),
- inputBinding.second.GetQuantizationOffset());
+ 1);
},
&inputTensorShape);
}