PlatePoa ALPR: OpenCV & PyTorch License Plate Extraction Pipeline

# PlatePoa AI Vehicle Plate Extractor
import cv2
import torch
import numpy as np

def extract_plate_number(image_path, model, ocr_reader):
    img = cv2.imread(image_path)
    results = model(img)
    
    for r in results:
        boxes = r.boxes.xyxy.cpu().numpy()
        for box in boxes:
            x1, y1, x2, y2 = map(int, box[:4])
            plate_crop = img[y1:y2, x1:x2]
            
            # Grayscale & Otsu Thresholding
            gray = cv2.cvtColor(plate_crop, cv2.COLOR_BGR2GRAY)
            ocr_res = ocr_reader.readtext(gray)
            
            if ocr_res:
                plate_text = "".join([t[1] for t in ocr_res]).replace(" ", "").upper()
                return {"plate": plate_text, "confidence": ocr_res[0][2]}
                
    return None