diff options
Diffstat (limited to 'tests/TfLiteResNetV2-Armnn')
-rw-r--r-- | tests/TfLiteResNetV2-Armnn/TfLiteResNetV2-Armnn.cpp | 5 |
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); } |