NRTR

NRTR: A No-Recurrence Sequence-to-Sequence Model For Scene Text Recognition

Abstract

Scene text recognition has attracted a great many researches due to its importance to various applications. Existing methods mainly adopt recurrence or convolution based networks. Though have obtained good performance, these methods still suffer from two limitations: slow training speed due to the internal recurrence of RNNs, and high complexity due to stacked convolutional layers for long-term feature extraction. This paper, for the first time, proposes a no-recurrence sequence-to-sequence text recognizer, named NRTR, that dispenses with recurrences and convolutions entirely. NRTR follows the encoder-decoder paradigm, where the encoder uses stacked self-attention to extract image features, and the decoder applies stacked self-attention to recognize texts based on encoder output. NRTR relies solely on self-attention mechanism thus could be trained with more parallelization and less complexity. Considering scene image has large variation in text and background, we further design a modality-transform block to effectively transform 2D input images to 1D sequences, combined with the encoder to extract more discriminative features. NRTR achieves state-of-the-art or highly competitive performance on both regular and irregular benchmarks, while requires only a small fraction of training time compared to the best model from the literature (at least 8 times faster).

Dataset

Train Dataset

trainset	instance_num	repeat_num	source
SynthText	7266686	1	synth
Syn90k	8919273	1	synth

Test Dataset

testset	instance_num	type
IIIT5K	3000	regular
SVT	647	regular
IC13	1015	regular
IC15	2077	irregular
SVTP	645	irregular
CT80	288	irregular

Results and Models

Methods	Backbone		Regular Text				Irregular Text		download
		IIIT5K	SVT	IC13-1015		IC15-2077	SVTP	CT80
NRTR	NRTRModalityTransform	0.9147	0.8841	0.9369		0.7246	0.7783	0.7500	model | log
NRTR-TTA	NRTRModalityTransform	0.9123	0.8825	0.9310		0.7492	0.7798	0.7535
NRTR	R31-1/8-1/4	0.9483	0.8918	0.9507		0.7578	0.8016	0.8889	model | log
NRTR-TTA	R31-1/8-1/4	0.9443	0.8903	0.9478		0.7790	0.8078	0.8854
NRTR	R31-1/16-1/8	0.9470	0.8918	0.9399		0.7376	0.7969	0.8854	model | log
NRTR-TTA	R31-1/16-1/8	0.9423	0.8903	0.9360		0.7641	0.8016	0.8854

Citation

@inproceedings{sheng2019nrtr,
  title={NRTR: A no-recurrence sequence-to-sequence model for scene text recognition},
  author={Sheng, Fenfen and Chen, Zhineng and Xu, Bo},
  booktitle={2019 International Conference on Document Analysis and Recognition (ICDAR)},
  pages={781--786},
  year={2019},
  organization={IEEE}
}