From 6940dd720ebb6b3d1df8ca203ab696daefe58189 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 20 Mar 2020 12:25:56 +0000 Subject: renamed Documentation folder 20.02 and added .nojekyll file Signed-off-by: Jim Flynn --- 20.02/_tf_lite_mobile_net_ssd-_armnn_8cpp.xhtml | 167 ++++++++++++++++++++++++ 1 file changed, 167 insertions(+) create mode 100644 20.02/_tf_lite_mobile_net_ssd-_armnn_8cpp.xhtml (limited to '20.02/_tf_lite_mobile_net_ssd-_armnn_8cpp.xhtml') diff --git a/20.02/_tf_lite_mobile_net_ssd-_armnn_8cpp.xhtml b/20.02/_tf_lite_mobile_net_ssd-_armnn_8cpp.xhtml new file mode 100644 index 0000000000..f891ae5df9 --- /dev/null +++ b/20.02/_tf_lite_mobile_net_ssd-_armnn_8cpp.xhtml @@ -0,0 +1,167 @@ + + + + + + + + + + + + + +ArmNN: tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp File Reference + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+Functions
+
+
TfLiteMobileNetSsd-Armnn.cpp File Reference
+
+
+
#include "../MobileNetSsdInferenceTest.hpp"
+#include "armnnTfLiteParser/ITfLiteParser.hpp"
+#include <algorithm>
+#include <iterator>
+
+

Go to the source code of this file.

+ + + + +

+Functions

int main (int argc, char *argv[])
 
+

Function Documentation

+ +

◆ main()

+ +
+
+ + + + + + + + + + + + + + + + + + +
int main (int argc,
char * argv[] 
)
+
+ +

Definition at line 14 of file TfLiteMobileNetSsd-Armnn.cpp.

+ +

References armnn::test::InferenceTestMain(), and armnn::test::ValidateDirectory().

+
15 {
16  int retVal = EXIT_FAILURE;
17  try
18  {
19  using DataType = float;
20  using Parser = armnnTfLiteParser::ITfLiteParser;
21  using Model = InferenceModel<Parser, DataType>;
22 
23  armnn::TensorShape inputTensorShape({ 1, 300, 300, 3 });
24 
25  std::vector<const char*> inputLayerNames =
26  {
27  "normalized_input_image_tensor"
28  };
29 
30  std::vector<const char*> outputLayerNames =
31  {
32  "TFLite_Detection_PostProcess",
33  "TFLite_Detection_PostProcess:1",
34  "TFLite_Detection_PostProcess:2",
35  "TFLite_Detection_PostProcess:3"
36  };
37 
38  retVal = InferenceTestMain(argc, argv, { 0 },
39  [&inputTensorShape, inputLayerNames, outputLayerNames]()
40  {
41  return make_unique<MobileNetSsdTestCaseProvider<Model>>(
42  [&]
43  (const InferenceTestOptions& commonOptions,
44  typename Model::CommandLineOptions modelOptions)
45  {
46  if (!ValidateDirectory(modelOptions.m_ModelDir))
47  {
48  return std::unique_ptr<Model>();
49  }
50 
51  typename Model::Params modelParams;
52  modelParams.m_ModelPath =
53  modelOptions.m_ModelDir + "ssd_mobilenet_v1.tflite";
54 
55  std::copy(inputLayerNames.begin(), inputLayerNames.end(),
56  std::back_inserter(modelParams.m_InputBindings));
57 
58  std::copy(outputLayerNames.begin(), outputLayerNames.end(),
59  std::back_inserter(modelParams.m_OutputBindings));
60 
61  modelParams.m_InputShapes = { inputTensorShape };
62  modelParams.m_IsModelBinary = true;
63  modelParams.m_ComputeDevices = modelOptions.GetComputeDevicesAsBackendIds();
64  modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel;
65  modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode;
66 
67  return std::make_unique<Model>(modelParams,
68  commonOptions.m_EnableProfiling,
69  commonOptions.m_DynamicBackendsPath);
70  });
71  });
72  }
73  catch (const std::exception& e)
74  {
75  std::cerr << "WARNING: " << *argv << ": An error has occurred when running "
76  "the classifier inference tests: " << e.what() << std::endl;
77  }
78  return retVal;
79 }
+ + + +
armnn::DataType
DataType
Definition: Types.hpp:32
+
armnn::test::ValidateDirectory
bool ValidateDirectory(std::string &dir)
Definition: InferenceTest.cpp:92
+
armnn::test::InferenceTestMain
int InferenceTestMain(int argc, char *argv[], const std::vector< unsigned int > &defaultTestCaseIds, TConstructTestCaseProvider constructTestCaseProvider)
Definition: InferenceTest.inl:301
+
+
+
+
+
+ + + + -- cgit v1.2.1