SVTRv2让CTC模型在文字识别中超越编码器-解码器模型。
SVTRv2: CTC Beats Encoder-Decoder Models in Scene Text Recognition
- 用多尺寸重缩放和特征重排解决文本扭曲与对齐难题
- 引入语义引导模块提升语言上下文建模,准确率超越多数EDTRs
- 推理无额外开销,适合实时场景的高精度文字识别
基于连接时序分类(CTC)的场景文字识别方法(如SVTR)因结构简单、推理快,广泛应用于OCR。但其在处理文本不规则和语言缺失时准确率常低于编码器-解码器模型(EDTR)。为此,我们提出SVTRv2:首先采用多尺寸重缩放策略,将文本实例调整至合适尺寸,有效避免严重失真;同时引入特征重排模块,使视觉特征满足CTC对齐要求,缓解对齐难题。其次,设计语义引导模块,将语言上下文融入视觉特征,提升识别准确率,该模块可于推理阶段移除,不影响速度。我们在标准及最新挑战性基准上全面评估,与多种STR模型在不同文本不规则类型、语言、长文本及是否预训练等场景下公平比较。结果表明,SVTRv2在准确率和推理速度上均超越多数EDTR。代码开源:https://github.com/Topdu/OpenOCR。
原文摘要 · Abstract (English)
Connectionist temporal classification (CTC)-based scene text recognition (STR) methods, e.g., SVTR, are widely employed in OCR applications, mainly due to their simple architecture, which only contains a visual model and a CTC-aligned linear classifier, and therefore fast inference. However, they generally exhibit worse accuracy than encoder-decoder-based methods (EDTRs) due to struggling with text irregularity and linguistic missing. To address these challenges, we propose SVTRv2, a CTC model endowed with the ability to handle text irregularities and model linguistic context. First, a multi-size resizing strategy is proposed to resize text instances to appropriate predefined sizes, effectively avoiding severe text distortion. Meanwhile, we introduce a feature rearrangement module to ensure that visual features accommodate the requirement of CTC, thus alleviating the alignment puzzle. Second, we propose a semantic guidance module. It integrates linguistic context into the visual features, allowing CTC model to leverage language information for accuracy improvement. This module can be omitted at the inference stage and would not increase the time cost. We extensively evaluate SVTRv2 in both standard and recent challenging benchmarks, where SVTRv2 is fairly compared to popular STR models across multiple scenarios, including different types of text irregularity, languages, long text, and whether employing pretraining. SVTRv2 surpasses most EDTRs across the scenarios in terms of accuracy and inference speed. Code: https://github.com/Topdu/OpenOCR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。