Create gh-pages branch via GitHub

index.html

@@ -186,82 +186,82 @@ In the following figure you can see that depending if the user choses k = 3, as
 <h1>
 <a name="7-code" class="anchor" href="#7-code"><span class="octicon octicon-link"></span></a>7. Code</h1>
 
-<pre><code>#!/usr/bin/env python
+<div class="highlight highlight-python"><pre><span class="c">#!/usr/bin/env python</span>
 
-import cv2, cv, sys, math, os, numpy
+<span class="kn">import</span> <span class="nn">cv2</span><span class="o">,</span> <span class="nn">cv</span><span class="o">,</span> <span class="nn">sys</span><span class="o">,</span> <span class="nn">math</span><span class="o">,</span> <span class="nn">os</span><span class="o">,</span> <span class="nn">numpy</span>
-from scipy.spatial import KDTree
+<span class="kn">from</span> <span class="nn">scipy.spatial</span> <span class="kn">import</span> <span class="n">KDTree</span>
 
-def extractFeatures(label):
+<span class="k">def</span> <span class="nf">extractFeatures</span><span class="p">(</span><span class="n">label</span><span class="p">):</span>
 
-    directory = "img/" + label + "/"
+    <span class="n">directory</span> <span class="o">=</span> <span class="s">"img/"</span> <span class="o">+</span> <span class="n">label</span> <span class="o">+</span> <span class="s">"/"</span>
 
-    features = []
+    <span class="n">features</span> <span class="o">=</span> <span class="p">[]</span>
 
-    for fn in os.listdir(directory):
+    <span class="k">for</span> <span class="n">fn</span> <span class="ow">in</span> <span class="n">os</span><span class="o">.</span><span class="n">listdir</span><span class="p">(</span><span class="n">directory</span><span class="p">):</span>
 
-        img = cv2.imread(directory + fn, 0)
+        <span class="n">img</span> <span class="o">=</span> <span class="n">cv2</span><span class="o">.</span><span class="n">imread</span><span class="p">(</span><span class="n">directory</span> <span class="o">+</span> <span class="n">fn</span><span class="p">,</span> <span class="mi">0</span><span class="p">)</span>
 
-        # find edges
+        <span class="c"># find edges</span>
-        canny = cv2.Canny(img, 50, 100)
+        <span class="n">canny</span> <span class="o">=</span> <span class="n">cv2</span><span class="o">.</span><span class="n">Canny</span><span class="p">(</span><span class="n">img</span><span class="p">,</span> <span class="mi">50</span><span class="p">,</span> <span class="mi">100</span><span class="p">)</span>
 
-        # find colored
+        <span class="c"># find colored</span>
-        black_pixels = numpy.count_nonzero(img)
+        <span class="n">black_pixels</span> <span class="o">=</span> <span class="n">numpy</span><span class="o">.</span><span class="n">count_nonzero</span><span class="p">(</span><span class="n">img</span><span class="p">)</span>
 
-        # find lines lines
+        <span class="c"># find lines lines</span>
-        lines = cv2.HoughLinesP(canny, 1, math.pi/360, 5, None, 10, 1)
+        <span class="n">lines</span> <span class="o">=</span> <span class="n">cv2</span><span class="o">.</span><span class="n">HoughLinesP</span><span class="p">(</span><span class="n">canny</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="n">math</span><span class="o">.</span><span class="n">pi</span><span class="o">/</span><span class="mi">360</span><span class="p">,</span> <span class="mi">5</span><span class="p">,</span> <span class="bp">None</span><span class="p">,</span> <span class="mi">10</span><span class="p">,</span> <span class="mi">1</span><span class="p">)</span>
 
-        lengths = []
+        <span class="n">lengths</span> <span class="o">=</span> <span class="p">[]</span>
-        angles = []
+        <span class="n">angles</span> <span class="o">=</span> <span class="p">[]</span>
-        try:
+        <span class="k">try</span><span class="p">:</span>
-            for line in lines[0]:
+            <span class="k">for</span> <span class="n">line</span> <span class="ow">in</span> <span class="n">lines</span><span class="p">[</span><span class="mi">0</span><span class="p">]:</span>
-                x1, y1, x2, y2 = line
+                <span class="n">x1</span><span class="p">,</span> <span class="n">y1</span><span class="p">,</span> <span class="n">x2</span><span class="p">,</span> <span class="n">y2</span> <span class="o">=</span> <span class="n">line</span>
 
-                # Pythagoras
+                <span class="c"># Pythagoras</span>
-                a2 = math.pow((x1-x2), 2)
+                <span class="n">a2</span> <span class="o">=</span> <span class="n">math</span><span class="o">.</span><span class="n">pow</span><span class="p">((</span><span class="n">x1</span><span class="o">-</span><span class="n">x2</span><span class="p">),</span> <span class="mi">2</span><span class="p">)</span>
-                b2 = math.pow((y1-y2), 2)
+                <span class="n">b2</span> <span class="o">=</span> <span class="n">math</span><span class="o">.</span><span class="n">pow</span><span class="p">((</span><span class="n">y1</span><span class="o">-</span><span class="n">y2</span><span class="p">),</span> <span class="mi">2</span><span class="p">)</span>
-                length = int(math.sqrt(a2 + b2))
+                <span class="n">length</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">math</span><span class="o">.</span><span class="n">sqrt</span><span class="p">(</span><span class="n">a2</span> <span class="o">+</span> <span class="n">b2</span><span class="p">))</span>
-                lengths.append(length)
+                <span class="n">lengths</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">length</span><span class="p">)</span>
 
-                angle = int(math.degrees(math.atan((y1-y2) / (x1-x2))))
+                <span class="n">angle</span> <span class="o">=</span> <span class="nb">int</span><span class="p">(</span><span class="n">math</span><span class="o">.</span><span class="n">degrees</span><span class="p">(</span><span class="n">math</span><span class="o">.</span><span class="n">atan</span><span class="p">((</span><span class="n">y1</span><span class="o">-</span><span class="n">y2</span><span class="p">)</span> <span class="o">/</span> <span class="p">(</span><span class="n">x1</span><span class="o">-</span><span class="n">x2</span><span class="p">))))</span>
-                angles.append(angle)
+                <span class="n">angles</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">angle</span><span class="p">)</span>
-        except:
+        <span class="k">except</span><span class="p">:</span>
-            pass
+            <span class="k">pass</span>
 
-        # print out everything
+        <span class="c"># print out everything</span>
-        lines_count = len(lengths)
+        <span class="n">lines_count</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">lengths</span><span class="p">)</span>
-        mid_length = sum(lengths) / lines_count
+        <span class="n">mid_length</span> <span class="o">=</span> <span class="nb">sum</span><span class="p">(</span><span class="n">lengths</span><span class="p">)</span> <span class="o">/</span> <span class="n">lines_count</span>
-        mid_angle = sum(angles) / lines_count
+        <span class="n">mid_angle</span> <span class="o">=</span> <span class="nb">sum</span><span class="p">(</span><span class="n">angles</span><span class="p">)</span> <span class="o">/</span> <span class="n">lines_count</span>
 
-        features.append([
+        <span class="n">features</span><span class="o">.</span><span class="n">append</span><span class="p">([</span>
-            [lines_count, mid_length, mid_angle, black_pixels],
+            <span class="p">[</span><span class="n">lines_count</span><span class="p">,</span> <span class="n">mid_length</span><span class="p">,</span> <span class="n">mid_angle</span><span class="p">,</span> <span class="n">black_pixels</span><span class="p">],</span>
-            label
+            <span class="n">label</span>
-        ])
+        <span class="p">])</span>
 
-    return features
+    <span class="k">return</span> <span class="n">features</span>
 
 
-if __name__ == "__main__":
+<span class="k">if</span> <span class="n">__name__</span> <span class="o">==</span> <span class="s">"__main__"</span><span class="p">:</span>
-    cats = extractFeatures("cat")
+    <span class="n">cats</span> <span class="o">=</span> <span class="n">extractFeatures</span><span class="p">(</span><span class="s">"cat"</span><span class="p">)</span>
-    dogs = extractFeatures("dog")
+    <span class="n">dogs</span> <span class="o">=</span> <span class="n">extractFeatures</span><span class="p">(</span><span class="s">"dog"</span><span class="p">)</span>
 
-    test_count = 5
+    <span class="n">test_count</span> <span class="o">=</span> <span class="mi">5</span>
 
-    test_data = dogs[:test_count] + cats[:test_count] 
+    <span class="n">test_data</span> <span class="o">=</span> <span class="n">dogs</span><span class="p">[:</span><span class="n">test_count</span><span class="p">]</span> <span class="o">+</span> <span class="n">cats</span><span class="p">[:</span><span class="n">test_count</span><span class="p">]</span> 
-    test_labels = map(lambda a: a[1], test_data)
+    <span class="n">test_labels</span> <span class="o">=</span> <span class="nb">map</span><span class="p">(</span><span class="k">lambda</span> <span class="n">a</span><span class="p">:</span> <span class="n">a</span><span class="p">[</span><span class="mi">1</span><span class="p">],</span> <span class="n">test_data</span><span class="p">)</span>
-    test_features = map(lambda a: a[0], test_data)
+    <span class="n">test_features</span> <span class="o">=</span> <span class="nb">map</span><span class="p">(</span><span class="k">lambda</span> <span class="n">a</span><span class="p">:</span> <span class="n">a</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">test_data</span><span class="p">)</span>
 
-    data = cats[test_count:] + dogs[test_count:]
+    <span class="n">data</span> <span class="o">=</span> <span class="n">cats</span><span class="p">[</span><span class="n">test_count</span><span class="p">:]</span> <span class="o">+</span> <span class="n">dogs</span><span class="p">[</span><span class="n">test_count</span><span class="p">:]</span>
-    labels = map(lambda a: a[1], data)
+    <span class="n">labels</span> <span class="o">=</span> <span class="nb">map</span><span class="p">(</span><span class="k">lambda</span> <span class="n">a</span><span class="p">:</span> <span class="n">a</span><span class="p">[</span><span class="mi">1</span><span class="p">],</span> <span class="n">data</span><span class="p">)</span>
-    features = map(lambda a: a[0], data)
+    <span class="n">features</span> <span class="o">=</span> <span class="nb">map</span><span class="p">(</span><span class="k">lambda</span> <span class="n">a</span><span class="p">:</span> <span class="n">a</span><span class="p">[</span><span class="mi">0</span><span class="p">],</span> <span class="n">data</span><span class="p">)</span>
 
-    tree = KDTree(features)
+    <span class="n">tree</span> <span class="o">=</span> <span class="n">KDTree</span><span class="p">(</span><span class="n">features</span><span class="p">)</span>
 
-    for t in xrange(0, test_count * 2):
+    <span class="k">for</span> <span class="n">t</span> <span class="ow">in</span> <span class="nb">xrange</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="n">test_count</span> <span class="o">*</span> <span class="mi">2</span><span class="p">):</span>
-        d, i = tree.query(test_features[t], k=3)
+        <span class="n">d</span><span class="p">,</span> <span class="n">i</span> <span class="o">=</span> <span class="n">tree</span><span class="o">.</span><span class="n">query</span><span class="p">(</span><span class="n">test_features</span><span class="p">[</span><span class="n">t</span><span class="p">],</span> <span class="n">k</span><span class="o">=</span><span class="mi">3</span><span class="p">)</span>
-        print "-"
+        <span class="k">print</span> <span class="s">"-"</span>
-        for j in xrange(0, len(i)):
+        <span class="k">for</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">xrange</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="nb">len</span><span class="p">(</span><span class="n">i</span><span class="p">)):</span>
-            print test_labels[t] + " is a " + labels[i[j]]
+            <span class="k">print</span> <span class="n">test_labels</span><span class="p">[</span><span class="n">t</span><span class="p">]</span> <span class="o">+</span> <span class="s">" is a "</span> <span class="o">+</span> <span class="n">labels</span><span class="p">[</span><span class="n">i</span><span class="p">[</span><span class="n">j</span><span class="p">]]</span>
-</code></pre>
+</pre></div>
       </section>
     </div>