tests/CaffeSqueezeNet1_0-Armnn/CaffeSqueezeNet1_0-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     return armnn::test::ClassifierInferenceTestMain<CaffePreprocessor, armnnCaffeParser::ICaffeParser>(
  12         argc, argv, "squeezenet.caffemodel", true,
  13         "data", "output", { 0 },
  14         [](const char* dataDir) { return CaffePreprocessor(dataDir); });
  15 }