diff options
| author | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-05-22 14:50:55 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-05-27 16:27:36 +0000 |
| commit | af8b72d3a97baa0c86583b03321e8a8d3f4bc2aa (patch) | |
| tree | 3b4d86842399842b852e593c3bcf131d86d3ae20 /tests/TfLiteVGG16Quantized-Armnn | |
| parent | 43aec5886449c1b024b740fd6f4500e827bde221 (diff) | |
| download | armnn-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/TfLiteVGG16Quantized-Armnn')
| -rw-r--r-- | tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp | 4 |
1 files changed, 1 insertions, 3 deletions
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, |