aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp
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-28 09:59:44 +0100
commita723ec5d2ac35948efb5dfd0c121a1a89cb977b7 (patch)
treee0a101fbb7ba41cd50e30a1e0ce31529404175ed /tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp
parent7f2c35a82ec11be50b3478bd15207320bbf3bd57 (diff)
downloadarmnn-a723ec5d2ac35948efb5dfd0c121a1a89cb977b7.tar.gz
IVGCVSW-3129 Image pre-processing fix for TFLitev19.05
* 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/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp')
-rw-r--r--tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp5
1 files changed, 3 insertions, 2 deletions
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);
}