Collections:
- Name: SDMGR
  Metadata:
    Training Data: KIEDataset
    Training Techniques:
      - Adam
    Training Resources: 1x NVIDIA A100-SXM4-80GB
    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
    Alias: SDMGR
    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.890
    Weights: https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_unet16_60e_wildreceipt/sdmgr_unet16_60e_wildreceipt_20220825_151648-22419f37.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.873
    Weights: https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_novisual_60e_wildreceipt/sdmgr_novisual_60e_wildreceipt_20220831_193317-827649d8.pth
  - Name: sdmgr_novisual_60e_wildreceipt_openset
    In Collection: SDMGR
    Config: configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt-openset.py
    Metadata:
      Training Data: wildreceipt-openset
    Results:
      - Task: Key Information Extraction
        Dataset: wildreceipt
        Metrics:
          macro_f1: 0.931
          micro_f1: 0.940
          edge_micro_f1: 0.792
    Weights: https://download.openmmlab.com/mmocr/kie/sdmgr/sdmgr_novisual_60e_wildreceipt-openset/sdmgr_novisual_60e_wildreceipt-openset_20220831_200807-dedf15ec.pth
