tests/CaffeSqueezeNet1_0-Armnn/CaffeSqueezeNet1_0-Armnn.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5 #include "../InferenceTest.hpp"
   6 #include "../CaffePreprocessor.hpp"
   7 #include "armnnCaffeParser/ICaffeParser.hpp"
   8
   9 int main(int argc, char* argv[])
  10 {
  11     using DataType = float;
  12     using DatabaseType = CaffePreprocessor;
  13     using ParserType = armnnCaffeParser::ICaffeParser;
  14     using ModelType = InferenceModel<ParserType, DataType>;
  15
  16     return armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>(
  17         argc, argv, "squeezenet.caffemodel", true,
  18         "input", "prob", { 0 },
  19         [](const char* dataDir, const ModelType &) { return CaffePreprocessor(dataDir); });
  20 }