|
@@ -0,0 +1,294 @@
|
|
|
+
|
|
|
+
|
|
|
+<!DOCTYPE html>
|
|
|
+<!--[if IE 8]><html class="no-js lt-ie9" lang="en" > <![endif]-->
|
|
|
+<!--[if gt IE 8]><!--> <html class="no-js" lang="en" > <!--<![endif]-->
|
|
|
+<head>
|
|
|
+ <meta charset="utf-8">
|
|
|
+
|
|
|
+ <meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
|
+
|
|
|
+ <title>ipfml.processing.segmentation — ipfml v0.4.1 documentation</title>
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ <link rel="stylesheet" href="../../../_static/css/theme.css" type="text/css" />
|
|
|
+ <link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" />
|
|
|
+ <link rel="index" title="Index" href="../../../genindex.html" />
|
|
|
+ <link rel="search" title="Search" href="../../../search.html" />
|
|
|
+
|
|
|
+
|
|
|
+ <script src="../../../_static/js/modernizr.min.js"></script>
|
|
|
+
|
|
|
+</head>
|
|
|
+
|
|
|
+<body class="wy-body-for-nav">
|
|
|
+
|
|
|
+
|
|
|
+ <div class="wy-grid-for-nav">
|
|
|
+
|
|
|
+
|
|
|
+ <nav data-toggle="wy-nav-shift" class="wy-nav-side">
|
|
|
+ <div class="wy-side-scroll">
|
|
|
+ <div class="wy-side-nav-search">
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ <a href="../../../index.html" class="icon icon-home"> ipfml
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ </a>
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ <div class="version">
|
|
|
+ 0.4.1
|
|
|
+ </div>
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+<div role="search">
|
|
|
+ <form id="rtd-search-form" class="wy-form" action="../../../search.html" method="get">
|
|
|
+ <input type="text" name="q" placeholder="Search docs" />
|
|
|
+ <input type="hidden" name="check_keywords" value="yes" />
|
|
|
+ <input type="hidden" name="area" value="default" />
|
|
|
+ </form>
|
|
|
+</div>
|
|
|
+
|
|
|
+
|
|
|
+ </div>
|
|
|
+
|
|
|
+ <div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="main navigation">
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ <p class="caption"><span class="caption-text">Contents:</span></p>
|
|
|
+<ul>
|
|
|
+<li class="toctree-l1"><a class="reference internal" href="../../../description.html">Description</a></li>
|
|
|
+<li class="toctree-l1"><a class="reference internal" href="../../../ipfml.html">Documentation of ipfml</a></li>
|
|
|
+<li class="toctree-l1"><a class="reference internal" href="../../../examples.html">Examples</a></li>
|
|
|
+<li class="toctree-l1"><a class="reference internal" href="../../../contributing.html">Contributing</a></li>
|
|
|
+</ul>
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ </div>
|
|
|
+ </div>
|
|
|
+ </nav>
|
|
|
+
|
|
|
+ <section data-toggle="wy-nav-shift" class="wy-nav-content-wrap">
|
|
|
+
|
|
|
+
|
|
|
+ <nav class="wy-nav-top" aria-label="top navigation">
|
|
|
+
|
|
|
+ <i data-toggle="wy-nav-top" class="fa fa-bars"></i>
|
|
|
+ <a href="../../../index.html">ipfml</a>
|
|
|
+
|
|
|
+ </nav>
|
|
|
+
|
|
|
+
|
|
|
+ <div class="wy-nav-content">
|
|
|
+
|
|
|
+ <div class="rst-content">
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+<div role="navigation" aria-label="breadcrumbs navigation">
|
|
|
+
|
|
|
+ <ul class="wy-breadcrumbs">
|
|
|
+
|
|
|
+ <li><a href="../../../index.html">Docs</a> »</li>
|
|
|
+
|
|
|
+ <li><a href="../../index.html">Module code</a> »</li>
|
|
|
+
|
|
|
+ <li>ipfml.processing.segmentation</li>
|
|
|
+
|
|
|
+
|
|
|
+ <li class="wy-breadcrumbs-aside">
|
|
|
+
|
|
|
+ </li>
|
|
|
+
|
|
|
+ </ul>
|
|
|
+
|
|
|
+
|
|
|
+ <hr/>
|
|
|
+</div>
|
|
|
+ <div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
|
|
|
+ <div itemprop="articleBody">
|
|
|
+
|
|
|
+ <h1>Source code for ipfml.processing.segmentation</h1><div class="highlight"><pre>
|
|
|
+<span></span><span class="sd">"""</span>
|
|
|
+<span class="sd">All segmentation methods applied on images</span>
|
|
|
+<span class="sd">"""</span>
|
|
|
+
|
|
|
+<span class="c1"># main imports</span>
|
|
|
+<span class="kn">import</span> <span class="nn">numpy</span> <span class="k">as</span> <span class="nn">np</span>
|
|
|
+
|
|
|
+<span class="c1"># image processing imports</span>
|
|
|
+<span class="kn">from</span> <span class="nn">PIL</span> <span class="k">import</span> <span class="n">Image</span>
|
|
|
+
|
|
|
+
|
|
|
+<div class="viewcode-block" id="divide_in_blocks"><a class="viewcode-back" href="../../../ipfml/ipfml.processing.segmentation.html#ipfml.processing.segmentation.divide_in_blocks">[docs]</a><span class="k">def</span> <span class="nf">divide_in_blocks</span><span class="p">(</span><span class="n">image</span><span class="p">,</span> <span class="n">block_size</span><span class="p">,</span> <span class="n">pil</span><span class="o">=</span><span class="kc">True</span><span class="p">):</span>
|
|
|
+ <span class="sd">'''Divide image into equal size blocks</span>
|
|
|
+
|
|
|
+<span class="sd"> Args:</span>
|
|
|
+<span class="sd"> image: PIL Image or Numpy array</span>
|
|
|
+<span class="sd"> block: tuple (width, height) representing the size of each dimension of the block</span>
|
|
|
+<span class="sd"> pil: block type returned as PIL Image (default True)</span>
|
|
|
+
|
|
|
+<span class="sd"> Returns:</span>
|
|
|
+<span class="sd"> list containing all 2D Numpy blocks (in RGB or not)</span>
|
|
|
+
|
|
|
+<span class="sd"> Raises:</span>
|
|
|
+<span class="sd"> ValueError: If `image_width` or `image_height` are not compatible to produce correct block sizes</span>
|
|
|
+
|
|
|
+<span class="sd"> Example:</span>
|
|
|
+
|
|
|
+<span class="sd"> >>> import numpy as np</span>
|
|
|
+<span class="sd"> >>> from PIL import Image</span>
|
|
|
+<span class="sd"> >>> from ipfml.processing import transform, segmentation</span>
|
|
|
+<span class="sd"> >>> image_values = np.random.randint(255, size=(800, 800, 3))</span>
|
|
|
+<span class="sd"> >>> blocks = segmentation.divide_in_blocks(image_values, (20, 20))</span>
|
|
|
+<span class="sd"> >>> len(blocks)</span>
|
|
|
+<span class="sd"> 1600</span>
|
|
|
+<span class="sd"> >>> blocks[0].width</span>
|
|
|
+<span class="sd"> 20</span>
|
|
|
+<span class="sd"> >>> blocks[0].height</span>
|
|
|
+<span class="sd"> 20</span>
|
|
|
+<span class="sd"> >>> img_l = Image.open('./images/test_img.png')</span>
|
|
|
+<span class="sd"> >>> L = transform.get_LAB_L(img_l)</span>
|
|
|
+<span class="sd"> >>> blocks_L = segmentation.divide_in_blocks(L, (100, 100))</span>
|
|
|
+<span class="sd"> >>> len(blocks_L)</span>
|
|
|
+<span class="sd"> 4</span>
|
|
|
+<span class="sd"> >>> blocks_L[0].width</span>
|
|
|
+<span class="sd"> 100</span>
|
|
|
+<span class="sd"> '''</span>
|
|
|
+
|
|
|
+ <span class="n">blocks</span> <span class="o">=</span> <span class="p">[]</span>
|
|
|
+ <span class="n">mode</span> <span class="o">=</span> <span class="s1">'RGB'</span>
|
|
|
+
|
|
|
+ <span class="c1"># convert in Numpy array</span>
|
|
|
+ <span class="n">image_array</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</span><span class="p">)</span>
|
|
|
+
|
|
|
+ <span class="c1"># check dimension of input image</span>
|
|
|
+ <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>
|
|
|
+ <span class="n">mode</span> <span class="o">=</span> <span class="s1">'L'</span>
|
|
|
+ <span class="n">image_width</span><span class="p">,</span> <span class="n">image_height</span> <span class="o">=</span> <span class="n">image_array</span><span class="o">.</span><span class="n">shape</span>
|
|
|
+ <span class="k">else</span><span class="p">:</span>
|
|
|
+ <span class="n">image_width</span><span class="p">,</span> <span class="n">image_height</span><span class="p">,</span> <span class="n">_</span> <span class="o">=</span> <span class="n">image_array</span><span class="o">.</span><span class="n">shape</span>
|
|
|
+
|
|
|
+ <span class="c1"># check size compatibility</span>
|
|
|
+ <span class="n">width</span><span class="p">,</span> <span class="n">height</span> <span class="o">=</span> <span class="n">block_size</span>
|
|
|
+
|
|
|
+ <span class="k">if</span> <span class="p">(</span><span class="n">image_width</span> <span class="o">%</span> <span class="n">width</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">):</span>
|
|
|
+ <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Width size issue, block size not compatible"</span><span class="p">)</span>
|
|
|
+
|
|
|
+ <span class="k">if</span> <span class="p">(</span><span class="n">image_height</span> <span class="o">%</span> <span class="n">height</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">):</span>
|
|
|
+ <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">"Height size issue, block size not compatible"</span><span class="p">)</span>
|
|
|
+
|
|
|
+ <span class="n">nb_block_width</span> <span class="o">=</span> <span class="n">image_width</span> <span class="o">/</span> <span class="n">width</span>
|
|
|
+ <span class="n">nb_block_height</span> <span class="o">=</span> <span class="n">image_height</span> <span class="o">/</span> <span class="n">height</span>
|
|
|
+
|
|
|
+ <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="nb">int</span><span class="p">(</span><span class="n">nb_block_width</span><span class="p">)):</span>
|
|
|
+
|
|
|
+ <span class="n">begin_x</span> <span class="o">=</span> <span class="n">i</span> <span class="o">*</span> <span class="n">width</span>
|
|
|
+
|
|
|
+ <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="nb">int</span><span class="p">(</span><span class="n">nb_block_height</span><span class="p">)):</span>
|
|
|
+
|
|
|
+ <span class="n">begin_y</span> <span class="o">=</span> <span class="n">j</span> <span class="o">*</span> <span class="n">height</span>
|
|
|
+
|
|
|
+ <span class="c1"># getting sub block information</span>
|
|
|
+ <span class="n">current_block</span> <span class="o">=</span> <span class="n">image_array</span><span class="p">[</span><span class="n">begin_x</span><span class="p">:(</span><span class="n">begin_x</span> <span class="o">+</span> <span class="n">width</span><span class="p">),</span> <span class="n">begin_y</span><span class="p">:(</span>
|
|
|
+ <span class="n">begin_y</span> <span class="o">+</span> <span class="n">height</span><span class="p">)]</span>
|
|
|
+
|
|
|
+ <span class="k">if</span> <span class="n">pil</span><span class="p">:</span>
|
|
|
+ <span class="n">blocks</span><span class="o">.</span><span class="n">append</span><span class="p">(</span>
|
|
|
+ <span class="n">Image</span><span class="o">.</span><span class="n">fromarray</span><span class="p">(</span><span class="n">current_block</span><span class="o">.</span><span class="n">astype</span><span class="p">(</span><span class="s1">'uint8'</span><span class="p">),</span> <span class="n">mode</span><span class="p">))</span>
|
|
|
+ <span class="k">else</span><span class="p">:</span>
|
|
|
+ <span class="n">blocks</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">current_block</span><span class="p">)</span>
|
|
|
+
|
|
|
+ <span class="k">return</span> <span class="n">blocks</span></div>
|
|
|
+</pre></div>
|
|
|
+
|
|
|
+ </div>
|
|
|
+
|
|
|
+ </div>
|
|
|
+ <footer>
|
|
|
+
|
|
|
+
|
|
|
+ <hr/>
|
|
|
+
|
|
|
+ <div role="contentinfo">
|
|
|
+ <p>
|
|
|
+ © Copyright 2019, Jérôme BUISINE
|
|
|
+
|
|
|
+ </p>
|
|
|
+ </div>
|
|
|
+ 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>.
|
|
|
+
|
|
|
+</footer>
|
|
|
+
|
|
|
+ </div>
|
|
|
+ </div>
|
|
|
+
|
|
|
+ </section>
|
|
|
+
|
|
|
+ </div>
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ <script type="text/javascript" id="documentation_options" data-url_root="../../../" src="../../../_static/documentation_options.js"></script>
|
|
|
+ <script type="text/javascript" src="../../../_static/jquery.js"></script>
|
|
|
+ <script type="text/javascript" src="../../../_static/underscore.js"></script>
|
|
|
+ <script type="text/javascript" src="../../../_static/doctools.js"></script>
|
|
|
+ <script type="text/javascript" src="../../../_static/language_data.js"></script>
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ <script type="text/javascript" src="../../../_static/js/theme.js"></script>
|
|
|
+
|
|
|
+ <script type="text/javascript">
|
|
|
+ jQuery(function () {
|
|
|
+ SphinxRtdTheme.Navigation.enable(true);
|
|
|
+ });
|
|
|
+ </script>
|
|
|
+
|
|
|
+</body>
|
|
|
+</html>
|