aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp
diff options
context:
space:
mode:
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);
}