blob: b1b4f3e4f741cbee9f7d2c838d995c042567ad3d [file] [log] [blame]
Ryan OSheade36e4a2020-03-13 16:26:19 +00001<!-- Copyright (c) 2020 ARM Limited. -->
2<!-- -->
3<!-- SPDX-License-Identifier: MIT -->
4<!-- -->
5<!-- HTML header for doxygen 1.8.13-->
6<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
7<html xmlns="http://www.w3.org/1999/xhtml">
8<head>
9<meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/>
10<meta http-equiv="X-UA-Compatible" content="IE=9"/>
11<meta name="generator" content="Doxygen 1.8.13"/>
12<meta name="robots" content="NOINDEX, NOFOLLOW" />
13<meta name="viewport" content="width=device-width, initial-scale=1"/>
14<title>ArmNN: tests/ImageTensorGenerator/ImageTensorGenerator.hpp File Reference</title>
15<link href="tabs.css" rel="stylesheet" type="text/css"/>
16<script type="text/javascript" src="jquery.js"></script>
17<script type="text/javascript" src="dynsections.js"></script>
18<link href="navtree.css" rel="stylesheet" type="text/css"/>
19<script type="text/javascript" src="resize.js"></script>
20<script type="text/javascript" src="navtreedata.js"></script>
21<script type="text/javascript" src="navtree.js"></script>
22<script type="text/javascript">
23 $(document).ready(initResizable);
24</script>
25<link href="search/search.css" rel="stylesheet" type="text/css"/>
26<script type="text/javascript" src="search/searchdata.js"></script>
27<script type="text/javascript" src="search/search.js"></script>
28<script type="text/x-mathjax-config">
29 MathJax.Hub.Config({
30 extensions: ["tex2jax.js"],
31 jax: ["input/TeX","output/HTML-CSS"],
32});
33</script><script type="text/javascript" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script>
34<link href="doxygen.css" rel="stylesheet" type="text/css" />
35<link href="stylesheet.css" rel="stylesheet" type="text/css"/>
36</head>
37<body>
38<div id="top"><!-- do not remove this div, it is closed by doxygen! -->
39<div id="titlearea">
40<table cellspacing="0" cellpadding="0">
41 <tbody>
42 <tr style="height: 56px;">
43 <img alt="ArmNN" src="Arm_NN_horizontal_blue.png" style="max-width: 10rem; margin-top: .5rem; margin-left 10px"/>
44 <td style="padding-left: 0.5em;">
45 <div id="projectname">
46 &#160;<span id="projectnumber">20.02</span>
47 </div>
48 </td>
49 </tr>
50 </tbody>
51</table>
52</div>
53<!-- end header part -->
54<!-- Generated by Doxygen 1.8.13 -->
55<script type="text/javascript">
56var searchBox = new SearchBox("searchBox", "search",false,'Search');
57</script>
58<script type="text/javascript" src="menudata.js"></script>
59<script type="text/javascript" src="menu.js"></script>
60<script type="text/javascript">
61$(function() {
62 initMenu('',true,false,'search.php','Search');
63 $(document).ready(function() { init_search(); });
64});
65</script>
66<div id="main-nav"></div>
67</div><!-- top -->
68<div id="side-nav" class="ui-resizable side-nav-resizable">
69 <div id="nav-tree">
70 <div id="nav-tree-contents">
71 <div id="nav-sync" class="sync"></div>
72 </div>
73 </div>
74 <div id="splitbar" style="-moz-user-select:none;"
75 class="ui-resizable-handle">
76 </div>
77</div>
78<script type="text/javascript">
79$(document).ready(function(){initNavTree('_image_tensor_generator_8hpp.xhtml','');});
80</script>
81<div id="doc-content">
82<!-- window showing the filter options -->
83<div id="MSearchSelectWindow"
84 onmouseover="return searchBox.OnSearchSelectShow()"
85 onmouseout="return searchBox.OnSearchSelectHide()"
86 onkeydown="return searchBox.OnSearchSelectKey(event)">
87</div>
88
89<!-- iframe showing the search results (closed by default) -->
90<div id="MSearchResultsWindow">
91<iframe src="javascript:void(0)" frameborder="0"
92 name="MSearchResults" id="MSearchResults">
93</iframe>
94</div>
95
96<div class="header">
97 <div class="summary">
98<a href="#nested-classes">Classes</a> &#124;
99<a href="#enum-members">Enumerations</a> &#124;
100<a href="#func-members">Functions</a> </div>
101 <div class="headertitle">
102<div class="title">ImageTensorGenerator.hpp File Reference</div> </div>
103</div><!--header-->
104<div class="contents">
105<div class="textblock"><code>#include &quot;<a class="el" href="_inference_test_image_8hpp_source.xhtml">../InferenceTestImage.hpp</a>&quot;</code><br />
106<code>#include &lt;<a class="el" href="_types_utils_8hpp_source.xhtml">armnn/TypesUtils.hpp</a>&gt;</code><br />
107<code>#include &lt;<a class="el" href="_permute_8hpp_source.xhtml">armnnUtils/Permute.hpp</a>&gt;</code><br />
108<code>#include &lt;algorithm&gt;</code><br />
109<code>#include &lt;fstream&gt;</code><br />
110<code>#include &lt;iterator&gt;</code><br />
111<code>#include &lt;string&gt;</code><br />
112</div>
113<p><a href="_image_tensor_generator_8hpp_source.xhtml">Go to the source code of this file.</a></p>
114<table class="memberdecls">
115<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="nested-classes"></a>
116Classes</h2></td></tr>
117<tr class="memitem:"><td class="memItemLeft" align="right" valign="top">struct &#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a></td></tr>
118<tr class="separator:"><td class="memSeparator" colspan="2">&#160;</td></tr>
119</table><table class="memberdecls">
120<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="enum-members"></a>
121Enumerations</h2></td></tr>
122<tr class="memitem:a7599848540a5f43e977119c054b3563e"><td class="memItemLeft" align="right" valign="top">enum &#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563e">SupportedFrontend</a> { <a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da">Caffe</a> = 0,
123<a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3">TensorFlow</a> = 1,
124<a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae">TFLite</a> = 2
125 }</td></tr>
126<tr class="separator:a7599848540a5f43e977119c054b3563e"><td class="memSeparator" colspan="2">&#160;</td></tr>
127</table><table class="memberdecls">
128<tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a>
129Functions</h2></td></tr>
130<tr class="memitem:a572494250ece8d94b68ea490d0bf46e4"><td class="memItemLeft" align="right" valign="top"><a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="_image_tensor_generator_8hpp.xhtml#a572494250ece8d94b68ea490d0bf46e4">GetNormalizationParameters</a> (const <a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563e">SupportedFrontend</a> &amp;modelFormat, const <a class="el" href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6">armnn::DataType</a> &amp;outputType)</td></tr>
131<tr class="memdesc:a572494250ece8d94b68ea490d0bf46e4"><td class="mdescLeft">&#160;</td><td class="mdescRight">Get normalization parameters. <a href="#a572494250ece8d94b68ea490d0bf46e4">More...</a><br /></td></tr>
132<tr class="separator:a572494250ece8d94b68ea490d0bf46e4"><td class="memSeparator" colspan="2">&#160;</td></tr>
133<tr class="memitem:af39456ac4838301c6fc5b44d6a57479c"><td class="memTemplParams" colspan="2">template&lt;typename ElemType &gt; </td></tr>
134<tr class="memitem:af39456ac4838301c6fc5b44d6a57479c"><td class="memTemplItemLeft" align="right" valign="top">std::vector&lt; ElemType &gt;&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="_image_tensor_generator_8hpp.xhtml#af39456ac4838301c6fc5b44d6a57479c">PrepareImageTensor</a> (const std::string &amp;imagePath, unsigned int newWidth, unsigned int newHeight, const <a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> &amp;normParams, unsigned int batchSize=1, const <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> &amp;outputLayout=<a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a>)</td></tr>
135<tr class="memdesc:af39456ac4838301c6fc5b44d6a57479c"><td class="mdescLeft">&#160;</td><td class="mdescRight">Prepare raw image tensor data by loading the image from imagePath and preprocessing it. <a href="#af39456ac4838301c6fc5b44d6a57479c">More...</a><br /></td></tr>
136<tr class="separator:af39456ac4838301c6fc5b44d6a57479c"><td class="memSeparator" colspan="2">&#160;</td></tr>
137<tr class="memitem:affacfe95f011fce3b897b2745880f82a"><td class="memTemplParams" colspan="2">template&lt;&gt; </td></tr>
138<tr class="memitem:affacfe95f011fce3b897b2745880f82a"><td class="memTemplItemLeft" align="right" valign="top">std::vector&lt; float &gt;&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="_image_tensor_generator_8hpp.xhtml#affacfe95f011fce3b897b2745880f82a">PrepareImageTensor&lt; float &gt;</a> (const std::string &amp;imagePath, unsigned int newWidth, unsigned int newHeight, const <a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> &amp;normParams, unsigned int batchSize, const <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> &amp;outputLayout)</td></tr>
139<tr class="separator:affacfe95f011fce3b897b2745880f82a"><td class="memSeparator" colspan="2">&#160;</td></tr>
140<tr class="memitem:ac1e8e68c74e5d1c8fd19a91a68de3cb7"><td class="memTemplParams" colspan="2">template&lt;&gt; </td></tr>
141<tr class="memitem:ac1e8e68c74e5d1c8fd19a91a68de3cb7"><td class="memTemplItemLeft" align="right" valign="top">std::vector&lt; int &gt;&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="_image_tensor_generator_8hpp.xhtml#ac1e8e68c74e5d1c8fd19a91a68de3cb7">PrepareImageTensor&lt; int &gt;</a> (const std::string &amp;imagePath, unsigned int newWidth, unsigned int newHeight, const <a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> &amp;normParams, unsigned int batchSize, const <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> &amp;outputLayout)</td></tr>
142<tr class="separator:ac1e8e68c74e5d1c8fd19a91a68de3cb7"><td class="memSeparator" colspan="2">&#160;</td></tr>
143<tr class="memitem:ac372cd2b257f33f29999217e1d2d8a37"><td class="memTemplParams" colspan="2">template&lt;&gt; </td></tr>
144<tr class="memitem:ac372cd2b257f33f29999217e1d2d8a37"><td class="memTemplItemLeft" align="right" valign="top">std::vector&lt; uint8_t &gt;&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="_image_tensor_generator_8hpp.xhtml#ac372cd2b257f33f29999217e1d2d8a37">PrepareImageTensor&lt; uint8_t &gt;</a> (const std::string &amp;imagePath, unsigned int newWidth, unsigned int newHeight, const <a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> &amp;normParams, unsigned int batchSize, const <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> &amp;outputLayout)</td></tr>
145<tr class="separator:ac372cd2b257f33f29999217e1d2d8a37"><td class="memSeparator" colspan="2">&#160;</td></tr>
146<tr class="memitem:a292c4c49fc943bfcb7ee1ce1435d4690"><td class="memTemplParams" colspan="2">template&lt;typename ElemType &gt; </td></tr>
147<tr class="memitem:a292c4c49fc943bfcb7ee1ce1435d4690"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="_image_tensor_generator_8hpp.xhtml#a292c4c49fc943bfcb7ee1ce1435d4690">WriteImageTensorImpl</a> (const std::vector&lt; ElemType &gt; &amp;imageData, std::ofstream &amp;imageTensorFile)</td></tr>
148<tr class="memdesc:a292c4c49fc943bfcb7ee1ce1435d4690"><td class="mdescLeft">&#160;</td><td class="mdescRight">Write image tensor to ofstream. <a href="#a292c4c49fc943bfcb7ee1ce1435d4690">More...</a><br /></td></tr>
149<tr class="separator:a292c4c49fc943bfcb7ee1ce1435d4690"><td class="memSeparator" colspan="2">&#160;</td></tr>
150<tr class="memitem:a2120283536d91fbdee0d19fff3156994"><td class="memTemplParams" colspan="2">template&lt;&gt; </td></tr>
151<tr class="memitem:a2120283536d91fbdee0d19fff3156994"><td class="memTemplItemLeft" align="right" valign="top">void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="_image_tensor_generator_8hpp.xhtml#a2120283536d91fbdee0d19fff3156994">WriteImageTensorImpl&lt; uint8_t &gt;</a> (const std::vector&lt; uint8_t &gt; &amp;imageData, std::ofstream &amp;imageTensorFile)</td></tr>
152<tr class="separator:a2120283536d91fbdee0d19fff3156994"><td class="memSeparator" colspan="2">&#160;</td></tr>
153</table>
154<h2 class="groupheader">Enumeration Type Documentation</h2>
155<a id="a7599848540a5f43e977119c054b3563e"></a>
156<h2 class="memtitle"><span class="permalink"><a href="#a7599848540a5f43e977119c054b3563e">&#9670;&nbsp;</a></span>SupportedFrontend</h2>
157
158<div class="memitem">
159<div class="memproto">
160<table class="mlabels">
161 <tr>
162 <td class="mlabels-left">
163 <table class="memname">
164 <tr>
165 <td class="memname">enum <a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563e">SupportedFrontend</a></td>
166 </tr>
167 </table>
168 </td>
169 <td class="mlabels-right">
170<span class="mlabels"><span class="mlabel">strong</span></span> </td>
171 </tr>
172</table>
173</div><div class="memdoc">
174<table class="fieldtable">
175<tr><th colspan="2">Enumerator</th></tr><tr><td class="fieldname"><a id="a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da"></a>Caffe&#160;</td><td class="fielddoc"></td></tr>
176<tr><td class="fieldname"><a id="a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3"></a>TensorFlow&#160;</td><td class="fielddoc"></td></tr>
177<tr><td class="fieldname"><a id="a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae"></a>TFLite&#160;</td><td class="fielddoc"></td></tr>
178</table>
179
180<p class="definition">Definition at line <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00025">25</a> of file <a class="el" href="_image_tensor_generator_8hpp_source.xhtml">ImageTensorGenerator.hpp</a>.</p>
181<div class="fragment"><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; <a class="code" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da">Caffe</a> = 0,</div><div class="line"><a name="l00028"></a><span class="lineno"> 28</span>&#160; <a class="code" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3">TensorFlow</a> = 1,</div><div class="line"><a name="l00029"></a><span class="lineno"> 29</span>&#160; <a class="code" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae">TFLite</a> = 2,</div><div class="line"><a name="l00030"></a><span class="lineno"> 30</span>&#160;};</div><div class="ttc" id="_image_tensor_generator_8hpp_xhtml_a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da"><div class="ttname"><a href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da">SupportedFrontend::Caffe</a></div></div>
182<div class="ttc" id="_image_tensor_generator_8hpp_xhtml_a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae"><div class="ttname"><a href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae">SupportedFrontend::TFLite</a></div></div>
183<div class="ttc" id="_image_tensor_generator_8hpp_xhtml_a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3"><div class="ttname"><a href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3">SupportedFrontend::TensorFlow</a></div></div>
184</div><!-- fragment -->
185</div>
186</div>
187<h2 class="groupheader">Function Documentation</h2>
188<a id="a572494250ece8d94b68ea490d0bf46e4"></a>
189<h2 class="memtitle"><span class="permalink"><a href="#a572494250ece8d94b68ea490d0bf46e4">&#9670;&nbsp;</a></span>GetNormalizationParameters()</h2>
190
191<div class="memitem">
192<div class="memproto">
193 <table class="memname">
194 <tr>
195 <td class="memname"><a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> GetNormalizationParameters </td>
196 <td>(</td>
197 <td class="paramtype">const <a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563e">SupportedFrontend</a> &amp;&#160;</td>
198 <td class="paramname"><em>modelFormat</em>, </td>
199 </tr>
200 <tr>
201 <td class="paramkey"></td>
202 <td></td>
203 <td class="paramtype">const <a class="el" href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6">armnn::DataType</a> &amp;&#160;</td>
204 <td class="paramname"><em>outputType</em>&#160;</td>
205 </tr>
206 <tr>
207 <td></td>
208 <td>)</td>
209 <td></td><td></td>
210 </tr>
211 </table>
212</div><div class="memdoc">
213
214<p>Get normalization parameters. </p>
215<p>Note that different flavours of models and different model data types have different normalization methods. This tool currently only supports Caffe, TF and TFLite models</p>
216<dl class="params"><dt>Parameters</dt><dd>
217 <table class="params">
218 <tr><td class="paramdir">[in]</td><td class="paramname">modelFormat</td><td>One of the supported frontends </td></tr>
219 <tr><td class="paramdir">[in]</td><td class="paramname">outputType</td><td>Output type of the image tensor, also the type of the intended model </td></tr>
220 </table>
221 </dd>
222</dl>
223
224<p class="definition">Definition at line <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00039">39</a> of file <a class="el" href="_image_tensor_generator_8hpp_source.xhtml">ImageTensorGenerator.hpp</a>.</p>
225
226<p class="reference">References <a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da">Caffe</a>, <a class="el" href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204">armnn::Float32</a>, <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00021">NormalizationParameters::mean</a>, <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::NHWC</a>, <a class="el" href="_image_tensor_generator_8hpp.xhtml#af39456ac4838301c6fc5b44d6a57479c">PrepareImageTensor()</a>, <a class="el" href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6a0a3f57c876f5a230244c38e1453a8a6e">armnn::QAsymmU8</a>, <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00020">NormalizationParameters::scale</a>, <a class="el" href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6accedffbc6e5308e33d3843e8bdc0dad7">armnn::Signed32</a>, <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00022">NormalizationParameters::stddev</a>, <a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3">TensorFlow</a>, and <a class="el" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae">TFLite</a>.</p>
227
228<p class="reference">Referenced by <a class="el" href="_image_tensor_generator_8cpp_source.xhtml#l00248">main()</a>.</p>
229<div class="fragment"><div class="line"><a name="l00041"></a><span class="lineno"> 41</span>&#160;{</div><div class="line"><a name="l00042"></a><span class="lineno"> 42</span>&#160; <a class="code" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> normParams;</div><div class="line"><a name="l00043"></a><span class="lineno"> 43</span>&#160; <span class="comment">// Explicitly set default parameters</span></div><div class="line"><a name="l00044"></a><span class="lineno"> 44</span>&#160; normParams.<a class="code" href="struct_normalization_parameters.xhtml#a1d28dec57cce925ad92342891bd71e7c">scale</a> = 1.0;</div><div class="line"><a name="l00045"></a><span class="lineno"> 45</span>&#160; normParams.<a class="code" href="struct_normalization_parameters.xhtml#a6e5ac6f9d3272e6670522daaacc04442">mean</a> = { 0.0, 0.0, 0.0 };</div><div class="line"><a name="l00046"></a><span class="lineno"> 46</span>&#160; normParams.<a class="code" href="struct_normalization_parameters.xhtml#a8fdb461f0b4417f4ebff10b3413a948d">stddev</a> = { 1.0, 1.0, 1.0 };</div><div class="line"><a name="l00047"></a><span class="lineno"> 47</span>&#160; <span class="keywordflow">switch</span> (modelFormat)</div><div class="line"><a name="l00048"></a><span class="lineno"> 48</span>&#160; {</div><div class="line"><a name="l00049"></a><span class="lineno"> 49</span>&#160; <span class="keywordflow">case</span> <a class="code" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da">SupportedFrontend::Caffe</a>:</div><div class="line"><a name="l00050"></a><span class="lineno"> 50</span>&#160; <span class="keywordflow">break</span>;</div><div class="line"><a name="l00051"></a><span class="lineno"> 51</span>&#160; <span class="keywordflow">case</span> <a class="code" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3">SupportedFrontend::TensorFlow</a>:</div><div class="line"><a name="l00052"></a><span class="lineno"> 52</span>&#160; <span class="keywordflow">case</span> <a class="code" href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae">SupportedFrontend::TFLite</a>:</div><div class="line"><a name="l00053"></a><span class="lineno"> 53</span>&#160; <span class="keywordflow">default</span>:</div><div class="line"><a name="l00054"></a><span class="lineno"> 54</span>&#160; <span class="keywordflow">switch</span> (outputType)</div><div class="line"><a name="l00055"></a><span class="lineno"> 55</span>&#160; {</div><div class="line"><a name="l00056"></a><span class="lineno"> 56</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204">armnn::DataType::Float32</a>:</div><div class="line"><a name="l00057"></a><span class="lineno"> 57</span>&#160; normParams.<a class="code" href="struct_normalization_parameters.xhtml#a1d28dec57cce925ad92342891bd71e7c">scale</a> = 127.5;</div><div class="line"><a name="l00058"></a><span class="lineno"> 58</span>&#160; normParams.<a class="code" href="struct_normalization_parameters.xhtml#a6e5ac6f9d3272e6670522daaacc04442">mean</a> = { 1.0, 1.0, 1.0 };</div><div class="line"><a name="l00059"></a><span class="lineno"> 59</span>&#160; <span class="keywordflow">break</span>;</div><div class="line"><a name="l00060"></a><span class="lineno"> 60</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6accedffbc6e5308e33d3843e8bdc0dad7">armnn::DataType::Signed32</a>:</div><div class="line"><a name="l00061"></a><span class="lineno"> 61</span>&#160; normParams.<a class="code" href="struct_normalization_parameters.xhtml#a6e5ac6f9d3272e6670522daaacc04442">mean</a> = { 128.0, 128.0, 128.0 };</div><div class="line"><a name="l00062"></a><span class="lineno"> 62</span>&#160; <span class="keywordflow">break</span>;</div><div class="line"><a name="l00063"></a><span class="lineno"> 63</span>&#160; <span class="keywordflow">case</span> <a class="code" href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6a0a3f57c876f5a230244c38e1453a8a6e">armnn::DataType::QAsymmU8</a>:</div><div class="line"><a name="l00064"></a><span class="lineno"> 64</span>&#160; <span class="keywordflow">default</span>:</div><div class="line"><a name="l00065"></a><span class="lineno"> 65</span>&#160; <span class="keywordflow">break</span>;</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">break</span>;</div><div class="line"><a name="l00068"></a><span class="lineno"> 68</span>&#160; }</div><div class="line"><a name="l00069"></a><span class="lineno"> 69</span>&#160; <span class="keywordflow">return</span> normParams;</div><div class="line"><a name="l00070"></a><span class="lineno"> 70</span>&#160;}</div><div class="ttc" id="_image_tensor_generator_8hpp_xhtml_a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da"><div class="ttname"><a href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea4df6064eb76f848def300474b7aef0da">SupportedFrontend::Caffe</a></div></div>
230<div class="ttc" id="_image_tensor_generator_8hpp_xhtml_a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae"><div class="ttname"><a href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea35120628a2097980f88fd49246124cae">SupportedFrontend::TFLite</a></div></div>
231<div class="ttc" id="namespacearmnn_xhtml_ad8ed01ff3ff33333d8e19db4d2818bb6accedffbc6e5308e33d3843e8bdc0dad7"><div class="ttname"><a href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6accedffbc6e5308e33d3843e8bdc0dad7">armnn::DataType::Signed32</a></div></div>
232<div class="ttc" id="_image_tensor_generator_8hpp_xhtml_a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3"><div class="ttname"><a href="_image_tensor_generator_8hpp.xhtml#a7599848540a5f43e977119c054b3563ea074dd699710da0ec1eb45f13b31788e3">SupportedFrontend::TensorFlow</a></div></div>
233<div class="ttc" id="struct_normalization_parameters_xhtml_a8fdb461f0b4417f4ebff10b3413a948d"><div class="ttname"><a href="struct_normalization_parameters.xhtml#a8fdb461f0b4417f4ebff10b3413a948d">NormalizationParameters::stddev</a></div><div class="ttdeci">std::array&lt; float, 3 &gt; stddev</div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00022">ImageTensorGenerator.hpp:22</a></div></div>
234<div class="ttc" id="struct_normalization_parameters_xhtml"><div class="ttname"><a href="struct_normalization_parameters.xhtml">NormalizationParameters</a></div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00018">ImageTensorGenerator.hpp:18</a></div></div>
235<div class="ttc" id="namespacearmnn_xhtml_ad8ed01ff3ff33333d8e19db4d2818bb6a0a3f57c876f5a230244c38e1453a8a6e"><div class="ttname"><a href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6a0a3f57c876f5a230244c38e1453a8a6e">armnn::DataType::QAsymmU8</a></div></div>
236<div class="ttc" id="struct_normalization_parameters_xhtml_a6e5ac6f9d3272e6670522daaacc04442"><div class="ttname"><a href="struct_normalization_parameters.xhtml#a6e5ac6f9d3272e6670522daaacc04442">NormalizationParameters::mean</a></div><div class="ttdeci">std::array&lt; float, 3 &gt; mean</div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00021">ImageTensorGenerator.hpp:21</a></div></div>
237<div class="ttc" id="namespacearmnn_xhtml_ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204"><div class="ttname"><a href="namespacearmnn.xhtml#ad8ed01ff3ff33333d8e19db4d2818bb6a166495adc0d0f53bee6baecc577f5204">armnn::DataType::Float32</a></div></div>
238<div class="ttc" id="struct_normalization_parameters_xhtml_a1d28dec57cce925ad92342891bd71e7c"><div class="ttname"><a href="struct_normalization_parameters.xhtml#a1d28dec57cce925ad92342891bd71e7c">NormalizationParameters::scale</a></div><div class="ttdeci">float scale</div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00020">ImageTensorGenerator.hpp:20</a></div></div>
239</div><!-- fragment -->
240</div>
241</div>
242<a id="af39456ac4838301c6fc5b44d6a57479c"></a>
243<h2 class="memtitle"><span class="permalink"><a href="#af39456ac4838301c6fc5b44d6a57479c">&#9670;&nbsp;</a></span>PrepareImageTensor()</h2>
244
245<div class="memitem">
246<div class="memproto">
247 <table class="memname">
248 <tr>
249 <td class="memname">std::vector&lt;ElemType&gt; PrepareImageTensor </td>
250 <td>(</td>
251 <td class="paramtype">const std::string &amp;&#160;</td>
252 <td class="paramname"><em>imagePath</em>, </td>
253 </tr>
254 <tr>
255 <td class="paramkey"></td>
256 <td></td>
257 <td class="paramtype">unsigned int&#160;</td>
258 <td class="paramname"><em>newWidth</em>, </td>
259 </tr>
260 <tr>
261 <td class="paramkey"></td>
262 <td></td>
263 <td class="paramtype">unsigned int&#160;</td>
264 <td class="paramname"><em>newHeight</em>, </td>
265 </tr>
266 <tr>
267 <td class="paramkey"></td>
268 <td></td>
269 <td class="paramtype">const <a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> &amp;&#160;</td>
270 <td class="paramname"><em>normParams</em>, </td>
271 </tr>
272 <tr>
273 <td class="paramkey"></td>
274 <td></td>
275 <td class="paramtype">unsigned int&#160;</td>
276 <td class="paramname"><em>batchSize</em> = <code>1</code>, </td>
277 </tr>
278 <tr>
279 <td class="paramkey"></td>
280 <td></td>
281 <td class="paramtype">const <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> &amp;&#160;</td>
282 <td class="paramname"><em>outputLayout</em> = <code><a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0ad066db54b89b0912e7e7c6da51e2da51">armnn::DataLayout::NHWC</a></code>&#160;</td>
283 </tr>
284 <tr>
285 <td></td>
286 <td>)</td>
287 <td></td><td></td>
288 </tr>
289 </table>
290</div><div class="memdoc">
291
292<p>Prepare raw image tensor data by loading the image from imagePath and preprocessing it. </p>
293<dl class="params"><dt>Parameters</dt><dd>
294 <table class="params">
295 <tr><td class="paramdir">[in]</td><td class="paramname">imagePath</td><td>Path to the image file </td></tr>
296 <tr><td class="paramdir">[in]</td><td class="paramname">newWidth</td><td>The new width of the output image tensor </td></tr>
297 <tr><td class="paramdir">[in]</td><td class="paramname">newHeight</td><td>The new height of the output image tensor </td></tr>
298 <tr><td class="paramdir">[in]</td><td class="paramname">normParams</td><td>Normalization parameters for the normalization of the image </td></tr>
299 <tr><td class="paramdir">[in]</td><td class="paramname">batchSize</td><td>Batch size </td></tr>
300 <tr><td class="paramdir">[in]</td><td class="paramname">outputLayout</td><td>Data layout of the output image tensor </td></tr>
301 </table>
302 </dd>
303</dl>
304
305<p class="reference">Referenced by <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00039">GetNormalizationParameters()</a>.</p>
306
307</div>
308</div>
309<a id="affacfe95f011fce3b897b2745880f82a"></a>
310<h2 class="memtitle"><span class="permalink"><a href="#affacfe95f011fce3b897b2745880f82a">&#9670;&nbsp;</a></span>PrepareImageTensor< float >()</h2>
311
312<div class="memitem">
313<div class="memproto">
314 <table class="memname">
315 <tr>
316 <td class="memname">std::vector&lt;float&gt; <a class="el" href="_image_tensor_generator_8hpp.xhtml#af39456ac4838301c6fc5b44d6a57479c">PrepareImageTensor</a>&lt; float &gt; </td>
317 <td>(</td>
318 <td class="paramtype">const std::string &amp;&#160;</td>
319 <td class="paramname"><em>imagePath</em>, </td>
320 </tr>
321 <tr>
322 <td class="paramkey"></td>
323 <td></td>
324 <td class="paramtype">unsigned int&#160;</td>
325 <td class="paramname"><em>newWidth</em>, </td>
326 </tr>
327 <tr>
328 <td class="paramkey"></td>
329 <td></td>
330 <td class="paramtype">unsigned int&#160;</td>
331 <td class="paramname"><em>newHeight</em>, </td>
332 </tr>
333 <tr>
334 <td class="paramkey"></td>
335 <td></td>
336 <td class="paramtype">const <a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> &amp;&#160;</td>
337 <td class="paramname"><em>normParams</em>, </td>
338 </tr>
339 <tr>
340 <td class="paramkey"></td>
341 <td></td>
342 <td class="paramtype">unsigned int&#160;</td>
343 <td class="paramname"><em>batchSize</em>, </td>
344 </tr>
345 <tr>
346 <td class="paramkey"></td>
347 <td></td>
348 <td class="paramtype">const <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> &amp;&#160;</td>
349 <td class="paramname"><em>outputLayout</em>&#160;</td>
350 </tr>
351 <tr>
352 <td></td>
353 <td>)</td>
354 <td></td><td></td>
355 </tr>
356 </table>
357</div><div class="memdoc">
358
359<p class="definition">Definition at line <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00091">91</a> of file <a class="el" href="_image_tensor_generator_8hpp_source.xhtml">ImageTensorGenerator.hpp</a>.</p>
360
361<p class="reference">References <a class="el" href="class_inference_test_image.xhtml#ae7a403f69a7717c1eaae1d74b7bb7bcea773b27aa8d21604182ba90d029ad2e13">InferenceTestImage::BilinearAndNormalized</a>, <a class="el" href="_exceptions_8hpp_source.xhtml#l00192">CHECK_LOCATION</a>, <a class="el" href="_inference_test_image_8hpp_source.xhtml#l00075">InferenceTestImage::GetWidth()</a>, <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00021">NormalizationParameters::mean</a>, <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f">armnn::NCHW</a>, <a class="el" href="_parser_helper_8cpp_source.xhtml#l00016">armnnUtils::NHWCToArmNN</a>, <a class="el" href="_permute_8cpp_source.xhtml#l00121">armnnUtils::Permute()</a>, <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00020">NormalizationParameters::scale</a>, and <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00022">NormalizationParameters::stddev</a>.</p>
362
363<p class="reference">Referenced by <a class="el" href="_image_tensor_generator_8cpp_source.xhtml#l00248">main()</a>, <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00128">PrepareImageTensor&lt; int &gt;()</a>, and <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00148">PrepareImageTensor&lt; uint8_t &gt;()</a>.</p>
364<div class="fragment"><div class="line"><a name="l00097"></a><span class="lineno"> 97</span>&#160;{</div><div class="line"><a name="l00098"></a><span class="lineno"> 98</span>&#160; <span class="comment">// Generate image tensor</span></div><div class="line"><a name="l00099"></a><span class="lineno"> 99</span>&#160; std::vector&lt;float&gt; imageData;</div><div class="line"><a name="l00100"></a><span class="lineno"> 100</span>&#160; <a class="code" href="class_inference_test_image.xhtml">InferenceTestImage</a> testImage(imagePath.c_str());</div><div class="line"><a name="l00101"></a><span class="lineno"> 101</span>&#160; <span class="keywordflow">if</span> (newWidth == 0)</div><div class="line"><a name="l00102"></a><span class="lineno"> 102</span>&#160; {</div><div class="line"><a name="l00103"></a><span class="lineno"> 103</span>&#160; newWidth = testImage.<a class="code" href="class_inference_test_image.xhtml#a4ec823e49ce46950b3d4cee1efff050b">GetWidth</a>();</div><div class="line"><a name="l00104"></a><span class="lineno"> 104</span>&#160; }</div><div class="line"><a name="l00105"></a><span class="lineno"> 105</span>&#160; <span class="keywordflow">if</span> (newHeight == 0)</div><div class="line"><a name="l00106"></a><span class="lineno"> 106</span>&#160; {</div><div class="line"><a name="l00107"></a><span class="lineno"> 107</span>&#160; newHeight = testImage.GetHeight();</div><div class="line"><a name="l00108"></a><span class="lineno"> 108</span>&#160; }</div><div class="line"><a name="l00109"></a><span class="lineno"> 109</span>&#160; <span class="comment">// Resize the image to new width and height or keep at original dimensions if the new width and height are specified</span></div><div class="line"><a name="l00110"></a><span class="lineno"> 110</span>&#160; <span class="comment">// as 0 Centre/Normalise the image.</span></div><div class="line"><a name="l00111"></a><span class="lineno"> 111</span>&#160; imageData = testImage.Resize(newWidth, newHeight, <a class="code" href="_exceptions_8hpp.xhtml#aa3be76aec4ce713822a5ea1ecbb7bc61">CHECK_LOCATION</a>(),</div><div class="line"><a name="l00112"></a><span class="lineno"> 112</span>&#160; <a class="code" href="class_inference_test_image.xhtml#ae7a403f69a7717c1eaae1d74b7bb7bcea773b27aa8d21604182ba90d029ad2e13">InferenceTestImage::ResizingMethods::BilinearAndNormalized</a>, normParams.<a class="code" href="struct_normalization_parameters.xhtml#a6e5ac6f9d3272e6670522daaacc04442">mean</a>,</div><div class="line"><a name="l00113"></a><span class="lineno"> 113</span>&#160; normParams.<a class="code" href="struct_normalization_parameters.xhtml#a8fdb461f0b4417f4ebff10b3413a948d">stddev</a>, normParams.<a class="code" href="struct_normalization_parameters.xhtml#a1d28dec57cce925ad92342891bd71e7c">scale</a>);</div><div class="line"><a name="l00114"></a><span class="lineno"> 114</span>&#160; <span class="keywordflow">if</span> (outputLayout == <a class="code" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f">armnn::DataLayout::NCHW</a>)</div><div class="line"><a name="l00115"></a><span class="lineno"> 115</span>&#160; {</div><div class="line"><a name="l00116"></a><span class="lineno"> 116</span>&#160; <span class="comment">// Convert to NCHW format</span></div><div class="line"><a name="l00117"></a><span class="lineno"> 117</span>&#160; <span class="keyword">const</span> <a class="code" href="classarmnn_1_1_permutation_vector.xhtml">armnn::PermutationVector</a> <a class="code" href="namespacearmnn_utils.xhtml#a12124184ac6aec018beb98b9715330c7">NHWCToArmNN</a> = { 0, 2, 3, 1 };</div><div class="line"><a name="l00118"></a><span class="lineno"> 118</span>&#160; <a class="code" href="classarmnn_1_1_tensor_shape.xhtml">armnn::TensorShape</a> dstShape({ batchSize, 3, newHeight, newWidth });</div><div class="line"><a name="l00119"></a><span class="lineno"> 119</span>&#160; std::vector&lt;float&gt; tempImage(imageData.size());</div><div class="line"><a name="l00120"></a><span class="lineno"> 120</span>&#160; <a class="code" href="namespacearmnn_utils.xhtml#af3c74017185773dd61d8ca6662d65d43">armnnUtils::Permute</a>(dstShape, NHWCToArmNN, imageData.data(), tempImage.data(), <span class="keyword">sizeof</span>(float));</div><div class="line"><a name="l00121"></a><span class="lineno"> 121</span>&#160; imageData.swap(tempImage);</div><div class="line"><a name="l00122"></a><span class="lineno"> 122</span>&#160; }</div><div class="line"><a name="l00123"></a><span class="lineno"> 123</span>&#160; <span class="keywordflow">return</span> imageData;</div><div class="line"><a name="l00124"></a><span class="lineno"> 124</span>&#160;}</div><div class="ttc" id="class_inference_test_image_xhtml_ae7a403f69a7717c1eaae1d74b7bb7bcea773b27aa8d21604182ba90d029ad2e13"><div class="ttname"><a href="class_inference_test_image.xhtml#ae7a403f69a7717c1eaae1d74b7bb7bcea773b27aa8d21604182ba90d029ad2e13">InferenceTestImage::ResizingMethods::BilinearAndNormalized</a></div></div>
365<div class="ttc" id="struct_normalization_parameters_xhtml_a8fdb461f0b4417f4ebff10b3413a948d"><div class="ttname"><a href="struct_normalization_parameters.xhtml#a8fdb461f0b4417f4ebff10b3413a948d">NormalizationParameters::stddev</a></div><div class="ttdeci">std::array&lt; float, 3 &gt; stddev</div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00022">ImageTensorGenerator.hpp:22</a></div></div>
366<div class="ttc" id="classarmnn_1_1_tensor_shape_xhtml"><div class="ttname"><a href="classarmnn_1_1_tensor_shape.xhtml">armnn::TensorShape</a></div><div class="ttdef"><b>Definition:</b> <a href="_tensor_8hpp_source.xhtml#l00020">Tensor.hpp:20</a></div></div>
367<div class="ttc" id="namespacearmnn_utils_xhtml_a12124184ac6aec018beb98b9715330c7"><div class="ttname"><a href="namespacearmnn_utils.xhtml#a12124184ac6aec018beb98b9715330c7">armnnUtils::NHWCToArmNN</a></div><div class="ttdeci">const armnn::PermutationVector NHWCToArmNN</div><div class="ttdef"><b>Definition:</b> <a href="_parser_helper_8cpp_source.xhtml#l00016">ParserHelper.cpp:16</a></div></div>
368<div class="ttc" id="namespacearmnn_utils_xhtml_af3c74017185773dd61d8ca6662d65d43"><div class="ttname"><a href="namespacearmnn_utils.xhtml#af3c74017185773dd61d8ca6662d65d43">armnnUtils::Permute</a></div><div class="ttdeci">void Permute(const armnn::TensorShape &amp;dstShape, const armnn::PermutationVector &amp;mappings, const void *src, void *dst, size_t dataTypeSize)</div><div class="ttdef"><b>Definition:</b> <a href="_permute_8cpp_source.xhtml#l00121">Permute.cpp:121</a></div></div>
369<div class="ttc" id="class_inference_test_image_xhtml_a4ec823e49ce46950b3d4cee1efff050b"><div class="ttname"><a href="class_inference_test_image.xhtml#a4ec823e49ce46950b3d4cee1efff050b">InferenceTestImage::GetWidth</a></div><div class="ttdeci">unsigned int GetWidth() const</div><div class="ttdef"><b>Definition:</b> <a href="_inference_test_image_8hpp_source.xhtml#l00075">InferenceTestImage.hpp:75</a></div></div>
370<div class="ttc" id="_exceptions_8hpp_xhtml_aa3be76aec4ce713822a5ea1ecbb7bc61"><div class="ttname"><a href="_exceptions_8hpp.xhtml#aa3be76aec4ce713822a5ea1ecbb7bc61">CHECK_LOCATION</a></div><div class="ttdeci">#define CHECK_LOCATION()</div><div class="ttdef"><b>Definition:</b> <a href="_exceptions_8hpp_source.xhtml#l00192">Exceptions.hpp:192</a></div></div>
371<div class="ttc" id="classarmnn_1_1_permutation_vector_xhtml"><div class="ttname"><a href="classarmnn_1_1_permutation_vector.xhtml">armnn::PermutationVector</a></div><div class="ttdef"><b>Definition:</b> <a href="_types_8hpp_source.xhtml#l00173">Types.hpp:173</a></div></div>
372<div class="ttc" id="class_inference_test_image_xhtml"><div class="ttname"><a href="class_inference_test_image.xhtml">InferenceTestImage</a></div><div class="ttdef"><b>Definition:</b> <a href="_inference_test_image_8hpp_source.xhtml#l00051">InferenceTestImage.hpp:51</a></div></div>
373<div class="ttc" id="struct_normalization_parameters_xhtml_a6e5ac6f9d3272e6670522daaacc04442"><div class="ttname"><a href="struct_normalization_parameters.xhtml#a6e5ac6f9d3272e6670522daaacc04442">NormalizationParameters::mean</a></div><div class="ttdeci">std::array&lt; float, 3 &gt; mean</div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00021">ImageTensorGenerator.hpp:21</a></div></div>
374<div class="ttc" id="struct_normalization_parameters_xhtml_a1d28dec57cce925ad92342891bd71e7c"><div class="ttname"><a href="struct_normalization_parameters.xhtml#a1d28dec57cce925ad92342891bd71e7c">NormalizationParameters::scale</a></div><div class="ttdeci">float scale</div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00020">ImageTensorGenerator.hpp:20</a></div></div>
375<div class="ttc" id="namespacearmnn_xhtml_ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f"><div class="ttname"><a href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0a6b99f356fe3b30a2a850b5ea897c289f">armnn::DataLayout::NCHW</a></div></div>
376</div><!-- fragment -->
377</div>
378</div>
379<a id="ac1e8e68c74e5d1c8fd19a91a68de3cb7"></a>
380<h2 class="memtitle"><span class="permalink"><a href="#ac1e8e68c74e5d1c8fd19a91a68de3cb7">&#9670;&nbsp;</a></span>PrepareImageTensor< int >()</h2>
381
382<div class="memitem">
383<div class="memproto">
384 <table class="memname">
385 <tr>
386 <td class="memname">std::vector&lt;int&gt; <a class="el" href="_image_tensor_generator_8hpp.xhtml#af39456ac4838301c6fc5b44d6a57479c">PrepareImageTensor</a>&lt; int &gt; </td>
387 <td>(</td>
388 <td class="paramtype">const std::string &amp;&#160;</td>
389 <td class="paramname"><em>imagePath</em>, </td>
390 </tr>
391 <tr>
392 <td class="paramkey"></td>
393 <td></td>
394 <td class="paramtype">unsigned int&#160;</td>
395 <td class="paramname"><em>newWidth</em>, </td>
396 </tr>
397 <tr>
398 <td class="paramkey"></td>
399 <td></td>
400 <td class="paramtype">unsigned int&#160;</td>
401 <td class="paramname"><em>newHeight</em>, </td>
402 </tr>
403 <tr>
404 <td class="paramkey"></td>
405 <td></td>
406 <td class="paramtype">const <a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> &amp;&#160;</td>
407 <td class="paramname"><em>normParams</em>, </td>
408 </tr>
409 <tr>
410 <td class="paramkey"></td>
411 <td></td>
412 <td class="paramtype">unsigned int&#160;</td>
413 <td class="paramname"><em>batchSize</em>, </td>
414 </tr>
415 <tr>
416 <td class="paramkey"></td>
417 <td></td>
418 <td class="paramtype">const <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> &amp;&#160;</td>
419 <td class="paramname"><em>outputLayout</em>&#160;</td>
420 </tr>
421 <tr>
422 <td></td>
423 <td>)</td>
424 <td></td><td></td>
425 </tr>
426 </table>
427</div><div class="memdoc">
428
429<p class="definition">Definition at line <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00128">128</a> of file <a class="el" href="_image_tensor_generator_8hpp_source.xhtml">ImageTensorGenerator.hpp</a>.</p>
430
431<p class="reference">References <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00091">PrepareImageTensor&lt; float &gt;()</a>.</p>
432
433<p class="reference">Referenced by <a class="el" href="_image_tensor_generator_8cpp_source.xhtml#l00248">main()</a>.</p>
434<div class="fragment"><div class="line"><a name="l00134"></a><span class="lineno"> 134</span>&#160;{</div><div class="line"><a name="l00135"></a><span class="lineno"> 135</span>&#160; <span class="comment">// Get float32 image tensor</span></div><div class="line"><a name="l00136"></a><span class="lineno"> 136</span>&#160; std::vector&lt;float&gt; imageDataFloat =</div><div class="line"><a name="l00137"></a><span class="lineno"> 137</span>&#160; <a class="code" href="_image_tensor_generator_8hpp.xhtml#affacfe95f011fce3b897b2745880f82a">PrepareImageTensor&lt;float&gt;</a>(imagePath, newWidth, newHeight, normParams, batchSize, outputLayout);</div><div class="line"><a name="l00138"></a><span class="lineno"> 138</span>&#160; <span class="comment">// Convert to int32 image tensor with static cast</span></div><div class="line"><a name="l00139"></a><span class="lineno"> 139</span>&#160; std::vector&lt;int&gt; imageDataInt;</div><div class="line"><a name="l00140"></a><span class="lineno"> 140</span>&#160; imageDataInt.reserve(imageDataFloat.size());</div><div class="line"><a name="l00141"></a><span class="lineno"> 141</span>&#160; std::transform(imageDataFloat.begin(), imageDataFloat.end(), std::back_inserter(imageDataInt),</div><div class="line"><a name="l00142"></a><span class="lineno"> 142</span>&#160; [](<span class="keywordtype">float</span> val) { <span class="keywordflow">return</span> <span class="keyword">static_cast&lt;</span><span class="keywordtype">int</span><span class="keyword">&gt;</span>(val); });</div><div class="line"><a name="l00143"></a><span class="lineno"> 143</span>&#160; <span class="keywordflow">return</span> imageDataInt;</div><div class="line"><a name="l00144"></a><span class="lineno"> 144</span>&#160;}</div><div class="ttc" id="_image_tensor_generator_8hpp_xhtml_affacfe95f011fce3b897b2745880f82a"><div class="ttname"><a href="_image_tensor_generator_8hpp.xhtml#affacfe95f011fce3b897b2745880f82a">PrepareImageTensor&lt; float &gt;</a></div><div class="ttdeci">std::vector&lt; float &gt; PrepareImageTensor&lt; float &gt;(const std::string &amp;imagePath, unsigned int newWidth, unsigned int newHeight, const NormalizationParameters &amp;normParams, unsigned int batchSize, const armnn::DataLayout &amp;outputLayout)</div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00091">ImageTensorGenerator.hpp:91</a></div></div>
435</div><!-- fragment -->
436</div>
437</div>
438<a id="ac372cd2b257f33f29999217e1d2d8a37"></a>
439<h2 class="memtitle"><span class="permalink"><a href="#ac372cd2b257f33f29999217e1d2d8a37">&#9670;&nbsp;</a></span>PrepareImageTensor< uint8_t >()</h2>
440
441<div class="memitem">
442<div class="memproto">
443 <table class="memname">
444 <tr>
445 <td class="memname">std::vector&lt;uint8_t&gt; <a class="el" href="_image_tensor_generator_8hpp.xhtml#af39456ac4838301c6fc5b44d6a57479c">PrepareImageTensor</a>&lt; uint8_t &gt; </td>
446 <td>(</td>
447 <td class="paramtype">const std::string &amp;&#160;</td>
448 <td class="paramname"><em>imagePath</em>, </td>
449 </tr>
450 <tr>
451 <td class="paramkey"></td>
452 <td></td>
453 <td class="paramtype">unsigned int&#160;</td>
454 <td class="paramname"><em>newWidth</em>, </td>
455 </tr>
456 <tr>
457 <td class="paramkey"></td>
458 <td></td>
459 <td class="paramtype">unsigned int&#160;</td>
460 <td class="paramname"><em>newHeight</em>, </td>
461 </tr>
462 <tr>
463 <td class="paramkey"></td>
464 <td></td>
465 <td class="paramtype">const <a class="el" href="struct_normalization_parameters.xhtml">NormalizationParameters</a> &amp;&#160;</td>
466 <td class="paramname"><em>normParams</em>, </td>
467 </tr>
468 <tr>
469 <td class="paramkey"></td>
470 <td></td>
471 <td class="paramtype">unsigned int&#160;</td>
472 <td class="paramname"><em>batchSize</em>, </td>
473 </tr>
474 <tr>
475 <td class="paramkey"></td>
476 <td></td>
477 <td class="paramtype">const <a class="el" href="namespacearmnn.xhtml#ad1d5cce2d9e9a5d61c243e5c989112e0">armnn::DataLayout</a> &amp;&#160;</td>
478 <td class="paramname"><em>outputLayout</em>&#160;</td>
479 </tr>
480 <tr>
481 <td></td>
482 <td>)</td>
483 <td></td><td></td>
484 </tr>
485 </table>
486</div><div class="memdoc">
487
488<p class="definition">Definition at line <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00148">148</a> of file <a class="el" href="_image_tensor_generator_8hpp_source.xhtml">ImageTensorGenerator.hpp</a>.</p>
489
490<p class="reference">References <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00091">PrepareImageTensor&lt; float &gt;()</a>.</p>
491
492<p class="reference">Referenced by <a class="el" href="_image_tensor_generator_8cpp_source.xhtml#l00248">main()</a>.</p>
493<div class="fragment"><div class="line"><a name="l00154"></a><span class="lineno"> 154</span>&#160;{</div><div class="line"><a name="l00155"></a><span class="lineno"> 155</span>&#160; <span class="comment">// Get float32 image tensor</span></div><div class="line"><a name="l00156"></a><span class="lineno"> 156</span>&#160; std::vector&lt;float&gt; imageDataFloat =</div><div class="line"><a name="l00157"></a><span class="lineno"> 157</span>&#160; <a class="code" href="_image_tensor_generator_8hpp.xhtml#affacfe95f011fce3b897b2745880f82a">PrepareImageTensor&lt;float&gt;</a>(imagePath, newWidth, newHeight, normParams, batchSize, outputLayout);</div><div class="line"><a name="l00158"></a><span class="lineno"> 158</span>&#160; std::vector&lt;uint8_t&gt; imageDataQasymm8;</div><div class="line"><a name="l00159"></a><span class="lineno"> 159</span>&#160; imageDataQasymm8.reserve(imageDataFloat.size());</div><div class="line"><a name="l00160"></a><span class="lineno"> 160</span>&#160; <span class="comment">// Convert to uint8 image tensor with static cast</span></div><div class="line"><a name="l00161"></a><span class="lineno"> 161</span>&#160; std::transform(imageDataFloat.begin(), imageDataFloat.end(), std::back_inserter(imageDataQasymm8),</div><div class="line"><a name="l00162"></a><span class="lineno"> 162</span>&#160; [](<span class="keywordtype">float</span> val) { <span class="keywordflow">return</span> <span class="keyword">static_cast&lt;</span>uint8_t<span class="keyword">&gt;</span>(val); });</div><div class="line"><a name="l00163"></a><span class="lineno"> 163</span>&#160; <span class="keywordflow">return</span> imageDataQasymm8;</div><div class="line"><a name="l00164"></a><span class="lineno"> 164</span>&#160;}</div><div class="ttc" id="_image_tensor_generator_8hpp_xhtml_affacfe95f011fce3b897b2745880f82a"><div class="ttname"><a href="_image_tensor_generator_8hpp.xhtml#affacfe95f011fce3b897b2745880f82a">PrepareImageTensor&lt; float &gt;</a></div><div class="ttdeci">std::vector&lt; float &gt; PrepareImageTensor&lt; float &gt;(const std::string &amp;imagePath, unsigned int newWidth, unsigned int newHeight, const NormalizationParameters &amp;normParams, unsigned int batchSize, const armnn::DataLayout &amp;outputLayout)</div><div class="ttdef"><b>Definition:</b> <a href="_image_tensor_generator_8hpp_source.xhtml#l00091">ImageTensorGenerator.hpp:91</a></div></div>
494</div><!-- fragment -->
495</div>
496</div>
497<a id="a292c4c49fc943bfcb7ee1ce1435d4690"></a>
498<h2 class="memtitle"><span class="permalink"><a href="#a292c4c49fc943bfcb7ee1ce1435d4690">&#9670;&nbsp;</a></span>WriteImageTensorImpl()</h2>
499
500<div class="memitem">
501<div class="memproto">
502 <table class="memname">
503 <tr>
504 <td class="memname">void WriteImageTensorImpl </td>
505 <td>(</td>
506 <td class="paramtype">const std::vector&lt; ElemType &gt; &amp;&#160;</td>
507 <td class="paramname"><em>imageData</em>, </td>
508 </tr>
509 <tr>
510 <td class="paramkey"></td>
511 <td></td>
512 <td class="paramtype">std::ofstream &amp;&#160;</td>
513 <td class="paramname"><em>imageTensorFile</em>&#160;</td>
514 </tr>
515 <tr>
516 <td></td>
517 <td>)</td>
518 <td></td><td></td>
519 </tr>
520 </table>
521</div><div class="memdoc">
522
523<p>Write image tensor to ofstream. </p>
524<dl class="params"><dt>Parameters</dt><dd>
525 <table class="params">
526 <tr><td class="paramdir">[in]</td><td class="paramname">imageData</td><td>Image tensor data </td></tr>
527 <tr><td class="paramdir">[in]</td><td class="paramname">imageTensorFile</td><td>Output filestream (ofstream) to which the image tensor data is written </td></tr>
528 </table>
529 </dd>
530</dl>
531
532<p class="definition">Definition at line <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00172">172</a> of file <a class="el" href="_image_tensor_generator_8hpp_source.xhtml">ImageTensorGenerator.hpp</a>.</p>
533
534<p class="reference">Referenced by <a class="el" href="_image_tensor_generator_8cpp_source.xhtml#l00248">main()</a>.</p>
535<div class="fragment"><div class="line"><a name="l00173"></a><span class="lineno"> 173</span>&#160;{</div><div class="line"><a name="l00174"></a><span class="lineno"> 174</span>&#160; std::copy(imageData.begin(), imageData.end(), std::ostream_iterator&lt;ElemType&gt;(imageTensorFile, <span class="stringliteral">&quot; &quot;</span>));</div><div class="line"><a name="l00175"></a><span class="lineno"> 175</span>&#160;}</div></div><!-- fragment -->
536</div>
537</div>
538<a id="a2120283536d91fbdee0d19fff3156994"></a>
539<h2 class="memtitle"><span class="permalink"><a href="#a2120283536d91fbdee0d19fff3156994">&#9670;&nbsp;</a></span>WriteImageTensorImpl< uint8_t >()</h2>
540
541<div class="memitem">
542<div class="memproto">
543 <table class="memname">
544 <tr>
545 <td class="memname">void <a class="el" href="_image_tensor_generator_8hpp.xhtml#a292c4c49fc943bfcb7ee1ce1435d4690">WriteImageTensorImpl</a>&lt; uint8_t &gt; </td>
546 <td>(</td>
547 <td class="paramtype">const std::vector&lt; uint8_t &gt; &amp;&#160;</td>
548 <td class="paramname"><em>imageData</em>, </td>
549 </tr>
550 <tr>
551 <td class="paramkey"></td>
552 <td></td>
553 <td class="paramtype">std::ofstream &amp;&#160;</td>
554 <td class="paramname"><em>imageTensorFile</em>&#160;</td>
555 </tr>
556 <tr>
557 <td></td>
558 <td>)</td>
559 <td></td><td></td>
560 </tr>
561 </table>
562</div><div class="memdoc">
563
564<p class="definition">Definition at line <a class="el" href="_image_tensor_generator_8hpp_source.xhtml#l00180">180</a> of file <a class="el" href="_image_tensor_generator_8hpp_source.xhtml">ImageTensorGenerator.hpp</a>.</p>
565<div class="fragment"><div class="line"><a name="l00181"></a><span class="lineno"> 181</span>&#160;{</div><div class="line"><a name="l00182"></a><span class="lineno"> 182</span>&#160; std::copy(imageData.begin(), imageData.end(), std::ostream_iterator&lt;int&gt;(imageTensorFile, <span class="stringliteral">&quot; &quot;</span>));</div><div class="line"><a name="l00183"></a><span class="lineno"> 183</span>&#160;}</div></div><!-- fragment -->
566</div>
567</div>
568</div><!-- contents -->
569</div><!-- doc-content -->
570<!-- start footer part -->
571<div id="nav-path" class="navpath"><!-- id is needed for treeview function! -->
572 <ul>
573 <li class="navelem"><a class="el" href="dir_59425e443f801f1f2fd8bbe4959a3ccf.xhtml">tests</a></li><li class="navelem"><a class="el" href="dir_db50b4cd131a46c9b297c117b868403e.xhtml">ImageTensorGenerator</a></li><li class="navelem"><a class="el" href="_image_tensor_generator_8hpp.xhtml">ImageTensorGenerator.hpp</a></li>
574 <li class="footer">Generated on Fri Mar 13 2020 16:09:16 for ArmNN by
575 <a href="http://www.doxygen.org/index.html">
576 <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.13 </li>
577 </ul>
578</div>
579</body>
580</html>