diff options
Diffstat (limited to 'tests/TfInceptionV3-Armnn/TfInceptionV3-Armnn.cpp')
-rw-r--r-- | tests/TfInceptionV3-Armnn/TfInceptionV3-Armnn.cpp | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/tests/TfInceptionV3-Armnn/TfInceptionV3-Armnn.cpp b/tests/TfInceptionV3-Armnn/TfInceptionV3-Armnn.cpp index 441b07c9c9..09e70018d3 100644 --- a/tests/TfInceptionV3-Armnn/TfInceptionV3-Armnn.cpp +++ b/tests/TfInceptionV3-Armnn/TfInceptionV3-Armnn.cpp @@ -3,7 +3,7 @@ // See LICENSE file in the project root for full license information. // #include "../InferenceTest.hpp" -#include "../MobileNetDatabase.hpp" +#include "../ImagePreprocessor.hpp" #include "armnnTfParser/ITfParser.hpp" int main(int argc, char* argv[]) @@ -21,11 +21,18 @@ int main(int argc, char* argv[]) armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 }); + using DataType = float; + using DatabaseType = ImagePreprocessor<float>; + using ParserType = armnnTfParser::ITfParser; + using ModelType = InferenceModel<ParserType, DataType>; + // Coverity fix: InferenceTestMain() may throw uncaught exceptions. - retVal = armnn::test::ClassifierInferenceTestMain<MobileNetDatabase, armnnTfParser::ITfParser>( + retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>( argc, argv, "inception_v3_2016_08_28_frozen_transformed.pb", true, "input", "InceptionV3/Predictions/Reshape_1", { 0, 1, 2, }, - [&imageSet](const char* dataDir) { return MobileNetDatabase(dataDir, 299, 299, imageSet); }, + [&imageSet](const char* dataDir, const ModelType&) { + return DatabaseType(dataDir, 299, 299, imageSet); + }, &inputTensorShape); } catch (const std::exception& e) |