File size: 2,178 Bytes
9c06ac7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
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:
    URL: https://arxiv.org/pdf/1910.04396.pdf
    Title: 'On Recognizing Texts of Arbitrary Shapes with 2D Self-Attention'
  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: Text Recognition
        Dataset: IIIT5K
        Metrics:
          word_acc: 96.1
      - Task: Text Recognition
        Dataset: SVT
        Metrics:
          word_acc: 93.5
      - Task: Text Recognition
        Dataset: ICDAR2013
        Metrics:
          word_acc: 95.7
      - Task: Text Recognition
        Dataset: ICDAR2015
        Metrics:
          word_acc: 84.1
      - Task: Text Recognition
        Dataset: SVTP
        Metrics:
          word_acc: 88.5
      - Task: Text Recognition
        Dataset: CT80
        Metrics:
          word_acc: 90.3
    Weights: https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_academic_20211009-cb8b1580.pth

  - Name: satrn_small
    In Collection: SATRN
    Config: configs/textrecog/satrn/satrn_small.py
    Metadata:
      Training Data:
        - SynthText
        - Syn90k
    Results:
      - Task: Text Recognition
        Dataset: IIIT5K
        Metrics:
          word_acc: 94.7
      - Task: Text Recognition
        Dataset: SVT
        Metrics:
          word_acc: 91.3
      - Task: Text Recognition
        Dataset: ICDAR2013
        Metrics:
          word_acc: 95.4
      - Task: Text Recognition
        Dataset: ICDAR2015
        Metrics:
          word_acc: 81.9
      - Task: Text Recognition
        Dataset: SVTP
        Metrics:
          word_acc: 85.9
      - Task: Text Recognition
        Dataset: CT80
        Metrics:
          word_acc: 86.5
    Weights: https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_small_20211009-2cf13355.pth