mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Enhancement] Update links to SDMGR (#1252) 

* [Enhancement] Update links to SDMGR links

* fix tests
pull/1255/head
Tong Gao 2022-08-03 18:53:05 +08:00 committed by GitHub
parent b1d1c7f401
commit 4ffbb23623
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
configs/kie/sdmgr/README.md
View File

@ -18,8 +18,8 @@ Key information extraction from document images is of paramount importance in of
| Method | Modality | Macro F1-Score | Download |
| :--------------------------------------------------------------------: | :--------------: | :------------: | :--------------------------------------------------------------------------------------------------: |
| [sdmgr_unet16](/configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py) | Visual + Textual | 0.888 | [model](https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_unet16_60e_wildreceipt_20210520-7489e6de.pth) \| [log](https://download.openmmlab.com/mmocr/kie/sdmgr/20210520_132236.log.json) |
| [sdmgr_novisual](/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt.py) | Textual | 0.870 | [model](https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_novisual_60e_wildreceipt_20210517-a44850da.pth) \| [log](https://download.openmmlab.com/mmocr/kie/sdmgr/20210517_205829.log.json) |
| [sdmgr_unet16](/configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py) | Visual + Textual | 0.888 | [model](https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_unet16_60e_wildreceipt_20220706-57c220a6.pth) \| [log](https://download.openmmlab.com/mmocr/kie/sdmgr/20220706_220455.log.json) |
| [sdmgr_novisual](/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt.py) | Textual | 0.871 | [model](https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_novisual_60e_wildreceipt_20220803-d06d4a1a.pth) \| [log](https://download.openmmlab.com/mmocr/kie/sdmgr/20220803_010223.log.json) |
```{note}
1. For `sdmgr_novisual`, images are not needed for training and testing. So fake `img_prefix` can be used in configs. As well, fake `file_name` can be used in annotation files.
@ -29,7 +29,7 @@ Key information extraction from document images is of paramount importance in of
| Method | Modality | Edge F1-Score | Node Macro F1-Score | Node Micro F1-Score | Download |
| :-------------------------------------------------------------------: | :------: | :-----------: | :-----------------: | :-----------------: | :----------------------------------------------------------------------: |
| [sdmgr_novisual](/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt_openset.py) | Textual | 0.786 | 0.926 | 0.935 | [model](https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_novisual_60e_wildreceipt_openset_20210917-d236b3ea.pth) \| [log](https://download.openmmlab.com/mmocr/kie/sdmgr/20210917_050824.log.json) |
| [sdmgr_novisual](/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt_openset.py) | Textual | 0.796 | 0.930 | 0.938 | [model](https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_novisual_60e_wildreceipt_openset_20220803-427ec7a8.pth) \| [log](https://download.openmmlab.com/mmocr/kie/sdmgr/20220803_010235.log.json) |
```{note}
1. In the case of openset, the number of node categories is unknown or unfixed, and more node category can be added.

8
configs/kie/sdmgr/metafile.yml
View File

@ -23,8 +23,8 @@ Models:
- Task: Key Information Extraction
Dataset: wildreceipt
Metrics:
macro_f1: 0.876
Weights: https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_unet16_60e_wildreceipt_20210405-16a47642.pth
macro_f1: 0.888
Weights: https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_unet16_60e_wildreceipt_20220706-57c220a6.pth
- Name: sdmgr_novisual_60e_wildreceipt
In Collection: SDMGR
@ -35,5 +35,5 @@ Models:
- Task: Key Information Extraction
Dataset: wildreceipt
Metrics:
macro_f1: 0.864
Weights: https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_novisual_60e_wildreceipt_20210405-07bc26ad.pth
macro_f1: 0.871
Weights: https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_novisual_60e_wildreceipt_20220803-d06d4a1a.pth

2
mmocr/utils/ocr.py
View File

@ -346,7 +346,7 @@ class MMOCR:
'SDMGR': {
'config': 'sdmgr/sdmgr_unet16_60e_wildreceipt.py',
'ckpt':
'sdmgr/sdmgr_unet16_60e_wildreceipt_20210520-7489e6de.pth'
'sdmgr/sdmgr_unet16_60e_wildreceipt_20220706-57c220a6.pth'
}
}

6
tests/test_utils/test_ocr.py
View File

@ -57,7 +57,7 @@ cfg_default_prefix = os.path.join(str(Path.cwd()), 'configs/')
'https://download.openmmlab.com/mmocr/textrecog/'
'crnn/crnn_academic-a723a1c5.pth',
'https://download.openmmlab.com/mmocr/kie/'
'sdmgr/sdmgr_unet16_60e_wildreceipt_20210520-7489e6de.pth'
'sdmgr/sdmgr_unet16_60e_wildreceipt_20220706-57c220a6.pth'
]),
('DB_r18', 'CRNN', 'SDMGR', 'test/', [
'test/textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py',
@ -69,7 +69,7 @@ cfg_default_prefix = os.path.join(str(Path.cwd()), 'configs/')
'https://download.openmmlab.com/mmocr/textrecog/'
'crnn/crnn_academic-a723a1c5.pth',
'https://download.openmmlab.com/mmocr/kie/'
'sdmgr/sdmgr_unet16_60e_wildreceipt_20210520-7489e6de.pth'
'sdmgr/sdmgr_unet16_60e_wildreceipt_20220706-57c220a6.pth'
])],
)
@mock.patch('mmocr.utils.ocr.init_detector')
@ -125,7 +125,7 @@ def test_ocr_init(mock_loading, mock_config, mock_build_detector,
'https://download.openmmlab.com/mmocr/textrecog/'
'crnn/crnn_academic-a723a1c5.pth',
'https://download.openmmlab.com/mmocr/kie/'
'sdmgr/sdmgr_unet16_60e_wildreceipt_20210520-7489e6de.pth'
'sdmgr/sdmgr_unet16_60e_wildreceipt_20220706-57c220a6.pth'
]),
('DB_r18', '', 'test.ckpt', 'CRNN', '', 'test.ckpt', 'SDMGR', '',
'test.ckpt', '', [