Merge branch 'master' of github.com:taddeus/licenseplates

src/GetPlate.py

@@ -0,0 +1,52 @@
+from pylab import array, zeros, inv, dot, svd, shape
+from Interpolation import pV
+
+def getPlateAt(image, x1, y1, x2, y2, x3, y3, x4, y4, M, N):
+    '''Returns an image of size MxN of the licenseplate (or any rectangular 
+    object) defined by the corner points (x1, y1) to (x4, y4).'''
+    # Construct the matrix M
+    x1_a, y1_a = 0, 0
+    x2_a, y2_a = M, 0
+    x3_a, y3_a = M, N
+    x4_a, y4_a = 0, N
+    
+    mat_M = array([[x1, y1, 1, 0,  0,  0, -x1_a * x1, -x1_a * y1, -x1_a], \
+                   [0,  0,  0, x1, y1, 1, -y1_a * x1, -y1_a * y1, -y1_a], \
+                   [x2, y2, 1, 0,  0,  0, -x2_a * x2, -x2_a * y2, -x2_a], \
+                   [0,  0,  0, x2, y2, 1, -y2_a * x2, -y2_a * y2, -y2_a], \
+                   [x3, y3, 1, 0,  0,  0, -x3_a * x3, -x3_a * y3, -x3_a], \
+                   [0,  0,  0, x3, y3, 1, -y3_a * x3, -y3_a * y3, -y3_a], \
+                   [x4, y4, 1, 0,  0,  0, -x4_a * x4, -x4_a * y4, -x4_a], \
+                   [0,  0,  0, x4, y4, 1, -y4_a * x4, -y4_a * y4, -y4_a]])
+    
+    # Get the vector p and the values that are in there by taking the SVD. 
+    # Since D is diagonal with the eigenvalues sorted from large to small on
+    # the diagonal, the optimal q in min ||Dq|| is q = [[0]..[1]]. Therefore, 
+    # p = Vq means p is the last column in V.
+    U, D, V = svd(mat_M)
+    p = V[8][:]                
+    a, b, c, d, e, f, g, h, i = p[0], \
+                                p[1], \
+                                p[2], \
+                                p[3], \
+                                p[4], \
+                                p[5], \
+                                p[6], \
+                                p[7], \
+                                p[8]
+    
+    # P is the resulting matrix that describes the transformation
+    P = array([[a, b, c], \
+               [d, e, f], \
+               [g, h, i]])
+    
+    # Create the new image
+    b = array([zeros(M, float)] * N)
+    for i in range(0, M):
+        for j in range(0, N):
+            or_coor = dot(inv(P),([[i],[j],[1]]))
+            or_coor_h = or_coor[1][0] / or_coor[2][0], \
+                      or_coor[0][0] / or_coor[2][0]
+            b[j][i] = pV(image, or_coor_h[0], or_coor_h[1])
+    
+    return b

src/Interpolation.py

@@ -0,0 +1,30 @@
+from pylab import floor
+
+def pV(image, x, y):
+    '''Get the value of a point x,y in the given image, where x and y are not
+    necessary integers, so the value is interpolated from its neighbouring
+    pixels.'''
+    if inImage(image, x, y):
+        x_low = floor(x)
+        x_high = floor(x + 1)
+        y_low = floor(y)
+        y_high = floor(y + 1)
+        x_y = (x_high - x_low) * (y_high - y_low)
+        
+        interpolatedValue = image[x_low, y_low] / x_y * (x_high - x) * \
+                                (y_high - y)\
+                          + image[x_high][y_low] / x_y * (x - x_low) * \
+                                (y_high - y)\
+                          + image[x_low][y_high] / x_y * (x_high - x) * \
+                                (y - y_low)\
+                          + image[x_high][y_high] / x_y * (x - x_low) * \
+                                (y - y_low)
+        return interpolatedValue
+    else:
+        constantValue = 0
+        return constantValue
+    
+def inImage(image, x, y):
+    '''Return if the pixels is within the image bounds.'''
+    return (x > 0 and x < image.get_height() - 1 \
+        and y > 0 and y < image.get_width() - 1)

src/testGetPlate.py

@@ -0,0 +1,20 @@
+from GetPlate import getPlateAt
+from pylab import imread, imshow, show, figure
+from GrayscaleImage import GrayscaleImage
+
+# Define the coordinates of the licenseplate
+x_vals = [310, 382, 382, 310]
+y_vals = [383, 381, 396, 398]
+
+# Get the image
+image = GrayscaleImage('../images/test_plate.png')
+
+# Let the code get the licenseplate
+output_image = getPlateAt(image, x_vals[0], y_vals[0],\
+                                 x_vals[1], y_vals[1],\
+                                 x_vals[2], y_vals[2],\
+                                 x_vals[3], y_vals[3], 100, 20)
+
+# Show the licenseplate                 
+output_image = GrayscaleImage(None, output_image)
+output_image.show()