Update app.py

app.py

@@ -30,6 +30,35 @@ def unsharp_mask(image, kernel_size=(5, 5), sigma=1.0, amount=1.0, threshold=0):
         np.copyto(sharpened, image, where=low_contrast_mask)
     return sharpened
 
+def convdigplate(text) :
+  dict = {'0':'O','1':'I','2':'S','3':'E','4':'A','5':'S','6':'B','7':'T','8':'B','9':'B'}
+  
+  dictL = {'A':'4','B':'8','C':'0','D':'0','E':'3','F':'3','G':'6','H':'4','I':'1',
+           'J':'6','K':'4','L':'1','M':'4','N':'4','O':'0','P':'8','Q':'0','R':'8',
+           'S':'2','T':'1','U':'0','V':'4','X':'4','Y':'9','W':'3','Z':'2'}
+  
+  if len(text) > 7: 
+    if len(text) >= 9: 
+      text = text[1:8]
+    else :
+      if text[3].isdigit() :
+        text = text[:7]
+      else :
+        text = text[1:]
+   
+  temp = list(text) 
+  
+  for index in range(len(temp)):
+    if index <3:
+      if text[index].isdigit():
+        temp[index] = dict[temp[index]]
+    else :
+       if not text[index].isdigit() and index <> 4:
+        temp[index] = dictL[temp[index]]
+        
+  text = "".join(temp)
+  return text
+    
 def increase_brightness(img, value):
     hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
     h, s, v = cv2.split(hsv)
@@ -66,15 +95,8 @@ def execute_text_recognition_tflite( boxes, frame, interpreter, input_details, o
     if not len(text): return 
     license_plate = text
     text[:3].replace("0",'O')
-
-    if len(text) > 7: 
-      if len(text) >= 9: 
-        text = text[1:8]
-      else :
-        if text[3].isdigit() :
-          text = text[:7]
-        else :
-          text = text[1:]
+    
+    text = convdigplate(text)
 
     return text,cv2.resize(save_frame,(94,24))