This repository has been archived by the owner on Mar 22, 2020. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathMain.py
180 lines (134 loc) · 7.96 KB
/
Main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
# Main.py
import cv2
import numpy as np
import os
import re
import DetectChars
import DetectPlates
import PossiblePlate
import csv
from datetime import datetime
import time
# module level variables ##########################################################################
SCALAR_BLACK = (0.0, 0.0, 0.0)
SCALAR_WHITE = (255.0, 255.0, 255.0)
SCALAR_YELLOW = (0.0, 255.0, 255.0)
SCALAR_GREEN = (0.0, 255.0, 0.0)
SCALAR_RED = (0.0, 0.0, 255.0)
showSteps = False
###################################################################################################
def main(imgdata):
print("entered")
blnKNNTrainingSuccessful = DetectChars.loadKNNDataAndTrainKNN() # attempt KNN training
if blnKNNTrainingSuccessful == False: # if KNN training was not successful
print("\nerror: KNN traning was not successful\n") # show error message
return # and exit program
# end if0
imgOriginalScene = imgdata # open image
if imgOriginalScene is None: # if image was not read successfully
print("\nerror: image not read from file \n\n") # print error message to std out
os.system("pause") # pause so user can see error message
return # and exit program
# end if
listOfPossiblePlates = DetectPlates.detectPlatesInScene(imgOriginalScene) # detect plates
listOfPossiblePlates = DetectChars.detectCharsInPlates(listOfPossiblePlates) # detect chars in plates
cv2.imshow("imgOriginalScene", imgOriginalScene) # show scene image
if len(listOfPossiblePlates) == 0: # if no plates were found
print("\nno license plates were detected\n") # inform user no plates were found
else: # else
# if we get in here list of possible plates has at leat one plate
# sort the list of possible plates in DESCENDING order (most number of chars to least number of chars)
listOfPossiblePlates.sort(key = lambda possiblePlate: len(possiblePlate.strChars), reverse = True)
# suppose the plate with the most recognized chars (the first plate in sorted by string length descending order) is the actual plate
licPlate = listOfPossiblePlates[0]
cv2.imshow("imgPlate", licPlate.imgPlate) # show crop of plate and threshold of plate
cv2.imshow("imgThresh", licPlate.imgThresh)
if len(licPlate.strChars) == 0: # if no chars were found in the plate
print("\nno characters were detected\n\n") # show message
return # and exit program
# end if
#drawRedRectangleAroundPlate(imgOriginalScene, licPlate) # draw red rectangle around plate
print("\nlicense plate read from image = " + licPlate.strChars + "\n") # write license plate text to std out
print("----------------------------------------")
#writeLicensePlateCharsOnImage(imgOriginalScene, licPlate) # write license plate text on the image
#cv2.imshow("imgOriginalScene", imgOriginalScene) # re-show scene image
date=str(datetime.now().strftime('%Y_%m_%d'))
time=str(datetime.now().strftime('%H_%M_%S'))
filename = str(datetime.now().strftime('%Y_%m_%d_%H_%M_%S'))
flag=1
with open(r"logs/log.txt", newline = "") as file:
reader = csv.reader(file)
for col in reader:
c = 0
for row in col:
if c==2:
datechek=row
if c==1:
numberchek=row
c += 1
if datechek==date and numberchek==licPlate.strChars:
flag=0
#file.close()
if flag==1:
file=open(r"logs/log.txt","a+")
file.write(filename+".png,"+licPlate.strChars+","+date+","+time+","+filename+"\n")
cv2.imwrite(r"logs/images/{}.png".format(filename), imgOriginalScene) # write image out to file
else:
print("already exist")
# end if else
cv2.waitKey(0) # hold windows open until user presses a key
return
# end main
###################################################################################################
#def drawRedRectangleAroundPlate(imgOriginalScene, licPlate):
#
# p2fRectPoints = cv2.boxPoints(licPlate.rrLocationOfPlateInScene) # get 4 vertices of rotated rect
#
# cv2.line(imgOriginalScene, tuple(p2fRectPoints[0]), tuple(p2fRectPoints[1]), SCALAR_RED, 2) # draw 4 red lines
# cv2.line(imgOriginalScene, tuple(p2fRectPoints[1]), tuple(p2fRectPoints[2]), SCALAR_RED, 2)
# cv2.line(imgOriginalScene, tuple(p2fRectPoints[2]), tuple(p2fRectPoints[3]), SCALAR_RED, 2)
# cv2.line(imgOriginalScene, tuple(p2fRectPoints[3]), tuple(p2fRectPoints[0]), SCALAR_RED, 2)
## end function
#
####################################################################################################
#def writeLicensePlateCharsOnImage(imgOriginalScene, licPlate):
# ptCenterOfTextAreaX = 0 # this will be the center of the area the text will be written to
# ptCenterOfTextAreaY = 0
#
# ptLowerLeftTextOriginX = 0 # this will be the bottom left of the area that the text will be written to
# ptLowerLeftTextOriginY = 0
#
# sceneHeight, sceneWidth, sceneNumChannels = imgOriginalScene.shape
# plateHeight, plateWidth, plateNumChannels = licPlate.imgPlate.shape
#
# intFontFace = cv2.FONT_HERSHEY_SIMPLEX # choose a plain jane font
# fltFontScale = float(plateHeight) / 30.0 # base font scale on height of plate area
# intFontThickness = int(round(fltFontScale * 1.5)) # base font thickness on font scale
#
# textSize, baseline = cv2.getTextSize(licPlate.strChars, intFontFace, fltFontScale, intFontThickness) # call getTextSize
#
# # unpack roatated rect into center point, width and height, and angle
# ( (intPlateCenterX, intPlateCenterY), (intPlateWidth, intPlateHeight), fltCorrectionAngleInDeg ) = licPlate.rrLocationOfPlateInScene
#
# intPlateCenterX = int(intPlateCenterX) # make sure center is an integer
# intPlateCenterY = int(intPlateCenterY)
#
# ptCenterOfTextAreaX = int(intPlateCenterX) # the horizontal location of the text area is the same as the plate
#
# if intPlateCenterY < (sceneHeight * 0.75): # if the license plate is in the upper 3/4 of the image
# ptCenterOfTextAreaY = int(round(intPlateCenterY)) + int(round(plateHeight * 1.6)) # write the chars in below the plate
# else: # else if the license plate is in the lower 1/4 of the image
# ptCenterOfTextAreaY = int(round(intPlateCenterY)) - int(round(plateHeight * 1.6)) # write the chars in above the plate
# # end if
#
# textSizeWidth, textSizeHeight = textSize # unpack text size width and height
#
# ptLowerLeftTextOriginX = int(ptCenterOfTextAreaX - (textSizeWidth / 2)) # calculate the lower left origin of the text area
# ptLowerLeftTextOriginY = int(ptCenterOfTextAreaY + (textSizeHeight / 2)) # based on the text area center, width, and height
#
# # write the text on the image
# cv2.putText(imgOriginalScene, licPlate.strChars, (ptLowerLeftTextOriginX, ptLowerLeftTextOriginY), intFontFace, fltFontScale, SCALAR_YELLOW, intFontThickness)
## end function
###################################################################################################
if __name__ == "__main__":
main()