tests/TfInceptionV3-Armnn/TfInceptionV3-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 "../MobileNetDatabase.hpp"
   7 #include "armnnTfParser/ITfParser.hpp"
   8
   9 int main(int argc, char* argv[])
  10 {
  11     std::vector<ImageSet> imageSet =
  12     {
  13         { "Dog.jpg", 208 },
  14         { "Cat.jpg", 283 },
  15         { "shark.jpg", 3 },
  16     };
  17     armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 });
  18     return armnn::test::ClassifierInferenceTestMain<MobileNetDatabase, armnnTfParser::ITfParser>(
  19         argc, argv, "inception_v3_2016_08_28_frozen_transformed.pb", true,
  20         "input", "InceptionV3/Predictions/Reshape_1", { 0, 1, 2, },
  21         [&imageSet](const char* dataDir) { return MobileNetDatabase(dataDir, 299, 299, imageSet); },
  22         &inputTensorShape);
  23 }