Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / f060b2e5ad65554cc8b567fb016bbaf57f6aca05 / . / Documentation / _tf_lite_mobile_net_ssd-_armnn_8cpp.html
blob: 835bad04d02c4f3f8ac728094175684d7df5c78d [file] [log] [blame]
Ryan OSheade36e4a2020-03-13 16:26:19 +0000[diff] [blame]1<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
2<html xmlns="http://www.w3.org/1999/xhtml">
3<head>
4<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
5<meta http-equiv="X-UA-Compatible" content="IE=9"/>
6<meta name="generator" content="Doxygen 1.8.13"/>
7<meta name="viewport" content="width=device-width, initial-scale=1"/>
8<title>ArmNN: tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp File Reference</title>
9<link href="tabs.css" rel="stylesheet" type="text/css"/>
10<script type="text/javascript" src="jquery.js"></script>
11<script type="text/javascript" src="dynsections.js"></script>
12<link href="navtree.css" rel="stylesheet" type="text/css"/>
13<script type="text/javascript" src="resize.js"></script>
14<script type="text/javascript" src="navtreedata.js"></script>
15<script type="text/javascript" src="navtree.js"></script>
16<script type="text/javascript">
17 $(document).ready(initResizable);
18</script>
19<link href="search/search.css" rel="stylesheet" type="text/css"/>
20<script type="text/javascript" src="search/searchdata.js"></script>
21<script type="text/javascript" src="search/search.js"></script>
22<link href="doxygen.css" rel="stylesheet" type="text/css" />
23</head>
24<body>
25<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
26<div id="titlearea">
27<table cellspacing="0" cellpadding="0">
28 <tbody>
29 <tr style="height: 56px;">
30 <td id="projectalign" style="padding-left: 0.5em;">
31 <div id="projectname">ArmNN
32 &#160;<span id="projectnumber">NotReleased</span>
33 </div>
34 </td>
35 </tr>
36 </tbody>
37</table>
38</div>
39<!-- end header part -->
40<!-- Generated by Doxygen 1.8.13 -->
41<script type="text/javascript">
42var searchBox = new SearchBox("searchBox", "search",false,'Search');
43</script>
44<script type="text/javascript" src="menudata.js"></script>
45<script type="text/javascript" src="menu.js"></script>
46<script type="text/javascript">
47$(function() {
48 initMenu('',true,false,'search.php','Search');
49 $(document).ready(function() { init_search(); });
50});
51</script>
52<div id="main-nav"></div>
53</div><!-- top -->
54<div id="side-nav" class="ui-resizable side-nav-resizable">
55 <div id="nav-tree">
56 <div id="nav-tree-contents">
57 <div id="nav-sync" class="sync"></div>
58 </div>
59 </div>
60 <div id="splitbar" style="-moz-user-select:none;"
61 class="ui-resizable-handle">
62 </div>
63</div>
64<script type="text/javascript">
65$(document).ready(function(){initNavTree('_tf_lite_mobile_net_ssd-_armnn_8cpp.html','');});
66</script>
67<div id="doc-content">
68<!-- window showing the filter options -->
69<div id="MSearchSelectWindow"
70 onmouseover="return searchBox.OnSearchSelectShow()"
71 onmouseout="return searchBox.OnSearchSelectHide()"
72 onkeydown="return searchBox.OnSearchSelectKey(event)">
73</div>
74
75<!-- iframe showing the search results (closed by default) -->
76<div id="MSearchResultsWindow">
77<iframe src="javascript:void(0)" frameborder="0"
78 name="MSearchResults" id="MSearchResults">
79</iframe>
80</div>
81
82<div class="header">
83 <div class="summary">
84<a href="#func-members">Functions</a> </div>
85 <div class="headertitle">
86<div class="title">TfLiteMobileNetSsd-Armnn.cpp File Reference</div> </div>
87</div><!--header-->
88<div class="contents">
89<div class="textblock"><code>#include &quot;<a class="el" href="_mobile_net_ssd_inference_test_8hpp_source.html">../MobileNetSsdInferenceTest.hpp</a>&quot;</code><br />
90<code>#include &quot;<a class="el" href="_i_tf_lite_parser_8hpp_source.html">armnnTfLiteParser/ITfLiteParser.hpp</a>&quot;</code><br />
91<code>#include &lt;algorithm&gt;</code><br />
92<code>#include &lt;iterator&gt;</code><br />
93</div>
94<p><a href="_tf_lite_mobile_net_ssd-_armnn_8cpp_source.html">Go to the source code of this file.</a></p>
95<table class="memberdecls">
96<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
97Functions</h2></td></tr>
98<tr class="memitem:a0ddf1224851353fc92bfbff6f499fa97"><td class="memItemLeft" align="right" valign="top">int&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="_tf_lite_mobile_net_ssd-_armnn_8cpp.html#a0ddf1224851353fc92bfbff6f499fa97">main</a> (int argc, char *argv[])</td></tr>
99<tr class="separator:a0ddf1224851353fc92bfbff6f499fa97"><td class="memSeparator" colspan="2">&#160;</td></tr>
100</table>
101<h2 class="groupheader">Function Documentation</h2>
102<a id="a0ddf1224851353fc92bfbff6f499fa97"></a>
103<h2 class="memtitle"><span class="permalink"><a href="#a0ddf1224851353fc92bfbff6f499fa97">&#9670;&nbsp;</a></span>main()</h2>
104
105<div class="memitem">
106<div class="memproto">
107 <table class="memname">
108 <tr>
109 <td class="memname">int main </td>
110 <td>(</td>
111 <td class="paramtype">int&#160;</td>
112 <td class="paramname"><em>argc</em>, </td>
113 </tr>
114 <tr>
115 <td class="paramkey"></td>
116 <td></td>
117 <td class="paramtype">char *&#160;</td>
118 <td class="paramname"><em>argv</em>[]&#160;</td>
119 </tr>
120 <tr>
121 <td></td>
122 <td>)</td>
123 <td></td><td></td>
124 </tr>
125 </table>
126</div><div class="memdoc">
127
128<p class="definition">Definition at line <a class="el" href="_tf_lite_mobile_net_ssd-_armnn_8cpp_source.html#l00014">14</a> of file <a class="el" href="_tf_lite_mobile_net_ssd-_armnn_8cpp_source.html">TfLiteMobileNetSsd-Armnn.cpp</a>.</p>
129
130<p class="reference">References <a class="el" href="_inference_test_8inl_source.html#l00301">armnn::test::InferenceTestMain()</a>, and <a class="el" href="_inference_test_8cpp_source.html#l00092">armnn::test::ValidateDirectory()</a>.</p>
131<div class="fragment"><div class="line"><a name="l00015"></a><span class="lineno"> 15</span>&#160;{</div><div class="line"><a name="l00016"></a><span class="lineno"> 16</span>&#160; <span class="keywordtype">int</span> retVal = EXIT_FAILURE;</div><div class="line"><a name="l00017"></a><span class="lineno"> 17</span>&#160; <span class="keywordflow">try</span></div><div class="line"><a name="l00018"></a><span class="lineno"> 18</span>&#160; {</div><div class="line"><a name="l00019"></a><span class="lineno"> 19</span>&#160; <span class="keyword">using</span> <a class="code" href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6">DataType</a> = float;</div><div class="line"><a name="l00020"></a><span class="lineno"> 20</span>&#160; <span class="keyword">using</span> Parser = <a class="code" href="classarmnn_tf_lite_parser_1_1_i_tf_lite_parser.html">armnnTfLiteParser::ITfLiteParser</a>;</div><div class="line"><a name="l00021"></a><span class="lineno"> 21</span>&#160; <span class="keyword">using</span> <a class="code" href="struct_model.html">Model</a> = <a class="code" href="class_inference_model.html">InferenceModel&lt;Parser, DataType&gt;</a>;</div><div class="line"><a name="l00022"></a><span class="lineno"> 22</span>&#160;</div><div class="line"><a name="l00023"></a><span class="lineno"> 23</span>&#160; <a class="code" href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a> inputTensorShape({ 1, 300, 300, 3 });</div><div class="line"><a name="l00024"></a><span class="lineno"> 24</span>&#160;</div><div class="line"><a name="l00025"></a><span class="lineno"> 25</span>&#160; std::vector&lt;const char*&gt; inputLayerNames =</div><div class="line"><a name="l00026"></a><span class="lineno"> 26</span>&#160; {</div><div class="line"><a name="l00027"></a><span class="lineno"> 27</span>&#160; <span class="stringliteral">&quot;normalized_input_image_tensor&quot;</span></div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160; };</div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160;</div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160; std::vector&lt;const char*&gt; outputLayerNames =</div><div class="line"><a name="l00031"></a><span class="lineno"> 31</span>&#160; {</div><div class="line"><a name="l00032"></a><span class="lineno"> 32</span>&#160; <span class="stringliteral">&quot;TFLite_Detection_PostProcess&quot;</span>,</div><div class="line"><a name="l00033"></a><span class="lineno"> 33</span>&#160; <span class="stringliteral">&quot;TFLite_Detection_PostProcess:1&quot;</span>,</div><div class="line"><a name="l00034"></a><span class="lineno"> 34</span>&#160; <span class="stringliteral">&quot;TFLite_Detection_PostProcess:2&quot;</span>,</div><div class="line"><a name="l00035"></a><span class="lineno"> 35</span>&#160; <span class="stringliteral">&quot;TFLite_Detection_PostProcess:3&quot;</span></div><div class="line"><a name="l00036"></a><span class="lineno"> 36</span>&#160; };</div><div class="line"><a name="l00037"></a><span class="lineno"> 37</span>&#160;</div><div class="line"><a name="l00038"></a><span class="lineno"> 38</span>&#160; retVal = <a class="code" href="namespacearmnn_1_1test.html#a7ad751ee33bdcba0656fdb6e98eb17f1">InferenceTestMain</a>(argc, argv, { 0 },</div><div class="line"><a name="l00039"></a><span class="lineno"> 39</span>&#160; [&amp;inputTensorShape, inputLayerNames, outputLayerNames]()</div><div class="line"><a name="l00040"></a><span class="lineno"> 40</span>&#160; {</div><div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160; <span class="keywordflow">return</span> make_unique&lt;MobileNetSsdTestCaseProvider&lt;Model&gt;&gt;(</div><div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160; [&amp;]</div><div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; (<span class="keyword">const</span> <a class="code" href="structarmnn_1_1test_1_1_inference_test_options.html">InferenceTestOptions</a>&amp; commonOptions,</div><div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160; <span class="keyword">typename</span> Model::CommandLineOptions modelOptions)</div><div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160; {</div><div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; <span class="keywordflow">if</span> (!<a class="code" href="namespacearmnn_1_1test.html#a8a678fb3a8864e31aeb77f857ae6fc43">ValidateDirectory</a>(modelOptions.m_ModelDir))</div><div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160; {</div><div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; <span class="keywordflow">return</span> std::unique_ptr&lt;Model&gt;();</div><div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; }</div><div class="line"><a name="l00050"></a><span class="lineno"> 50</span>&#160;</div><div class="line"><a name="l00051"></a><span class="lineno"> 51</span>&#160; <span class="keyword">typename</span> Model::Params modelParams;</div><div class="line"><a name="l00052"></a><span class="lineno"> 52</span>&#160; modelParams.m_ModelPath =</div><div class="line"><a name="l00053"></a><span class="lineno"> 53</span>&#160; modelOptions.m_ModelDir + <span class="stringliteral">&quot;ssd_mobilenet_v1.tflite&quot;</span>;</div><div class="line"><a name="l00054"></a><span class="lineno"> 54</span>&#160;</div><div class="line"><a name="l00055"></a><span class="lineno"> 55</span>&#160; std::copy(inputLayerNames.begin(), inputLayerNames.end(),</div><div class="line"><a name="l00056"></a><span class="lineno"> 56</span>&#160; std::back_inserter(modelParams.m_InputBindings));</div><div class="line"><a name="l00057"></a><span class="lineno"> 57</span>&#160;</div><div class="line"><a name="l00058"></a><span class="lineno"> 58</span>&#160; std::copy(outputLayerNames.begin(), outputLayerNames.end(),</div><div class="line"><a name="l00059"></a><span class="lineno"> 59</span>&#160; std::back_inserter(modelParams.m_OutputBindings));</div><div class="line"><a name="l00060"></a><span class="lineno"> 60</span>&#160;</div><div class="line"><a name="l00061"></a><span class="lineno"> 61</span>&#160; modelParams.m_InputShapes = { inputTensorShape };</div><div class="line"><a name="l00062"></a><span class="lineno"> 62</span>&#160; modelParams.m_IsModelBinary = <span class="keyword">true</span>;</div><div class="line"><a name="l00063"></a><span class="lineno"> 63</span>&#160; modelParams.m_ComputeDevices = modelOptions.GetComputeDevicesAsBackendIds();</div><div class="line"><a name="l00064"></a><span class="lineno"> 64</span>&#160; modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel;</div><div class="line"><a name="l00065"></a><span class="lineno"> 65</span>&#160; modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode;</div><div class="line"><a name="l00066"></a><span class="lineno"> 66</span>&#160;</div><div class="line"><a name="l00067"></a><span class="lineno"> 67</span>&#160; <span class="keywordflow">return</span> std::make_unique&lt;Model&gt;(modelParams,</div><div class="line"><a name="l00068"></a><span class="lineno"> 68</span>&#160; commonOptions.m_EnableProfiling,</div><div class="line"><a name="l00069"></a><span class="lineno"> 69</span>&#160; commonOptions.m_DynamicBackendsPath);</div><div class="line"><a name="l00070"></a><span class="lineno"> 70</span>&#160; });</div><div class="line"><a name="l00071"></a><span class="lineno"> 71</span>&#160; });</div><div class="line"><a name="l00072"></a><span class="lineno"> 72</span>&#160; }</div><div class="line"><a name="l00073"></a><span class="lineno"> 73</span>&#160; <span class="keywordflow">catch</span> (<span class="keyword">const</span> std::exception&amp; e)</div><div class="line"><a name="l00074"></a><span class="lineno"> 74</span>&#160; {</div><div class="line"><a name="l00075"></a><span class="lineno"> 75</span>&#160; std::cerr &lt;&lt; <span class="stringliteral">&quot;WARNING: &quot;</span> &lt;&lt; *argv &lt;&lt; <span class="stringliteral">&quot;: An error has occurred when running &quot;</span></div><div class="line"><a name="l00076"></a><span class="lineno"> 76</span>&#160; <span class="stringliteral">&quot;the classifier inference tests: &quot;</span> &lt;&lt; e.what() &lt;&lt; std::endl;</div><div class="line"><a name="l00077"></a><span class="lineno"> 77</span>&#160; }</div><div class="line"><a name="l00078"></a><span class="lineno"> 78</span>&#160; <span class="keywordflow">return</span> retVal;</div><div class="line"><a name="l00079"></a><span class="lineno"> 79</span>&#160;}</div><div class="ttc" id="class_inference_model_html"><div class="ttname"><a href="class_inference_model.html">InferenceModel</a></div><div class="ttdef"><b>Definition:</b> <a href="_inference_model_8hpp_source.html#l00316">InferenceModel.hpp:316</a></div></div>
132<div class="ttc" id="struct_model_html"><div class="ttname"><a href="struct_model.html">Model</a></div><div class="ttdef"><b>Definition:</b> <a href="_timeline_model_8h_source.html#l00062">TimelineModel.h:62</a></div></div>
133<div class="ttc" id="structarmnn_1_1test_1_1_inference_test_options_html"><div class="ttname"><a href="structarmnn_1_1test_1_1_inference_test_options.html">armnn::test::InferenceTestOptions</a></div><div class="ttdef"><b>Definition:</b> <a href="_inference_test_8hpp_source.html#l00055">InferenceTest.hpp:55</a></div></div>
134<div class="ttc" id="namespacearmnn_html_ad8ed01ff3ff33333d8e19db4d2818bb6"><div class="ttname"><a href="namespacearmnn.html#ad8ed01ff3ff33333d8e19db4d2818bb6">armnn::DataType</a></div><div class="ttdeci">DataType</div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.html#l00032">Types.hpp:32</a></div></div>
135<div class="ttc" id="classarmnn_1_1_tensor_shape_html"><div class="ttname"><a href="classarmnn_1_1_tensor_shape.html">armnn::TensorShape</a></div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.html#l00020">Tensor.hpp:20</a></div></div>
136<div class="ttc" id="namespacearmnn_1_1test_html_a7ad751ee33bdcba0656fdb6e98eb17f1"><div class="ttname"><a href="namespacearmnn_1_1test.html#a7ad751ee33bdcba0656fdb6e98eb17f1">armnn::test::InferenceTestMain</a></div><div class="ttdeci">int InferenceTestMain(int argc, char *argv[], const std::vector&lt; unsigned int &gt; &amp;defaultTestCaseIds, TConstructTestCaseProvider constructTestCaseProvider)</div><div class="ttdef"><b>Definition:</b> <a href="_inference_test_8inl_source.html#l00301">InferenceTest.inl:301</a></div></div>
137<div class="ttc" id="classarmnn_tf_lite_parser_1_1_i_tf_lite_parser_html"><div class="ttname"><a href="classarmnn_tf_lite_parser_1_1_i_tf_lite_parser.html">armnnTfLiteParser::ITfLiteParser</a></div><div class="ttdef"><b>Definition:</b> <a href="_i_tf_lite_parser_8hpp_source.html#l00025">ITfLiteParser.hpp:25</a></div></div>
138<div class="ttc" id="namespacearmnn_1_1test_html_a8a678fb3a8864e31aeb77f857ae6fc43"><div class="ttname"><a href="namespacearmnn_1_1test.html#a8a678fb3a8864e31aeb77f857ae6fc43">armnn::test::ValidateDirectory</a></div><div class="ttdeci">bool ValidateDirectory(std::string &amp;dir)</div><div class="ttdef"><b>Definition:</b> <a href="_inference_test_8cpp_source.html#l00092">InferenceTest.cpp:92</a></div></div>
139</div><!-- fragment -->
140</div>
141</div>
142</div><!-- contents -->
143</div><!-- doc-content -->
144<!-- start footer part -->
145<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
146 <ul>
147 <li class="navelem"><a class="el" href="dir_59425e443f801f1f2fd8bbe4959a3ccf.html">tests</a></li><li class="navelem"><a class="el" href="dir_fdc95ac0b83376e0461da52976cdade9.html">TfLiteMobileNetSsd-Armnn</a></li><li class="navelem"><a class="el" href="_tf_lite_mobile_net_ssd-_armnn_8cpp.html">TfLiteMobileNetSsd-Armnn.cpp</a></li>
148 <li class="footer">Generated on Fri Mar 13 2020 16:07:01 for ArmNN by
149 <a href="http://www.doxygen.org/index.html">
150 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.13 </li>
151 </ul>
152</div>
153</body>
154</html>