[Fix] Update checkpoint links for SATRN (#518)

open-mmlab · Oct 10, 2021 · abfe75d · abfe75d
1 parent 66ce0ef
commit abfe75d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/configs/textrecog/satrn/README.md b/configs/textrecog/satrn/README.md
@@ -37,5 +37,5 @@
 |                             Methods                             |        | Regular Text |      |     |      | Irregular Text |      |                                                                                               download                                                                                                |
 | :-------------------------------------------------------------: | :----: | :----------: | :--: | :-: | :--: | :------------: | :--: | :---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------: |
 |                                                                 | IIIT5K |     SVT      | IC13 |     | IC15 |      SVTP      | CT80 |
-| [Satrn](/configs/textrecog/satrn/satrn_academic.py) |  96.1  |     93.5     | 95.7 |     | 84.1 |      88.5      | 90.3 |      [model](https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_academic_20210809-59c8c92d.pth) \| [log](https://download.openmmlab.com/mmocr/textrecog/satrn/20210809_093244.log.json)      |
-| [Satrn_small](/configs/textrecog/satrn/satrn_small.py) |  94.7  |     91.3     | 95.4 |     | 81.9 |      85.9      | 86.5 |      [model](https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_small_20210811-2badf6fc.pth) \| [log](https://download.openmmlab.com/mmocr/textrecog/satrn/20210811_053047.log.json)      |
+| [Satrn](/configs/textrecog/satrn/satrn_academic.py) |  96.1  |     93.5     | 95.7 |     | 84.1 |      88.5      | 90.3 |      [model](https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_academic_20211009-cb8b1580.pth) \| [log](https://download.openmmlab.com/mmocr/textrecog/satrn/20210809_093244.log.json)      |
+| [Satrn_small](/configs/textrecog/satrn/satrn_small.py) |  94.7  |     91.3     | 95.4 |     | 81.9 |      85.9      | 86.5 |      [model](https://download.openmmlab.com/mmocr/textrecog/satrn/satrn_small_20211009-2cf13355.pth) \| [log](https://download.openmmlab.com/mmocr/textrecog/satrn/20210811_053047.log.json)      |
diff --git a/mmocr/utils/ocr.py b/mmocr/utils/ocr.py
@@ -281,11 +281,11 @@ def __init__(self,
             },
             'SATRN': {
                 'config': 'satrn/satrn_academic.py',
-                'ckpt': 'satrn/satrn_academic_20210809-59c8c92d.pth'
+                'ckpt': 'satrn/satrn_academic_20211009-cb8b1580.pth'
             },
             'SATRN_sm': {
                 'config': 'satrn/satrn_small.py',
-                'ckpt': 'satrn/satrn_small_20210811-2badf6fc.pth'
+                'ckpt': 'satrn/satrn_small_20211009-2cf13355.pth'
             },
             'SEG': {
                 'config': 'seg/seg_r31_1by16_fpnocr_academic.py',