Update app.py

app.py

@@ -41,6 +41,19 @@ if submit:
         #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 != '':
@@ -48,10 +61,20 @@ if submit:
         if name in counter and name !='':
             counter[name] +=1
         plate_name = list((sorted(counter.items(),key = lambda item:item[1])))[-1][0]
-        print(plate_name)
+        #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)
     
         #cv2.imshow("Plate", plate_img)
-        st.write(plate_name)
+        st.write(new_plate_name)
     
     
     except Exception as e: