movement.html 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336
  1. <!DOCTYPE html>
  2. <!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
  3. <!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
  4. <head>
  5. <meta charset="utf-8">
  6. <meta name="viewport" content="width=device-width, initial-scale=1.0">
  7. <title>ipfml.processing.movement &mdash; ipfml v0.4.8 documentation</title>
  8. <link rel="stylesheet" href="../../../_static/css/theme.css" type="text/css" />
  9. <link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" />
  10. <link rel="index" title="Index" href="../../../genindex.html" />
  11. <link rel="search" title="Search" href="../../../search.html" />
  12. <script src="../../../_static/js/modernizr.min.js"></script>
  13. </head>
  14. <body class="wy-body-for-nav">
  15. <div class="wy-grid-for-nav">
  16. <nav data-toggle="wy-nav-shift" class="wy-nav-side">
  17. <div class="wy-side-scroll">
  18. <div class="wy-side-nav-search">
  19. <a href="../../../index.html" class="icon icon-home"> ipfml
  20. </a>
  21. <div class="version">
  22. 0.4.8
  23. </div>
  24. <div role="search">
  25. <form id="rtd-search-form" class="wy-form" action="../../../search.html" method="get">
  26. <input type="text" name="q" placeholder="Search docs" />
  27. <input type="hidden" name="check_keywords" value="yes" />
  28. <input type="hidden" name="area" value="default" />
  29. </form>
  30. </div>
  31. </div>
  32. <div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation">
  33. <p class="caption"><span class="caption-text">Contents:</span></p>
  34. <ul>
  35. <li class="toctree-l1"><a class="reference internal" href="../../../description.html">Description</a></li>
  36. <li class="toctree-l1"><a class="reference internal" href="../../../ipfml.html">Documentation</a></li>
  37. <li class="toctree-l1"><a class="reference internal" href="../../../examples.html">Examples</a></li>
  38. <li class="toctree-l1"><a class="reference internal" href="../../../contributing.html">Contributing</a></li>
  39. </ul>
  40. </div>
  41. </div>
  42. </nav>
  43. <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
  44. <nav class="wy-nav-top" aria-label="top navigation">
  45. <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
  46. <a href="../../../index.html">ipfml</a>
  47. </nav>
  48. <div class="wy-nav-content">
  49. <div class="rst-content">
  50. <div role="navigation" aria-label="breadcrumbs navigation">
  51. <ul class="wy-breadcrumbs">
  52. <li><a href="../../../index.html">Docs</a> &raquo;</li>
  53. <li><a href="../../index.html">Module code</a> &raquo;</li>
  54. <li>ipfml.processing.movement</li>
  55. <li class="wy-breadcrumbs-aside">
  56. </li>
  57. </ul>
  58. <hr/>
  59. </div>
  60. <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
  61. <div itemprop="articleBody">
  62. <h1>Source code for ipfml.processing.movement</h1><div class="highlight"><pre>
  63. <span></span><span class="sd">&quot;&quot;&quot;</span>
  64. <span class="sd">All movements that can be applied on image such as rotations, fusions, flips</span>
  65. <span class="sd">&quot;&quot;&quot;</span>
  66. <span class="c1"># main imports</span>
  67. <span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
  68. <span class="c1"># image processing imports</span>
  69. <span class="kn">from</span> <span class="nn">PIL</span> <span class="k">import</span> <span class="n">Image</span>
  70. <span class="kn">from</span> <span class="nn">skimage</span> <span class="k">import</span> <span class="n">transform</span> <span class="k">as</span> <span class="n">sk_transform</span>
  71. <span class="c1"># ipfml imports</span>
  72. <span class="kn">from</span> <span class="nn">ipfml.exceptions</span> <span class="k">import</span> <span class="n">NumpyShapeComparisonException</span>
  73. <div class="viewcode-block" id="fusion_images"><a class="viewcode-back" href="../../../ipfml/ipfml.processing.movement.html#ipfml.processing.movement.fusion_images">[docs]</a><span class="k">def</span> <span class="nf">fusion_images</span><span class="p">(</span><span class="n">images</span><span class="p">,</span> <span class="n">pil</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
  74. <span class="sd">&#39;&#39;&#39;Fusion array of images into single image</span>
  75. <span class="sd"> Args:</span>
  76. <span class="sd"> images: array of images (PIL Image or Numpy array)</span>
  77. <span class="sd"> pil: block type returned as PIL Image (default True)</span>
  78. <span class="sd"> Returns:</span>
  79. <span class="sd"> merged image from array of images</span>
  80. <span class="sd"> Raises:</span>
  81. <span class="sd"> ValueError: if `images` is not an array or is empty</span>
  82. <span class="sd"> NumpyShapeComparisonException: if `images` array contains images with different shapes</span>
  83. <span class="sd"> Example:</span>
  84. <span class="sd"> &gt;&gt;&gt; import numpy as np</span>
  85. <span class="sd"> &gt;&gt;&gt; from ipfml.processing import movement</span>
  86. <span class="sd"> &gt;&gt;&gt; image_values_1 = np.random.randint(255, size=(800, 800, 3))</span>
  87. <span class="sd"> &gt;&gt;&gt; image_values_2 = np.random.randint(255, size=(800, 800, 3))</span>
  88. <span class="sd"> &gt;&gt;&gt; merged_image = movement.fusion_images([image_values_1, image_values_2], pil=False)</span>
  89. <span class="sd"> &gt;&gt;&gt; merged_image.shape</span>
  90. <span class="sd"> (800, 800, 3)</span>
  91. <span class="sd"> &#39;&#39;&#39;</span>
  92. <span class="n">mode</span> <span class="o">=</span> <span class="s1">&#39;RGB&#39;</span>
  93. <span class="n">dim</span> <span class="o">=</span> <span class="mi">1</span>
  94. <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">images</span><span class="p">)</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
  95. <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s1">&#39;Empty array of images provided...&#39;</span><span class="p">)</span>
  96. <span class="c1"># convert image in numpy array (perhaps not necessary)</span>
  97. <span class="n">images</span> <span class="o">=</span> <span class="p">[</span><span class="n">np</span><span class="o">.</span><span class="n">asarray</span><span class="p">(</span><span class="n">img</span><span class="p">)</span> <span class="k">for</span> <span class="n">img</span> <span class="ow">in</span> <span class="n">images</span><span class="p">]</span>
  98. <span class="n">image_array</span> <span class="o">=</span> <span class="n">images</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span>
  99. <span class="k">if</span> <span class="n">image_array</span><span class="o">.</span><span class="n">ndim</span> <span class="o">!=</span> <span class="mi">3</span><span class="p">:</span>
  100. <span class="n">mode</span> <span class="o">=</span> <span class="s1">&#39;L&#39;</span>
  101. <span class="n">width</span><span class="p">,</span> <span class="n">height</span> <span class="o">=</span> <span class="n">image_array</span><span class="o">.</span><span class="n">shape</span>
  102. <span class="k">else</span><span class="p">:</span>
  103. <span class="n">width</span><span class="p">,</span> <span class="n">height</span><span class="p">,</span> <span class="n">dim</span> <span class="o">=</span> <span class="n">image_array</span><span class="o">.</span><span class="n">shape</span>
  104. <span class="c1"># raise exception if all images do not have same shape</span>
  105. <span class="k">if</span> <span class="ow">not</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">([</span><span class="n">image_array</span><span class="o">.</span><span class="n">shape</span> <span class="o">==</span> <span class="n">a</span><span class="o">.</span><span class="n">shape</span> <span class="k">for</span> <span class="n">a</span> <span class="ow">in</span> <span class="n">images</span><span class="p">])</span><span class="o">.</span><span class="n">all</span><span class="p">():</span>
  106. <span class="k">raise</span> <span class="n">NumpyShapeComparisonException</span><span class="p">()</span>
  107. <span class="k">if</span> <span class="n">dim</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
  108. <span class="n">image_mean</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">empty</span><span class="p">([</span><span class="n">width</span><span class="p">,</span> <span class="n">height</span><span class="p">])</span>
  109. <span class="k">else</span><span class="p">:</span>
  110. <span class="n">image_mean</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">empty</span><span class="p">([</span><span class="n">width</span><span class="p">,</span> <span class="n">height</span><span class="p">,</span> <span class="n">dim</span><span class="p">])</span>
  111. <span class="n">nb_images</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">images</span><span class="p">)</span>
  112. <span class="c1"># construction of mean image from rotation</span>
  113. <span class="k">for</span> <span class="n">i</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">width</span><span class="p">):</span>
  114. <span class="k">for</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">height</span><span class="p">):</span>
  115. <span class="k">if</span> <span class="n">dim</span> <span class="o">==</span> <span class="mi">1</span><span class="p">:</span>
  116. <span class="n">grey_value</span> <span class="o">=</span> <span class="mi">0</span>
  117. <span class="c1"># for each image we merge pixel values</span>
  118. <span class="k">for</span> <span class="n">img</span> <span class="ow">in</span> <span class="n">images</span><span class="p">:</span>
  119. <span class="n">grey_value</span> <span class="o">+=</span> <span class="n">img</span><span class="p">[</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">]</span>
  120. <span class="n">image_mean</span><span class="p">[</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">]</span> <span class="o">=</span> <span class="n">grey_value</span> <span class="o">/</span> <span class="n">nb_images</span>
  121. <span class="k">else</span><span class="p">:</span>
  122. <span class="k">for</span> <span class="n">k</span> <span class="ow">in</span> <span class="nb">range</span><span class="p">(</span><span class="n">dim</span><span class="p">):</span>
  123. <span class="n">canal_value</span> <span class="o">=</span> <span class="mi">0</span>
  124. <span class="c1"># for each image we merge pixel values</span>
  125. <span class="k">for</span> <span class="n">img</span> <span class="ow">in</span> <span class="n">images</span><span class="p">:</span>
  126. <span class="n">canal_value</span> <span class="o">+=</span> <span class="n">img</span><span class="p">[</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">][</span><span class="n">k</span><span class="p">]</span>
  127. <span class="n">image_mean</span><span class="p">[</span><span class="n">i</span><span class="p">][</span><span class="n">j</span><span class="p">][</span><span class="n">k</span><span class="p">]</span> <span class="o">=</span> <span class="n">canal_value</span> <span class="o">/</span> <span class="n">nb_images</span>
  128. <span class="n">image_mean</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">(</span><span class="n">image_mean</span><span class="p">,</span> <span class="s1">&#39;uint8&#39;</span><span class="p">)</span>
  129. <span class="k">if</span> <span class="n">pil</span><span class="p">:</span>
  130. <span class="k">return</span> <span class="n">Image</span><span class="o">.</span><span class="n">fromarray</span><span class="p">(</span><span class="n">image_mean</span><span class="p">,</span> <span class="n">mode</span><span class="p">)</span>
  131. <span class="k">else</span><span class="p">:</span>
  132. <span class="k">return</span> <span class="n">image_mean</span></div>
  133. <div class="viewcode-block" id="rotate_image"><a class="viewcode-back" href="../../../ipfml/ipfml.processing.movement.html#ipfml.processing.movement.rotate_image">[docs]</a><span class="k">def</span> <span class="nf">rotate_image</span><span class="p">(</span><span class="n">image</span><span class="p">,</span> <span class="n">angle</span><span class="o">=</span><span class="mi">90</span><span class="p">,</span> <span class="n">pil</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
  134. <span class="sd">&quot;&quot;&quot;Rotate image using specific angle</span>
  135. <span class="sd"> Args:</span>
  136. <span class="sd"> image: PIL Image or Numpy array</span>
  137. <span class="sd"> angle: Angle value of the rotation</span>
  138. <span class="sd"> pil: block type returned as PIL Image (default True)</span>
  139. <span class="sd"> Returns:</span>
  140. <span class="sd"> Image with rotation applied</span>
  141. <span class="sd"> Example:</span>
  142. <span class="sd"> &gt;&gt;&gt; from PIL import Image</span>
  143. <span class="sd"> &gt;&gt;&gt; import numpy as np</span>
  144. <span class="sd"> &gt;&gt;&gt; from ipfml.processing import movement</span>
  145. <span class="sd"> &gt;&gt;&gt; image_values = Image.open(&#39;./images/test_img.png&#39;)</span>
  146. <span class="sd"> &gt;&gt;&gt; rotated_image = movement.rotate_image(image_values, 90, pil=False)</span>
  147. <span class="sd"> &gt;&gt;&gt; rotated_image.shape</span>
  148. <span class="sd"> (200, 200, 3)</span>
  149. <span class="sd"> &quot;&quot;&quot;</span>
  150. <span class="n">mode</span> <span class="o">=</span> <span class="s1">&#39;RGB&#39;</span>
  151. <span class="n">image_array</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">asarray</span><span class="p">(</span><span class="n">image</span><span class="p">)</span>
  152. <span class="k">if</span> <span class="n">image_array</span><span class="o">.</span><span class="n">ndim</span> <span class="o">!=</span> <span class="mi">3</span><span class="p">:</span>
  153. <span class="n">mode</span> <span class="o">=</span> <span class="s1">&#39;L&#39;</span>
  154. <span class="n">rotated_image</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">(</span>
  155. <span class="n">sk_transform</span><span class="o">.</span><span class="n">rotate</span><span class="p">(</span><span class="n">image_array</span><span class="p">,</span> <span class="n">angle</span><span class="p">)</span> <span class="o">*</span> <span class="mi">255</span><span class="p">,</span> <span class="s1">&#39;uint8&#39;</span><span class="p">)</span>
  156. <span class="k">if</span> <span class="n">pil</span><span class="p">:</span>
  157. <span class="k">return</span> <span class="n">Image</span><span class="o">.</span><span class="n">fromarray</span><span class="p">(</span><span class="n">rotated_image</span><span class="p">,</span> <span class="n">mode</span><span class="p">)</span>
  158. <span class="k">else</span><span class="p">:</span>
  159. <span class="k">return</span> <span class="n">rotated_image</span></div>
  160. </pre></div>
  161. </div>
  162. </div>
  163. <footer>
  164. <hr/>
  165. <div role="contentinfo">
  166. <p>
  167. &copy; Copyright 2019, Jérôme BUISINE
  168. </p>
  169. </div>
  170. Built with <a href="http://sphinx-doc.org/">Sphinx</a> using a <a href="https://github.com/rtfd/sphinx_rtd_theme">theme</a> provided by <a href="https://readthedocs.org">Read the Docs</a>.
  171. </footer>
  172. </div>
  173. </div>
  174. </section>
  175. </div>
  176. <script type="text/javascript" id="documentation_options" data-url_root="../../../" src="../../../_static/documentation_options.js"></script>
  177. <script type="text/javascript" src="../../../_static/jquery.js"></script>
  178. <script type="text/javascript" src="../../../_static/underscore.js"></script>
  179. <script type="text/javascript" src="../../../_static/doctools.js"></script>
  180. <script type="text/javascript" src="../../../_static/language_data.js"></script>
  181. <script type="text/javascript" src="../../../_static/js/theme.js"></script>
  182. <script type="text/javascript">
  183. jQuery(function () {
  184. SphinxRtdTheme.Navigation.enable(true);
  185. });
  186. </script>
  187. </body>
  188. </html>