patch/Documentation/_tf_lite_mnas_net-_armnn_8cpp_source.xhtml

 //
 // Copyright © 2017 Arm Ltd. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 #include "../InferenceTest.hpp"
 #include "../ImagePreprocessor.hpp"
 #include "armnnTfLiteParser/ITfLiteParser.hpp"

 using namespace armnnTfLiteParser;

 int main(int argc, char* argv[])
 {
     int retVal = EXIT_FAILURE;
     try
     {
         // Coverity fix: The following code may throw an exception of type std::length_error.
         std::vector<ImageSet> imageSet =
         {
             {"Dog.jpg", 209},
             {"Cat.jpg", 283},
             {"shark.jpg", 3},
         };

         armnn::TensorShape inputTensorShape({ 2, 224, 224, 3 });

         using DataType = float;
         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>(
                      argc, argv,
                      "mnasnet_1.3_224.tflite",            // model name
                      true,                                // model is binary
                      "input",                             // input tensor name
                      "output",                            // output tensor name
                      { 0, 1, 2 },                         // test images to test with as above
                      [&imageSet](const char* dataDir, const ModelType &) {
                          return DatabaseType(
                              dataDir,
                              224,
                              224,
                              imageSet,
                              127.5f,
                              {0.5f,0.5f,0.5f});
                      },
                      &inputTensorShape);
     }
     catch (const std::exception& e)
     {
         // Coverity fix: BOOST_LOG_TRIVIAL (typically used to report errors) may throw an
         // exception of type std::length_error.
         // Using stderr instead in this context as there is no point in nesting try-catch blocks here.
         std::cerr << "WARNING: " << *argv << ": An error has occurred when running "
                      "the classifier inference tests: " << e.what() << std::endl;
     }
     return retVal;
 }
InferenceModel
Definition: InferenceModel.hpp:316

-armnnTfLiteParser::ITfLiteParser
Definition: ITfLiteParser.hpp:25

-armnn::TensorShape
Definition: Tensor.hpp:20

-ITfLiteParser.hpp

-armnn::DataType
DataType
Definition: Types.hpp:32

-armnn::test::ClassifierInferenceTestMain
int ClassifierInferenceTestMain(int argc, char *argv[], const char *modelFilename, bool isModelBinary, const char *inputBindingName, const char *outputBindingName, const std::vector< unsigned int > &defaultTestCaseIds, TConstructDatabaseCallable constructDatabase, const armnn::TensorShape *inputTensorShape=nullptr)
Definition: InferenceTest.inl:352

-main
int main(int argc, char *argv[])
Definition: TfLiteMnasNet-Armnn.cpp:11

-armnnTfLiteParser
Definition: ITfLiteParser.hpp:17

-ImagePreprocessor
Definition: ImagePreprocessor.hpp:20

-