mirror of https://github.com/open-mmlab/mmocr.git
85 lines
2.1 KiB
YAML
85 lines
2.1 KiB
YAML
Collections:
|
|
- Name: SATRN
|
|
Metadata:
|
|
Training Data: OCRDataset
|
|
Training Techniques:
|
|
- Adam
|
|
Training Resources: 8x Tesla V100
|
|
Epochs: 6
|
|
Batch Size: 512
|
|
Architecture:
|
|
- ShallowCNN
|
|
- SatrnEncoder
|
|
- TFDecoder
|
|
Paper: https://arxiv.org/pdf/1910.04396.pdf
|
|
README: configs/textrecog/satrn/README.md
|
|
|
|
Models:
|
|
- Name: satrn_academic
|
|
In Collection: SATRN
|
|
Config: configs/textrecog/satrn/satrn_academic.py
|
|
Metadata:
|
|
Training Data:
|
|
- SynthText
|
|
- Syn90k
|
|
Results:
|
|
- Task: Regular Text Recognition
|
|
Dataset: IIIT5K
|
|
Metrics:
|
|
word_acc: 96.1
|
|
- Task: Regular Text Recognition
|
|
Dataset: SVT
|
|
Metrics:
|
|
word_acc: 93.5
|
|
- Task: Regular Text Recognition
|
|
Dataset: ICDAR2013
|
|
Metrics:
|
|
word_acc: 95.7
|
|
- Task: Irregular Text Recognition
|
|
Dataset: ICDAR2015
|
|
Metrics:
|
|
word_acc: 84.1
|
|
- Task: Irregular Text Recognition
|
|
Dataset: SVTP
|
|
Metrics:
|
|
word_acc: 88.5
|
|
- Task: Irregular Text Recognition
|
|
Dataset: CT80
|
|
Metrics:
|
|
word_acc: 90.3
|
|
Weights: https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_academic_20210809-59c8c92d.pth
|
|
|
|
- Name: satrn_small
|
|
In Collection: SATRN
|
|
Config: configs/textrecog/satrn/satrn_small.py
|
|
Metadata:
|
|
Training Data:
|
|
- SynthText
|
|
- Syn90k
|
|
Results:
|
|
- Task: Regular Text Recognition
|
|
Dataset: IIIT5K
|
|
Metrics:
|
|
word_acc: 94.7
|
|
- Task: Regular Text Recognition
|
|
Dataset: SVT
|
|
Metrics:
|
|
word_acc: 91.3
|
|
- Task: Regular Text Recognition
|
|
Dataset: ICDAR2013
|
|
Metrics:
|
|
word_acc: 95.4
|
|
- Task: Irregular Text Recognition
|
|
Dataset: ICDAR2015
|
|
Metrics:
|
|
word_acc: 81.9
|
|
- Task: Irregular Text Recognition
|
|
Dataset: SVTP
|
|
Metrics:
|
|
word_acc: 85.9
|
|
- Task: Irregular Text Recognition
|
|
Dataset: CT80
|
|
Metrics:
|
|
word_acc: 86.5
|
|
Weights: https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_small_20210811-2badf6fc.pth
|