1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #include "../InferenceTest.hpp"
6 #include "../ImagePreprocessor.hpp"
7 #include "armnnTfLiteParser/ITfLiteParser.hpp"
8
9 using namespace armnnTfLiteParser;
10
main(int argc,char * argv[])11 int main(int argc, char* argv[])
12 {
13 int retVal = EXIT_FAILURE;
14 try
15 {
16 // Coverity fix: The following code may throw an exception of type std::length_error.
17 std::vector<ImageSet> imageSet =
18 {
19 {"Dog.jpg", 209},
20 {"Cat.jpg", 283},
21 {"shark.jpg", 3},
22
23 };
24
25 armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 });
26
27 using DataType = float;
28 using DatabaseType = ImagePreprocessor<DataType>;
29 using ParserType = armnnTfLiteParser::ITfLiteParser;
30 using ModelType = InferenceModel<ParserType, DataType>;
31 // Coverity fix: ClassifierInferenceTestMain() may throw uncaught exceptions.
32 retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType,
33 ParserType>(
34 argc, argv,
35 "resnet_v2_101_299.tflite", // model name
36 true, // model is binary
37 "input", // input tensor name
38 "output", // output tensor name
39 { 0, 1, 2 }, // test images to test with as above
40 [&imageSet](const char* dataDir, const ModelType &) {
41 return DatabaseType(
42 dataDir,
43 299,
44 299,
45 imageSet,
46 127.5f,
47 {0.5f,0.5f,0.5f});
48 },
49 &inputTensorShape);
50 }
51 catch (const std::exception& e)
52 {
53 // Coverity fix: BOOST_LOG_TRIVIAL (typically used to report errors) may throw an
54 // exception of type std::length_error.
55 // Using stderr instead in this context as there is no point in nesting try-catch blocks here.
56 std::cerr << "WARNING: " << *argv << ": An error has occurred when running "
57 "the classifier inference tests: " << e.what() << std::endl;
58 }
59 return retVal;
60 }
61