Commit 34d11f9d authored by Jayke Meijer's avatar Jayke Meijer

Modified GetPlate and GetPlateTest to handle Point datatype instead of raw coordinates.

parent c6ab9323
from pylab import array, zeros, inv, dot, svd, shape from pylab import array, zeros, inv, dot, svd, shape
from Interpolation import pV from Interpolation import pV
def getPlateAt(image, x1, y1, x2, y2, x3, y3, x4, y4, M, N): def getPlateAt(image, points, M, N):
'''Returns an image of size MxN of the licenseplate (or any rectangular '''Returns an image of size MxN of the licenseplate (or any rectangular
object) defined by the corner points (x1, y1) to (x4, y4).''' object) defined by 4 corner points.'''
# Construct the matrix M # Construct the matrix M
x1_a, y1_a = 0, 0 x1_a, y1_a = 0, 0
x2_a, y2_a = M, 0 x2_a, y2_a = M, 0
x3_a, y3_a = M, N x3_a, y3_a = M, N
x4_a, y4_a = 0, N x4_a, y4_a = 0, N
mat_M = array([[x1, y1, 1, 0, 0, 0, -x1_a * x1, -x1_a * y1, -x1_a], \ p_i = []
[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], \ for point in points:
[0, 0, 0, x2, y2, 1, -y2_a * x2, -y2_a * y2, -y2_a], \ p_i.append([point.x, point.y])
[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], \ mat_M = array([[p_i[0][0], p_i[0][1], 1, 0, 0, 0, \
[x4, y4, 1, 0, 0, 0, -x4_a * x4, -x4_a * y4, -x4_a], \ -x1_a * p_i[0][0], -x1_a * p_i[0][1], -x1_a], \
[0, 0, 0, x4, y4, 1, -y4_a * x4, -y4_a * y4, -y4_a]]) [0, 0, 0, p_i[0][0], p_i[0][1], 1, \
-y1_a * p_i[0][0], -y1_a * p_i[0][1], -y1_a], \
[p_i[1][0], p_i[1][1], 1, 0, 0, 0, \
-x2_a * p_i[1][0], -x2_a * p_i[1][1], -x2_a], \
[0, 0, 0, p_i[1][0], p_i[1][1], 1, \
-y2_a * p_i[1][0], -y2_a * p_i[1][1], -y2_a], \
[p_i[2][0], p_i[2][1], 1, 0, 0, 0, \
-x3_a * p_i[2][0], -x3_a * p_i[2][1], -x3_a], \
[0, 0, 0, p_i[2][0], p_i[2][1], 1, \
-y3_a * p_i[2][0], -y3_a * p_i[2][1], -y3_a], \
[p_i[3][0], p_i[3][1], 1, 0, 0, 0, \
-x4_a * p_i[3][0], -x4_a * p_i[3][1], -x4_a], \
[0, 0, 0, p_i[3][0], p_i[3][1], 1, \
-y4_a * p_i[3][0], -y4_a * p_i[3][1], -y4_a]])
# Get the vector p and the values that are in there by taking the SVD. # 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 # Since D is diagonal with the eigenvalues sorted from large to small on
......
from GetPlate import getPlateAt from GetPlate import getPlateAt
from GrayscaleImage import GrayscaleImage from GrayscaleImage import GrayscaleImage
from Point import Point
# Define the coordinates of the licenseplate # Define the corners of the licenseplate
x_vals = [310, 382, 382, 310] points = [Point(None, (310, 383)), Point(None, (382, 381)), Point(None, (382, 396)), Point(None, (310, 398))]
y_vals = [383, 381, 396, 398]
# Get the image # Get the image
image = GrayscaleImage('../images/test_plate.png') image = GrayscaleImage('../images/test_plate.png')
# Let the code get the licenseplate # Let the code get the licenseplate
output_image = getPlateAt(image, x_vals[0], y_vals[0],\ output_image = getPlateAt(image, points, 100, 20)
x_vals[1], y_vals[1],\
x_vals[2], y_vals[2],\
x_vals[3], y_vals[3], 100, 20)
# Show the licenseplate # Show the licenseplate
output_image = GrayscaleImage(None, output_image) output_image = GrayscaleImage(None, output_image)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment