mirror of https://github.com/open-mmlab/mmocr.git
40 lines
1.2 KiB
YAML
40 lines
1.2 KiB
YAML
Collections:
|
|
- Name: SDMGR
|
|
Metadata:
|
|
Training Data: KIEDataset
|
|
Training Techniques:
|
|
- Adam
|
|
Training Resources: 1x GeForce GTX 1080 Ti
|
|
Architecture:
|
|
- UNet
|
|
- SDMGRHead
|
|
Paper:
|
|
URL: https://arxiv.org/abs/2103.14470.pdf
|
|
Title: 'Spatial Dual-Modality Graph Reasoning for Key Information Extraction'
|
|
README: configs/kie/sdmgr/README.md
|
|
|
|
Models:
|
|
- Name: sdmgr_unet16_60e_wildreceipt
|
|
In Collection: SDMGR
|
|
Config: configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py
|
|
Metadata:
|
|
Training Data: wildreceipt
|
|
Results:
|
|
- 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
|
|
|
|
- Name: sdmgr_novisual_60e_wildreceipt
|
|
In Collection: SDMGR
|
|
Config: configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt.py
|
|
Metadata:
|
|
Training Data: wildreceipt
|
|
Results:
|
|
- 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
|