PaddleOCR engine feature + tesseract-ocr pacakage fix

app.py

@@ -16,16 +16,18 @@ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 import torch
 from datetime import datetime
 import time
+from paddleocr import PaddleOCR
 
 # Initialisation of models
 def load_models():
-    global model_vehicle, model_plate, reader_easyocr, pipeline_kerasocr, processor_trocr, model_trocr
+    global model_vehicle, model_plate, reader_easyocr, pipeline_kerasocr, processor_trocr, model_trocr, ocr_paddle
     model_vehicle = YOLO('models/yolov8n.pt')
     model_plate = YOLO('models/best.pt')
     reader_easyocr = easyocr.Reader(['en'], gpu=False)
     pipeline_kerasocr = keras_ocr.pipeline.Pipeline()
     processor_trocr = TrOCRProcessor.from_pretrained('microsoft/trocr-base-handwritten')
     model_trocr = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-handwritten')
+    ocr_paddle = PaddleOCR(use_angle_cls=True, lang='en', use_gpu=False)
 load_models()
 
 # patterns plate layouts europe
@@ -81,6 +83,11 @@ def read_license_plate(license_plate_crop, ocr_engine='easyocr'):
         text_preprocessed = trocr_ocr(preprocess_image(license_plate_crop))
         detections_raw = [(None, text_raw.strip(), None)]
         detections_preprocessed = [(None, text_preprocessed.strip(), None)]
+    elif ocr_engine == 'paddleocr':
+        result_raw = ocr_paddle.ocr(license_plate_crop)
+        result_preprocessed = ocr_paddle.ocr(preprocessed_image)
+        detections_raw = [(None, result_raw[0][0][1][0], result_raw[0][0][1][1] if result_raw[0] else 0.0)]
+        detections_preprocessed = [(None, result_preprocessed[0][0][1][0], result_preprocessed[0][0][1][1] if result_preprocessed[0] else 0.0)]
     else:
         raise ValueError(f"OCR engine '{ocr_engine}' not supported.")
     
@@ -297,7 +304,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             with gr.Row():
                 with gr.Column(scale=1):
                     input_image = gr.Image(type="numpy", label="Input image")
-                    ocr_selector = gr.Radio(choices=['easyocr', 'pytesseract', 'kerasocr', 'trocr'], value='easyocr', label="Select OCR Engine")
+                    ocr_selector = gr.Radio(choices=['easyocr', 'paddleocr' 'pytesseract', 'kerasocr', 'trocr'], value='easyocr', label="Select OCR Engine")
                     confidence_slider = gr.Slider(minimum=0, maximum=1, value=0.5, step=0.01, label="Detection Confidence Threshold")
                     submit_btn = gr.Button("Detect License Plates", variant="primary")
                 

packages.txt

@@ -0,0 +1 @@
+tesseract-ocr

requirements.txt

@@ -10,4 +10,6 @@ pytesseract==0.3.10
 torch==2.4.0
 transformers==4.44.0
 ultralytics==8.2.74
-tensorflow==2.14.0
+tensorflow==2.14.0
+paddleocr
+paddlepaddle