首页 浏览 帮助
注册 登录
hungerzs
/
ImageFeatrueExtraction
关注 1
点赞 0
派生 0
文件 工单管理 0 合并请求 0 Wiki
目录树: 040c7200be
分支列表 标签列表
ImageFeatrue...
/
docs
/
_modules
/
Features.html

Features.html 18 KB
文件历史 原始文件

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223 
  1. 

  2. <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
    3. 

  3.   "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
    4. 

  4. 

  5. <html xmlns="http://www.w3.org/1999/xhtml">
    6. 

  6.   <head>
    7. 

  7.     <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
    8. 

  8.     <title>Features &#8212; Image Features Extraction 1.0.0 documentation</title>
    9. 

  9.     <link rel="stylesheet" href="../_static/alabaster.css" type="text/css" />
    10. 

  10.     <link rel="stylesheet" href="../_static/pygments.css" type="text/css" />
    11. 

  11.     <script type="text/javascript">
    12. 

  12.       var DOCUMENTATION_OPTIONS = {
    13. 

  13.         URL_ROOT:    '../',
    14. 

  14.         VERSION:     '1.0.0',
    15. 

  15.         COLLAPSE_INDEX: false,
    16. 

  16.         FILE_SUFFIX: '.html',
    17. 

  17.         HAS_SOURCE:  true,
    18. 

  18.         SOURCELINK_SUFFIX: '.txt'
    19. 

  19.       };
    20. 

  20.     </script>
    21. 

  21.     <script type="text/javascript" src="../_static/jquery.js"></script>
    22. 

  22.     <script type="text/javascript" src="../_static/underscore.js"></script>
    23. 

  23.     <script type="text/javascript" src="../_static/doctools.js"></script>
    24. 

  24.     <link rel="index" title="Index" href="../genindex.html" />
    25. 

  25.     <link rel="search" title="Search" href="../search.html" />
    26. 

  26.    
    27. 

  27.   <link rel="stylesheet" href="../_static/custom.css" type="text/css" />
    28. 

  28.   
    29. 

  29.   
    30. 

  30.   <meta name="viewport" content="width=device-width, initial-scale=0.9, maximum-scale=0.9" />
    31. 

  31. 

  32.   </head>
    33. 

  33.   <body>
    34. 

  34.   
    35. 

  35. 

  36.     <div class="document">
    37. 

  37.       <div class="documentwrapper">
    38. 

  38.         <div class="bodywrapper">
    39. 

  39.           <div class="body" role="main">
    40. 

  40.             
    41. 

  41.   <h1>Source code for Features</h1><div class="highlight"><pre>
    42. 

  42. <span></span><span class="kn">import</span> <span class="nn">pandas</span> <span class="k">as</span> <span class="nn">pd</span>
    43. 

  43. <span class="kn">import</span> <span class="nn">os.path</span>
    44. 

  44. <span class="kn">from</span> <span class="nn">image_features_extraction</span> <span class="k">import</span> <span class="n">MyException</span>
    45. 

  45. 

  46. <div class="viewcode-block" id="Features"><a class="viewcode-back" href="../code.html#Features.Features">[docs]</a><span class="k">class</span> <span class="nc">Features</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
    47. 

  47.     <span class="sd">&quot;&quot;&quot;</span>
    48. 

  48. <span class="sd">    contain the dataframe</span>
    49. 

  49. <span class="sd">    &quot;&quot;&quot;</span>
    50. 

  50.     <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">data_frame</span><span class="p">):</span>
    51. 

  51.         <span class="bp">self</span><span class="o">.</span><span class="n">__data_frame</span> <span class="o">=</span> <span class="n">data_frame</span>
    52. 

  52.         <span class="bp">self</span><span class="o">.</span><span class="n">__class_name</span> <span class="o">=</span> <span class="s1">&#39;&#39;</span>
    53. 

  53.         <span class="bp">self</span><span class="o">.</span><span class="n">__class_value</span><span class="o">=</span><span class="kc">None</span>
    54. 

  54. 

  55. 

  56. <div class="viewcode-block" id="Features.set_class_name"><a class="viewcode-back" href="../code.html#Features.Features.set_class_name">[docs]</a>    <span class="k">def</span> <span class="nf">set_class_name</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">class_name</span><span class="p">):</span>
    57. 

  57.         <span class="bp">self</span><span class="o">.</span><span class="n">__class_name</span> <span class="o">=</span> <span class="n">class_name</span></div>
    58. 

  58. 

  59. 

  60. <div class="viewcode-block" id="Features.set_class_value"><a class="viewcode-back" href="../code.html#Features.Features.set_class_value">[docs]</a>    <span class="k">def</span> <span class="nf">set_class_value</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">class_value</span><span class="p">):</span>
    61. 

  61.         <span class="bp">self</span><span class="o">.</span><span class="n">__class_value</span> <span class="o">=</span> <span class="n">class_value</span></div>
    62. 

  62. 

  63. 

  64. <div class="viewcode-block" id="Features.get_class_name"><a class="viewcode-back" href="../code.html#Features.Features.get_class_name">[docs]</a>    <span class="k">def</span> <span class="nf">get_class_name</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
    65. 

  65.         <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class_name</span></div>
    66. 

  66. 

  67. 

  68. <div class="viewcode-block" id="Features.get_class_value"><a class="viewcode-back" href="../code.html#Features.Features.get_class_value">[docs]</a>    <span class="k">def</span> <span class="nf">get_class_value</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
    69. 

  69.         <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class_value</span></div>
    70. 

  70. 

  71. 

  72. <div class="viewcode-block" id="Features.merge"><a class="viewcode-back" href="../code.html#Features.Features.merge">[docs]</a>    <span class="k">def</span> <span class="nf">merge</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">Features_Obj</span><span class="p">,</span> <span class="n">how_in</span><span class="o">=</span><span class="s1">&#39;inner&#39;</span><span class="p">):</span>
    73. 

  73.         <span class="sd">&quot;&quot;&quot;</span>
    74. 

  74. <span class="sd">        Merges in the current Features object a second Features object (ex. obtained from the Voronoi Object)</span>
    75. 

  75. 

  76. <span class="sd">        :param Features_Obj: External Features Object</span>
    77. 

  77. <span class="sd">        :type Features_Obj: Features (image_features_extraction package)</span>
    78. 

  78. <span class="sd">        :param how_in: &#39;inner&#39;, &#39;outer&#39;</span>
    79. 

  79. <span class="sd">        :type how_in: string</span>
    80. 

  80. <span class="sd">        &gt;&gt;&gt; import image_features_extraction.Images as fe</span>
    81. 

  81. <span class="sd">        &gt;&gt;&gt; IMGS = fe.Images(&#39;../images/CA/1&#39;)</span>
    82. 

  82. <span class="sd">        &gt;&gt;&gt; vor = IMG.Voronoi()</span>
    83. 

  83. <span class="sd">        &gt;&gt;&gt; features1 = IMG.features([&#39;area&#39;,&#39;perimeter&#39;,&#39;centroid&#39;,&#39;bbox&#39;, &#39;eccentricity&#39;])</span>
    84. 

  84. <span class="sd">        &gt;&gt;&gt; features2 = vor.features([&#39;area&#39;,&#39;perimeter&#39;,&#39;centroid&#39;,&#39;bbox&#39;, &#39;eccentricity&#39;])</span>
    85. 

  85. <span class="sd">        &gt;&gt;&gt; features3 = features1.merge(features2, how_in=&#39;inner&#39;)</span>
    86. 

  86. <span class="sd">        &gt;&gt;&gt; features3.get_dataframe().head()</span>
    87. 

  87. <span class="sd">        &quot;&quot;&quot;</span>
    88. 

  88.         <span class="n">df1</span> <span class="o">=</span> <span class="n">Features_Obj</span><span class="o">.</span><span class="n">get_dataframe</span><span class="p">();</span>
    89. 

  89.         <span class="n">df2</span> <span class="o">=</span> <span class="n">pd</span><span class="o">.</span><span class="n">merge</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">__data_frame</span><span class="p">,</span> <span class="n">df1</span><span class="p">,</span> <span class="n">on</span><span class="o">=</span><span class="s1">&#39;id&#39;</span><span class="p">,</span> <span class="n">how</span><span class="o">=</span><span class="n">how_in</span><span class="p">)</span>
    90. 

  90.         <span class="k">return</span> <span class="n">Features</span><span class="p">(</span><span class="n">df2</span><span class="p">)</span></div>
    91. 

  91. 

  92. 

  93. <div class="viewcode-block" id="Features.save"><a class="viewcode-back" href="../code.html#Features.Features.save">[docs]</a>    <span class="k">def</span> <span class="nf">save</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">storage_name</span><span class="p">,</span> <span class="n">type_storage</span><span class="o">=</span><span class="s1">&#39;file&#39;</span><span class="p">,</span> <span class="n">do_append</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
    94. 

  94.         <span class="sd">&quot;&quot;&quot;</span>
    95. 

  95. <span class="sd">        save the current dataframe into the type of storage given in input.</span>
    96. 

  96. 

  97. <span class="sd">        :param storage_name: storage name, (e.g., file name if &#39; storage type is &#39;file&#39;)</span>
    98. 

  98. <span class="sd">        :type storage_name: string</span>
    99. 

  99. <span class="sd">        :param type_storage:  type of storage (default is &#39;file&#39;) (&#39;db&#39;, &#39;json&#39; will be future implementations)</span>
    100. 

  100. <span class="sd">        :type type_storage: string</span>
    101. 

  101. <span class="sd">        :param do_append: if True it appends to existing storage. If False it creates a new storage</span>
    102. 

  102. <span class="sd">        :type do_append: boolean</span>
    103. 

  103. <span class="sd">        if  &#39;do_append=True&#39;: This version of the method does not check whether the new data are consistent with presisitng data into</span>
    104. 

  104. <span class="sd">        the existing storage. It just tries to append the date and might fail or not depending on the type of storage</span>
    105. 

  105. <span class="sd">        </span>
    106. 

  106. <span class="sd">        :returns: 1 if sucessuful otherwise 0</span>
    107. 

  107. <span class="sd">        :rtype: int</span>
    108. 

  108. 

  109. <span class="sd">        :example:</span>
    110. 

  110. <span class="sd">        &gt;&gt;&gt; import image_features_extraction as fe</span>
    111. 

  111. <span class="sd">        &gt;&gt;&gt; imgs = fe.Images(folder_name)</span>
    112. 

  112. <span class="sd">        &gt;&gt;&gt; img = imgs.item(1)</span>
    113. 

  113. <span class="sd">        &gt;&gt;&gt; reg = img.Regions().item(1)</span>
    114. 

  114. <span class="sd">        &gt;&gt;&gt; features = regs.get_features([&#39;label&#39;, &#39;area&#39;,&#39;perimeter&#39;, &#39;centroid&#39;], class_value=1)</span>
    115. 

  115. <span class="sd">        &gt;&gt;&gt; features.save(&#39;my_file_name&#39;)</span>
    116. 

  116. <span class="sd">        &quot;&quot;&quot;</span>
    117. 

  117.         <span class="k">try</span><span class="p">:</span>
    118. 

  118.             <span class="k">if</span> <span class="n">type_storage</span> <span class="o">==</span> <span class="s1">&#39;file&#39;</span><span class="p">:</span>
    119. 

  119.                 <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">__save_file</span><span class="p">(</span><span class="n">storage_name</span><span class="p">,</span> <span class="n">do_append</span><span class="p">)</span>
    120. 

  120.             <span class="k">else</span><span class="p">:</span>
    121. 

  121.                 <span class="k">raise</span> <span class="n">MyException</span><span class="o">.</span><span class="n">MyException</span><span class="p">(</span><span class="s2">&quot;error: storage type no specified or not found&quot;</span><span class="p">)</span>
    122. 

  122.             <span class="k">return</span> <span class="mi">0</span>
    123. 

  123.         <span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
    124. 

  124.             <span class="nb">print</span><span class="p">(</span><span class="s2">&quot;one or more input labels might be wrong:</span><span class="si">{}</span><span class="s2">&quot;</span><span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">e</span><span class="p">))</span>
    125. 

  125.             <span class="k">return</span> <span class="mi">0</span></div>
    126. 

  126. 

  127. 

  128.     <span class="k">def</span> <span class="nf">__save_file</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">file_name</span><span class="p">,</span> <span class="n">do_append</span><span class="p">):</span>
    129. 

  129.         <span class="sd">&quot;&quot;&quot;</span>
    130. 

  130. <span class="sd">        load the data from the local directory with the name indicated</span>
    131. 

  131. <span class="sd">        &quot;&quot;&quot;</span>
    132. 

  132.         <span class="k">if</span> <span class="n">do_append</span><span class="o">==</span><span class="kc">True</span><span class="p">:</span>
    133. 

  133.             <span class="n">add_header</span> <span class="o">=</span> <span class="kc">False</span>
    134. 

  134.             <span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">isfile</span><span class="p">(</span><span class="n">file_name</span><span class="p">)</span> <span class="o">==</span> <span class="kc">False</span><span class="p">:</span>
    135. 

  135.                 <span class="n">add_header</span> <span class="o">=</span> <span class="kc">True</span>
    136. 

  136.             <span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="n">file_name</span><span class="p">,</span> <span class="s1">&#39;a&#39;</span><span class="p">)</span> <span class="k">as</span> <span class="n">f</span><span class="p">:</span> <span class="c1"># it appends or creates a new file</span>
    137. 

  137.                 <span class="bp">self</span><span class="o">.</span><span class="n">__data_frame</span><span class="o">.</span><span class="n">to_csv</span><span class="p">(</span><span class="n">f</span><span class="p">,</span> <span class="n">header</span><span class="o">=</span><span class="n">add_header</span><span class="p">)</span>
    138. 

  138.         <span class="k">else</span><span class="p">:</span>
    139. 

  139.             <span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="n">file_name</span><span class="p">,</span> <span class="s1">&#39;w&#39;</span><span class="p">)</span> <span class="k">as</span> <span class="n">f</span><span class="p">:</span> <span class="c1"># it creates a new file</span>
    140. 

  140.                 <span class="bp">self</span><span class="o">.</span><span class="n">__data_frame</span><span class="o">.</span><span class="n">to_csv</span><span class="p">(</span><span class="n">f</span><span class="p">,</span> <span class="n">header</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
    141. 

  141.         <span class="k">return</span> <span class="mi">1</span>
    142. 

  142. 

  143. 

  144. <div class="viewcode-block" id="Features.get_dataframe"><a class="viewcode-back" href="../code.html#Features.Features.get_dataframe">[docs]</a>    <span class="k">def</span> <span class="nf">get_dataframe</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">include_class</span><span class="o">=</span><span class="kc">False</span><span class="p">):</span>
    145. 

  145.         <span class="sd">&quot;&quot;&quot;</span>
    146. 

  146. <span class="sd">        creates  a panda data frame contining the feature values</span>
    147. 

  147. 

  148. <span class="sd">        :param include_class: if True if includes the class name and value as the last column</span>
    149. 

  149. <span class="sd">        :type include_class: Boolean</span>
    150. 

  150. <span class="sd">        &quot;&quot;&quot;</span>
    151. 

  151.         <span class="k">try</span><span class="p">:</span>
    152. 

  152.             <span class="n">df</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">__data_frame</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span>
    153. 

  153. 

  154.             <span class="k">if</span> <span class="n">include_class</span> <span class="o">==</span> <span class="kc">True</span><span class="p">:</span>
    155. 

  155.                 <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class_name</span> <span class="o">==</span> <span class="s1">&#39;&#39;</span><span class="p">:</span>
    156. 

  156.                     <span class="k">raise</span> <span class="n">MyException</span><span class="o">.</span><span class="n">MyException</span><span class="p">(</span><span class="s2">&quot;error: class name not set&quot;</span><span class="p">)</span>
    157. 

  157.                 <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class_value</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
    158. 

  158.                     <span class="k">raise</span> <span class="n">MyException</span><span class="o">.</span><span class="n">MyException</span><span class="p">(</span><span class="s2">&quot;error: class value not set&quot;</span><span class="p">)</span>
    159. 

  159.                 <span class="n">df</span><span class="p">[</span><span class="bp">self</span><span class="o">.</span><span class="n">__class_name</span><span class="p">]</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">__class_value</span>
    160. 

  160. 

  161.             <span class="k">return</span> <span class="n">df</span>
    162. 

  162.         <span class="k">except</span> <span class="n">MyException</span><span class="o">.</span><span class="n">MyException</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
    163. 

  163.             <span class="nb">print</span><span class="p">(</span><span class="n">e</span><span class="o">.</span><span class="n">args</span><span class="p">)</span>
    164. 

  164.             <span class="k">return</span> <span class="kc">None</span></div></div>
    165. 

  165. </pre></div>
    166. 

  166. 

  167.           </div>
    168. 

  168.         </div>
    169. 

  169.       </div>
    170. 

  170.       <div class="sphinxsidebar" role="navigation" aria-label="main navigation">
    171. 

  171.         <div class="sphinxsidebarwrapper">
    172. 

  172. <h1 class="logo"><a href="../index.html">Image Features Extraction</a></h1>
    173. 

  173. 

  174. 

  175. 

  176. 

  177. 

  178. 

  179. 

  180. 

  181. <h3>Navigation</h3>
    182. 

  182. <ul>
    183. 

  183. <li class="toctree-l1"><a class="reference internal" href="../tutorial.html">Tutorial</a></li>
    184. 

  184. <li class="toctree-l1"><a class="reference internal" href="../code.html">Classes/Modules</a></li>
    185. 

  185. </ul>
    186. 

  186. 

  187. <div class="relations">
    188. 

  188. <h3>Related Topics</h3>
    189. 

  189. <ul>
    190. 

  190.   <li><a href="../index.html">Documentation overview</a><ul>
    191. 

  191.   <li><a href="index.html">Module code</a><ul>
    192. 

  192.   </ul></li>
    193. 

  193.   </ul></li>
    194. 

  194. </ul>
    195. 

  195. </div>
    196. 

  196. <div id="searchbox" style="display: none" role="search">
    197. 

  197.   <h3>Quick search</h3>
    198. 

  198.     <form class="search" action="../search.html" method="get">
    199. 

  199.       <div><input type="text" name="q" /></div>
    200. 

  200.       <div><input type="submit" value="Go" /></div>
    201. 

  201.       <input type="hidden" name="check_keywords" value="yes" />
    202. 

  202.       <input type="hidden" name="area" value="default" />
    203. 

  203.     </form>
    204. 

  204. </div>
    205. 

  205. <script type="text/javascript">$('#searchbox').show(0);</script>
    206. 

  206.         </div>
    207. 

  207.       </div>
    208. 

  208.       <div class="clearer"></div>
    209. 

  209.     </div>
    210. 

  210.     <div class="footer">
    211. 

  211.       &copy;2017, Remigio Picone.
    212. 

  212.       
    213. 

  213.       |
    214. 

  214.       Powered by <a href="http://sphinx-doc.org/">Sphinx 1.6.3</a>
    215. 

  215.       &amp; <a href="https://github.com/bitprophet/alabaster">Alabaster 0.7.10</a>
    216. 

  216.       
    217. 

  217.     </div>
    218. 

  218. 

  219.     
    220. 

  220. 

  221.     
    222. 

  222.   </body>
    223. 

  223. </html>
    224.