Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Visual Semantics Allow for Textual Reasoning Better in Scene Text Recognition

About

Existing Scene Text Recognition (STR) methods typically use a language model to optimize the joint probability of the 1D character sequence predicted by a visual recognition (VR) model, which ignore the 2D spatial context of visual semantics within and between character instances, making them not generalize well to arbitrary shape scene text. To address this issue, we make the first attempt to perform textual reasoning based on visual semantics in this paper. Technically, given the character segmentation maps predicted by a VR model, we construct a subgraph for each instance, where nodes represent the pixels in it and edges are added between nodes based on their spatial similarity. Then, these subgraphs are sequentially connected by their root nodes and merged into a complete graph. Based on this graph, we devise a graph convolutional network for textual reasoning (GTR) by supervising it with a cross-entropy loss. GTR can be easily plugged in representative STR models to improve their performance owing to better textual reasoning. Specifically, we construct our model, namely S-GTR, by paralleling GTR to the language model in a segmentation-based STR baseline, which can effectively exploit the visual-linguistic complementarity via mutual learning. S-GTR sets new state-of-the-art on six challenging STR benchmarks and generalizes well to multi-linguistic datasets. Code is available at https://github.com/adeline-cs/GTR.

Yue He, Chen Chen, Jing Zhang, Juhua Liu, Fengxiang He, Chaoyue Wang, Bo Du• 2021

Related benchmarks

TaskDatasetResultRank
Scene Text RecognitionSVT (test)
Word Accuracy94.1
289
Scene Text RecognitionIIIT5K (test)
Word Accuracy95.8
244
Scene Text RecognitionIC15 (test)
Word Accuracy84.6
210
Scene Text RecognitionIC13 (test)
Word Accuracy96.8
207
Scene Text RecognitionSVTP (test)
Word Accuracy87.9
153
Scene Text RecognitionIC13, IC15, IIIT, SVT, SVTP, CUTE80 Average of 6 benchmarks (test)
Average Accuracy91.92
105
Scene Text RecognitionCUTE80 (test)
Accuracy0.923
87
Scene Text Recognition6 common benchmarks (test)
Word Accuracy (IIIT)95.8
57
Scene Text RecognitionIIIT5K 3000 (test)
Accuracy95.8
51
Scene Text RecognitionCUTE80
Accuracy92.3
47
Showing 10 of 29 rows

Other info

Code

Follow for update