K-OCR — OCR biển số xe Việt Nam

Dự án nhận dạng ký tự (OCR) trên ảnh biển số xe đã được crop sẵn, dùng kiến trúc CNN (stem + residual stages) + BiLSTM + CTC, định nghĩa tại plate_recognizer.py.

1. Cấu trúc dự án

K-OCR/
  plate_recognizer.py           # Kiến trúc model (CNN + BiLSTM + CTC)
  charset.py                    # Bảng ký tự + encode/decode CTC
  dataset.py                    # PlateDataset, augmentation, collate_fn cho CTC
  train.py                      # Script train model từ đầu
  tune.py                       # Script fine-tune / train tiếp từ 1 checkpoint có sẵn
  infer.py                      # Script chạy thử / demo inference
  export.py                     # Script export model sang ONNX / OpenVINO (FP32/FP16/INT8)
  generate_dummy_data.py        # Tạo dữ liệu giả lập để test nhanh pipeline
  requirements.txt
  data/
    README.md                  # Hướng dẫn định dạng dữ liệu
    train/{images/, labels.txt}
    val/{images/, labels.txt}
  checkpoints/                  # Nơi lưu model đã train (best_model.pt, last_model.pt)
  exported_models/              # Nơi lưu model đã export (onnx, openvino IR) — tạo bởi export.py

2. Kiến trúc model (tóm tắt)

Input [B, 3, 112, 224] → Stem → 3 Residual stage → AdaptiveAvgPool theo chiều cao → sequence [B, 56, 256] → 2 lớp BiLSTM → Linear classifier → logits [B, 56, num_classes]. Chi tiết đầy đủ xem docstring trong plate_recognizer.py.

num_classes = số ký tự trong charset.py + 1 (lớp blank của CTC).

3. Cài đặt

python -m venv venv
venv\Scripts\activate          # Windows
pip install -r requirements.txt

Nếu chỉ có CPU, có thể cài PyTorch bản CPU theo hướng dẫn tại https://pytorch.org/get-started/locally/ trước khi cài các phần còn lại.

4. Chuẩn bị dữ liệu

Xem chi tiết tại data/README.md.

Chưa có dữ liệu thật? Tạo dữ liệu giả lập để test pipeline trước:

python generate_dummy_data.py --data-dir data --train-samples 200 --val-samples 40

5. Train

python train.py --data-dir data

--epochs, --batch-size, --lr mặc định là tự động — nếu không truyền tay, train.py sẽ đếm số ảnh trong data/train/labels.txt và tự chọn theo bảng dưới (xem auto_hparams() trong train.py). Data ít cần train nhiều epoch hơn để hội tụ nhưng batch nhỏ hơn; d