File size: 2,382 Bytes
9bf4bd7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
88
89
Collections:
- Name: SATRN
  Metadata:
    Training Data: OCRDataset
    Training Techniques:
      - Adam
    Training Resources: 4x Tesla A100
    Epochs: 6
    Batch Size: 512
    Architecture:
      - ShallowCNN
      - SatrnEncoder
      - NRTRDecoder
  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_shallow_5e_st_mj
    Alias: SATRN
    In Collection: SATRN
    Config: configs/textrecog/satrn/satrn_shallow_5e_st_mj.py
    Metadata:
      Training Data:
        - SynthText
        - Syn90k
    Results:
      - Task: Text Recognition
        Dataset: IIIT5K
        Metrics:
          word_acc: 0.9600
      - Task: Text Recognition
        Dataset: SVT
        Metrics:
          word_acc: 0.9196
      - Task: Text Recognition
        Dataset: ICDAR2013
        Metrics:
          word_acc: 0.9606
      - Task: Text Recognition
        Dataset: ICDAR2015
        Metrics:
          word_acc: 0.8031
      - Task: Text Recognition
        Dataset: SVTP
        Metrics:
          word_acc: 0.8837
      - Task: Text Recognition
        Dataset: CT80
        Metrics:
          word_acc: 0.8993
    Weights: https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_shallow_5e_st_mj/satrn_shallow_5e_st_mj_20220915_152443-5fd04a4c.pth

  - Name: satrn_shallow-small_5e_st_mj
    Alias: SATRN_sm
    In Collection: SATRN
    Config: configs/textrecog/satrn/satrn_shallow-small_5e_st_mj.py
    Metadata:
      Training Data:
        - SynthText
        - Syn90k
    Results:
      - Task: Text Recognition
        Dataset: IIIT5K
        Metrics:
          word_acc: 0.9423
      - Task: Text Recognition
        Dataset: SVT
        Metrics:
          word_acc: 0.8995
      - Task: Text Recognition
        Dataset: ICDAR2013
        Metrics:
          word_acc: 0.9567
      - Task: Text Recognition
        Dataset: ICDAR2015
        Metrics:
          word_acc: 0.7877
      - Task: Text Recognition
        Dataset: SVTP
        Metrics:
          word_acc: 0.8574
      - Task: Text Recognition
        Dataset: CT80
        Metrics:
          word_acc: 0.8507
    Weights: https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_shallow-small_5e_st_mj/satrn_shallow-small_5e_st_mj_20220915_152442-5591bf27.pth