3 个月前

TrOCR:基于预训练模型的Transformer光学字符识别

TrOCR:基于预训练模型的Transformer光学字符识别

摘要

文本识别是文档数字化领域长期存在的研究难题。现有的方法通常基于卷积神经网络(CNN)进行图像理解,并结合循环神经网络(RNN)实现字符级文本生成。此外,为提升整体识别准确率,通常还需引入另一语言模型作为后处理步骤。本文提出一种端到端的文本识别方法——TrOCR,该方法结合预训练的图像Transformer与文本Transformer模型,利用Transformer架构同时完成图像理解与词片(wordpiece)级别的文本生成。TrOCR模型结构简洁但效果显著,可基于大规模合成数据进行预训练,并在人工标注数据集上进行微调。实验结果表明,TrOCR在印刷体、手写体及场景文本识别任务上均优于当前最先进的模型。TrOCR模型及相关代码已公开,获取地址为:\url{https://aka.ms/trocr}。

基准测试

基准方法指标
handwritten-text-recognition-on-iamTrOCR-small 62M
CER: 4.22
handwritten-text-recognition-on-iamTrOCR-large 558M
CER: 2.89
handwritten-text-recognition-on-iamTrOCR-base 334M
CER: 3.42
handwritten-text-recognition-on-iam-lineTrOCR
Test CER: 3.4
Test WER: -
handwritten-text-recognition-on-lam-lineTrOCR
Test CER: 3.6
Test WER: 11.6

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
TrOCR:基于预训练模型的Transformer光学字符识别 | 论文 | HyperAI超神经