
摘要
文本识别是文档数字化领域长期存在的研究难题。现有的方法通常基于卷积神经网络(CNN)进行图像理解,并结合循环神经网络(RNN)实现字符级文本生成。此外,为提升整体识别准确率,通常还需引入另一语言模型作为后处理步骤。本文提出一种端到端的文本识别方法——TrOCR,该方法结合预训练的图像Transformer与文本Transformer模型,利用Transformer架构同时完成图像理解与词片(wordpiece)级别的文本生成。TrOCR模型结构简洁但效果显著,可基于大规模合成数据进行预训练,并在人工标注数据集上进行微调。实验结果表明,TrOCR在印刷体、手写体及场景文本识别任务上均优于当前最先进的模型。TrOCR模型及相关代码已公开,获取地址为:\url{https://aka.ms/trocr}。
代码仓库
oleehyo/texteller
paddle
GitHub 中提及
pwc-1/Paper-10/tree/main/trocr
mindspore
huggingface/transformers
pytorch
GitHub 中提及
pwc-1/Paper-9/tree/main/1/trocr
mindspore
microsoft/unilm/tree/master/trocr
官方
pytorch
MindCode-4/code-5/tree/main/trocr
mindspore
d-gurgurov/im2latex
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| handwritten-text-recognition-on-iam | TrOCR-small 62M | CER: 4.22 |
| handwritten-text-recognition-on-iam | TrOCR-large 558M | CER: 2.89 |
| handwritten-text-recognition-on-iam | TrOCR-base 334M | CER: 3.42 |
| handwritten-text-recognition-on-iam-line | TrOCR | Test CER: 3.4 Test WER: - |
| handwritten-text-recognition-on-lam-line | TrOCR | Test CER: 3.6 Test WER: 11.6 |