论文标题

学会增强:用于文本识别的联合数据增强和网络优化

Learn to Augment: Joint Data Augmentation and Network Optimization for Text Recognition

论文作者

Luo, Canjie, Zhu, Yuanzhi, Jin, Lianwen, Wang, Yongpan

论文摘要

手写的文本和场景文字遭受各种形状和扭曲的图案。因此,训练强大的识别模型需要大量数据以尽可能涵盖多样性。与数据收集和注释相反,数据增强是一种低成本的方式。在本文中,我们提出了一种新的文本图像增强方法。与传统的增强方法(例如旋转,缩放和透视转换)不同,我们提出的增强方法旨在学习适当有效的数据增强,这对于培训强大的识别器更有效,更有特殊。通过使用一组自定义基准点,提出的增强方法是灵活且可控的。此外,我们通过联合学习的数据增强过程和网络优化的孤立过程弥合了差距。代理网络从识别网络的输出中学习,并控制基准点,以生成识别网络的更合适的培训样本。对各种基准测试的广泛实验,包括常规场景文本,不规则场景文本和手写文本,表明拟议的增强和联合学习方法显着提高了识别网络的性能。提供了用于几何增强的一般工具包。

Handwritten text and scene text suffer from various shapes and distorted patterns. Thus training a robust recognition model requires a large amount of data to cover diversity as much as possible. In contrast to data collection and annotation, data augmentation is a low cost way. In this paper, we propose a new method for text image augmentation. Different from traditional augmentation methods such as rotation, scaling and perspective transformation, our proposed augmentation method is designed to learn proper and efficient data augmentation which is more effective and specific for training a robust recognizer. By using a set of custom fiducial points, the proposed augmentation method is flexible and controllable. Furthermore, we bridge the gap between the isolated processes of data augmentation and network optimization by joint learning. An agent network learns from the output of the recognition network and controls the fiducial points to generate more proper training samples for the recognition network. Extensive experiments on various benchmarks, including regular scene text, irregular scene text and handwritten text, show that the proposed augmentation and the joint learning methods significantly boost the performance of the recognition networks. A general toolkit for geometric augmentation is available.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源