arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2108.06949cs.CV
收藏

场景文本识别中的数据增强

Data Augmentation for Scene Text Recognition

University of the Philippines(菲律宾大学)

Rowel Atienza

摘要:

由于自然场景中存在大量可能出现的文本,场景文本识别(STR)是计算机视觉中一项具有挑战性的任务。大多数STR模型依赖于合成数据集进行训练,因为没有足够大且公开可用的标记真实数据集。由于STR模型是使用真实数据进行评估的,因此训练和测试数据分布之间的不匹配导致模型性能不佳,特别是在受噪声,伪影,几何形状,结构等影响的具有挑战性的文本上。本文介绍了STRAug,它由36个为STR设计的图像增强函数组成。每个函数都模仿了自然场景中可以找到的某些文本图像属性,由相机传感器引起,或者由信号处理操作引起,但在训练数据集中表现不佳。当使用RandAugment应用于强基线模型时,STRAug显著提高了STR模型在常规和不规则测试数据集上的整体绝对准确度,在Rosetta上提高了2.10%,在R2AM上提高了1.48%,在CRNN上提高了1.30%,在RARE上提高了1.35%,在TRBA上提高了1.06%,在GCRNN上提高了0.89%。STRAg函数提供的API的多样性和简单性使STR的现有数据增强方法易于复制和验证。STRAg可在https://github.com/roatienza/straug上获得。

英文摘要:

Scene text recognition (STR) is a challenging task in computer vision due to the large number of possible text appearances in natural scenes. Most STR models rely on synthetic datasets for training since there are no sufficiently big and publicly available labelled real datasets. Since STR models are evaluated using real data, the mismatch between training and testing data distributions results into poor performance of models especially on challenging text that are affected by noise, artifacts, geometry, structure, etc. In this paper, we introduce STRAug which is made of 36 image augmentation functions designed for STR. Each function mimics certain text image properties that can be found in natural scenes, caused by camera sensors, or induced by signal processing operations but poorly represented in the training dataset. When applied to strong baseline models using RandAugment, STRAug significantly increases the overall absolute accuracy of STR models across regular and irregular test datasets by as much as 2.10% on Rosetta, 1.48% on R2AM, 1.30% on CRNN, 1.35% on RARE, 1.06% on TRBA and 0.89% on GCRNN. The diversity and simplicity of API provided by STRAug functions enable easy replication and validation of existing data augmentation methods for STR. STRAug is available at https://github.com/roatienza/straug.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑