Google Git
Sign in
android / platform / external / armnn / eba482e6 / . / Documentation / _tf_lite_mobile_net_ssd-_armnn_8cpp.html
blob: 835bad04d02c4f3f8ac728094175684d7df5c78d [file] [log] [blame]
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
<html xmlns="http://www.w3.org/1999/xhtml">
<head>
<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
<meta http-equiv="X-UA-Compatible" content="IE=9"/>
<meta name="generator" content="Doxygen 1.8.13"/>
<meta name="viewport" content="width=device-width, initial-scale=1"/>
<title>ArmNN: tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp File Reference</title>
<link href="tabs.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="jquery.js"></script>
<script type="text/javascript" src="dynsections.js"></script>
<link href="navtree.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="resize.js"></script>
<script type="text/javascript" src="navtreedata.js"></script>
<script type="text/javascript" src="navtree.js"></script>
<script type="text/javascript">
$(document).ready(initResizable);
</script>
<link href="search/search.css" rel="stylesheet" type="text/css"/>
<script type="text/javascript" src="search/searchdata.js"></script>
<script type="text/javascript" src="search/search.js"></script>
<link href="doxygen.css" rel="stylesheet" type="text/css" />
</head>
<body>
<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
<div id="titlearea">
<table cellspacing="0" cellpadding="0">
<tbody>
<tr style="height: 56px;">
<td id="projectalign" style="padding-left: 0.5em;">
<div id="projectname">ArmNN
&#160;<span id="projectnumber">NotReleased</span>
</div>
</td>
</tr>
</tbody>
</table>
</div>
<!-- end header part -->
<!-- Generated by Doxygen 1.8.13 -->
<script type="text/javascript">
var searchBox = new SearchBox("searchBox", "search",false,'Search');
</script>
<script type="text/javascript" src="menudata.js"></script>
<script type="text/javascript" src="menu.js"></script>
<script type="text/javascript">
$(function() {
initMenu('',true,false,'search.php','Search');
$(document).ready(function() { init_search(); });
});
</script>
<div id="main-nav"></div>
</div><!-- top -->
<div id="side-nav" class="ui-resizable side-nav-resizable">
<div id="nav-tree">
<div id="nav-tree-contents">
<div id="nav-sync" class="sync"></div>
</div>
</div>
<div id="splitbar" style="-moz-user-select:none;"
class="ui-resizable-handle">
</div>
</div>
<script type="text/javascript">
$(document).ready(function(){initNavTree('_tf_lite_mobile_net_ssd-_armnn_8cpp.html','');});
</script>
<div id="doc-content">
<!-- window showing the filter options -->
<div id="MSearchSelectWindow"
onmouseover="return searchBox.OnSearchSelectShow()"
onmouseout="return searchBox.OnSearchSelectHide()"
onkeydown="return searchBox.OnSearchSelectKey(event)">
</div>
<!-- iframe showing the search results (closed by default) -->
<div id="MSearchResultsWindow">
<iframe src="javascript:void(0)" frameborder="0"
name="MSearchResults" id="MSearchResults">
</iframe>
</div>
<div class="header">
<div class="summary">
<a href="#func-members">Functions</a> </div>
<div class="headertitle">
<div class="title">TfLiteMobileNetSsd-Armnn.cpp File Reference</div> </div>
</div><!--header-->
<div class="contents">
<div class="textblock"><code>#include &quot;<a class="el" href="_mobile_net_ssd_inference_test_8hpp_source.html">../MobileNetSsdInferenceTest.hpp</a>&quot;</code><br />
<code>#include &quot;<a class="el" href="_i_tf_lite_parser_8hpp_source.html">armnnTfLiteParser/ITfLiteParser.hpp</a>&quot;</code><br />
<code>#include &lt;algorithm&gt;</code><br />
<code>#include &lt;iterator&gt;</code><br />
</div>
<p><a href="_tf_lite_mobile_net_ssd-_armnn_8cpp_source.html">Go to the source code of this file.</a></p>
<table class="memberdecls">
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
Functions</h2></td></tr>
<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>
<tr class="separator:a0ddf1224851353fc92bfbff6f499fa97"><td class="memSeparator" colspan="2">&#160;</td></tr>
</table>
<h2 class="groupheader">Function Documentation</h2>
<a id="a0ddf1224851353fc92bfbff6f499fa97"></a>
<h2 class="memtitle"><span class="permalink"><a href="#a0ddf1224851353fc92bfbff6f499fa97">&#9670;&nbsp;</a></span>main()</h2>
<div class="memitem">
<div class="memproto">
<table class="memname">
<tr>
<td class="memname">int main </td>
<td>(</td>
<td class="paramtype">int&#160;</td>
<td class="paramname"><em>argc</em>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype">char *&#160;</td>
<td class="paramname"><em>argv</em>[]&#160;</td>
</tr>
<tr>
<td></td>
<td>)</td>
<td></td><td></td>
</tr>
</table>
</div><div class="memdoc">
<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>
<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>
<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>
<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>
<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>
<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>
<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>
<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>
<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>
<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>
</div><!-- fragment -->
</div>
</div>
</div><!-- contents -->
</div><!-- doc-content -->
<!-- start footer part -->
<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
<ul>
<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>
<li class="footer">Generated on Fri Mar 13 2020 16:07:01 for ArmNN by
<a href="http://www.doxygen.org/index.html">
<img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.13 </li>
</ul>
</div>
</body>
</html>