Tạo nhận diện biển số xe bằng YOLO + OpenCV

from ultralytics import YOLO
model = YOLO("best.pt")  # Đường dẫn tới model của bạn

Nhận diện biển số xe thời gian thực từ webcam

Để xử lý ảnh từ camera và vẽ khung nhận diện, bạn cần hiểu 2 điều:

• YOLO trả về danh sách các bounding box (vị trí), bạn cần vẽ hoặc crop để xử lý tiếp

• OpenCV sẽ hỗ trợ hiển thị, trích xuất và thao tác ảnh gốc từ camera hoặc video

Dưới đây là ví dụ code đầy đủ để nhận diện biển số từ webcam:

import cv2
from ultralytics import YOLO

model = YOLO("best.pt")  # Mô hình nhận diện biển số
cap = cv2.VideoCapture(0)  # Webcam

while True:
    ret, frame = cap.read()
    if not ret:
        break

    results = model(frame)
    boxes = results[0].boxes
    names = results[0].names

    for box in boxes:
        cls_id = int(box.cls[0])
        label = names[cls_id]
        x1, y1, x2, y2 = map(int, box.xyxy[0])

        # Vẽ khung nhận diện
        cv2.rectangle(frame, (x1, y1), (x2, y2), (0,255,0), 2)
        cv2.putText(frame, label, (x1, y1-10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0,255,0), 2)

        # Crop vùng biển số (có thể dùng OCR sau)
        plate_crop = frame[y1:y2, x1:x2]

    cv2.imshow("License Plate Detection", frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

Nhận diện nội dung biển số bằng OCR

Sau khi nhận diện được vùng biển số, bạn có thể trích xuất nội dung bằng pytesseract (OCR) hoặc easyocr.

Cài đặt pytesseract:

pip install pytesseract

Dùng OCR để đọc chữ từ vùng plate_crop:

import pytesseract

# Resize & chuyển ảnh về đen trắng nếu cần để tăng độ chính xác
gray = cv2.cvtColor(plate_crop, cv2.COLOR_BGR2GRAY)
text = pytesseract.image_to_string(gray, config='--psm 7')
print("Biển số xe:", text)