HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Multi-Granularity Prediction for Scene Text Recognition

Peng Wang Cheng Da Cong Yao

Multi-Granularity Prediction for Scene Text Recognition

Abstract

Scene text recognition (STR) has been an active research topic in computer vision for years. To tackle this challenging problem, numerous innovative methods have been successively proposed and incorporating linguistic knowledge into STR models has recently become a prominent trend. In this work, we first draw inspiration from the recent progress in Vision Transformer (ViT) to construct a conceptually simple yet powerful vision STR model, which is built upon ViT and outperforms previous state-of-the-art models for scene text recognition, including both pure vision models and language-augmented methods. To integrate linguistic knowledge, we further propose a Multi-Granularity Prediction strategy to inject information from the language modality into the model in an implicit way, i.e. , subword representations (BPE and WordPiece) widely-used in NLP are introduced into the output space, in addition to the conventional character level representation, while no independent language model (LM) is adopted. The resultant algorithm (termed MGP-STR) is able to push the performance envelop of STR to an even higher level. Specifically, it achieves an average recognition accuracy of 93.35% on standard benchmarks. Code is available at https://github.com/AlibabaResearch/AdvancedLiterateMachinery/tree/main/OCR/MGP-STR.

Benchmarks

BenchmarkMethodologyMetrics
scene-text-recognition-on-coco-textMGP-STR
1:1 Accuracy: 81.7
scene-text-recognition-on-cute80MGP-STR
Accuracy: 99.31
scene-text-recognition-on-ic19-artMGP-STR
Accuracy (%): 85.5
scene-text-recognition-on-icdar2013MGP-STR
Accuracy: 98.5
scene-text-recognition-on-icdar2015MGP-STR
Accuracy: 90.9
scene-text-recognition-on-iiit5kMGP-STR
Accuracy: 98.8
scene-text-recognition-on-svtMGP-STR
Accuracy: 98.6
scene-text-recognition-on-svtpMGP-STR
Accuracy: 98.3
scene-text-recognition-on-uber-textMGP-STR
Accuracy (%): 91.0

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp