tests/CaffeVGG-Armnn/CaffeVGG-Armnn.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // See LICENSE file in the project root for full license information.
   4 //
   5 #include "../InferenceTest.hpp"
   6 #include "../CaffePreprocessor.hpp"
   7 #include "armnnCaffeParser/ICaffeParser.hpp"
   8
   9 int main(int argc, char* argv[])
  10 {
  11     armnn::TensorShape inputTensorShape({ 1, 3, 224, 224 });
  12     int retVal = EXIT_FAILURE;
  13     try
  14     {
  15         using DataType = float;
  16         using DatabaseType = CaffePreprocessor;
  17         using ParserType = armnnCaffeParser::ICaffeParser;
  18         using ModelType = InferenceModel<ParserType, DataType>;
  19
  20         // Coverity fix: ClassifierInferenceTestMain() may throw uncaught exceptions.
  21         retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>(
  22                     argc, argv, "VGG_CNN_S.caffemodel", true,
  23                     "input", "prob", { 0 },
  24                     [](const char* dataDir, const ModelType&) {
  25                         return DatabaseType(dataDir, 224, 224);
  26                     }, &inputTensorShape);
  27     }
  28     catch (const std::exception& e)
  29     {
  30         // Coverity fix: BOOST_LOG_TRIVIAL (typically used to report errors) may throw an
  31         // exception of type std::length_error.
  32         // Using stderr instead in this context as there is no point in nesting try-catch blocks here.
  33         std::cerr << "WARNING: CaffeVGG-Armnn: An error has occurred when running "
  34                      "the classifier inference tests: " << e.what() << std::endl;
  35     }
  36     return retVal;
  37 }