Update app.py

app.py

@@ -38,43 +38,68 @@ if submit:
     frame = np.array(image)[...,::-1]
     try:
         plate_img = alpr(frame,license)
-        #plate_img = cv2.resize(plate_img,(200,50))
         results = model(plate_img*255)
-        #print(results.xyxy[0])
-        #Post-processing pre-requisite
-        decoder = results.pandas().xyxy[0].sort_values('xmin').iloc[:,0].values
-        compare = list(decoder[2:])
-        maks = None
-        for i in range(len(compare)):
-            if i == len(compare) - 1:
-                break
-            if maks == None:
-                maks = abs(compare[i] - compare[i + 1])
-            if abs(compare[i] - compare[i + 1]) > maks:
-                maks = abs(compare[i] - compare[i + 1])
-                w_index = (maks, i + 1)
-          
-        name = results.pandas().xyxy[0].sort_values('xmin').iloc[:, -1]
-        name = "".join([i for i in name])
-        if name not in counter and name != '':
-            counter[name] = 1
-        if name in counter and name !='':
-            counter[name] +=1
-        plate_name = list((sorted(counter.items(),key = lambda item:item[1])))[-1][0]
-        #Post-processing happens after here
-        mid_chars = str(plate_name[2:int(w_index[1] + 2)])  # assign this as old mid chars
-
-        if "6" in mid_chars:
-            mid_chars = mid_chars.replace("6", "G")  # assign this as new
-        if "1" in mid_chars:
-            mid_chars = mid_chars.replace("1", "I")
-        if "0" in mid_chars:
-            mid_chars = mid_chars.replace("0", "O")
-
-        new_plate_name = plate_name.replace(plate_name[2:int(w_index[1] + 2)], mid_chars)
+        control = max(results.pandas().xyxy[0].sort_values('ymin').iloc[:,1].values)
+        if control > 50:
+            name = results.pandas().xyxy[0].sort_values('ymin') #.iloc[:, -1] #ymin alwas bigger than 50 with bottom characters
+            ind = [ix for ix,i in enumerate(name.iloc[:,1]) if i>50][0]
+            upper_f_2 = name.iloc[:ind].sort_values("xmin").iloc[:,-1][:2]
+            upper_sort = name.iloc[:ind].sort_values("xmin").iloc[:,-1][2:] #add name column
+            bottom_sort = name.iloc[ind:].sort_values("xmin").iloc[:,-1]
+            upper_name =  "".join([i for i in upper_sort])
+            upper_f_name = "".join([i for i in upper_f_2])
+            bottom_name = "".join([i for i in bottom_sort])
+            if "1" in upper_name:
+                upper_name.replace("1","I")
+            if "6" in upper_name:
+                upper_name.replace("6","G")
+            if "0" in upper_name:
+                upper_name.replace("0","O")
+        
+            name =  upper_f_name + upper_name + bottom_name
+            if name not in counter and name != '':
+                counter[name] = 1
+            if name in counter and name != '':    
+                counter[name] += 1
+            plate_name = list((sorted(counter.items(), key=lambda item: item[1])))[-1][0]
+            st.write(plate_name)
+        
+        else:
+        
+            #Post-processing pre-requisite
+            decoder = results.pandas().xyxy[0].sort_values('xmin').iloc[:,0].values
+            compare = list(decoder[2:])
+            maks = None
+            for i in range(len(compare)):
+                if i == len(compare) - 1:
+                    break
+                if maks == None:
+                    maks = abs(compare[i] - compare[i + 1])
+                if abs(compare[i] - compare[i + 1]) > maks:
+                    maks = abs(compare[i] - compare[i + 1])
+                    w_index = (maks, i + 1)
+              
+            name = results.pandas().xyxy[0].sort_values('xmin').iloc[:, -1]
+            name = "".join([i for i in name])
+            if name not in counter and name != '':
+                counter[name] = 1
+            if name in counter and name !='':
+                counter[name] +=1
+            plate_name = list((sorted(counter.items(),key = lambda item:item[1])))[-1][0]
+            #Post-processing happens after here
+            mid_chars = str(plate_name[2:int(w_index[1] + 2)])  # assign this as old mid chars
+    
+            if "6" in mid_chars:
+                mid_chars = mid_chars.replace("6", "G")  # assign this as new
+            if "1" in mid_chars:
+                mid_chars = mid_chars.replace("1", "I")
+            if "0" in mid_chars:
+                mid_chars = mid_chars.replace("0", "O")
     
-        #cv2.imshow("Plate", plate_img)
-        st.write(new_plate_name)
+            new_plate_name = plate_name.replace(plate_name[2:int(w_index[1] + 2)], mid_chars)
+        
+            #cv2.imshow("Plate", plate_img)
+            st.write(new_plate_name)
     
     
     except Exception as e: