Introduction

lightningOCR is an open-source toolbox based on PyTorch and Lightning-AI for text recognition.

Installation

TODO:

Examples

Example	Description
Create_dataset	Create a primitive dataset from scratch
Train ABINet Vision	Train the ABINet Vision model on the created dataset
Export ONNX	Export the model to onnx format

Model Zoo

• ABINet (CVPR 2021) from MMOCR
• MGP-STR (ECCV 2022) from transformers
• TrOCR from transformers

Citation

@article{lightningOCR,
  title   = {{lightningOCR}: Open-source toolbox based on PyTorch and Lightning-AI for text recognition},
  author  = {MiXaiLL76},
  year    = {2024}
}