mmocr/configs/kie/sdmgr
Tong Gao 34bd6b9bbb
[Docs] Docs for KIE CloseSet & OpenSet (#573)
* add kie docs

* fix paths

* sync cn docs

* fix layout error
2021-11-11 17:39:27 +08:00
..
README.md [Docs] Docs for KIE CloseSet & OpenSet (#573) 2021-11-11 17:39:27 +08:00
metafile.yml fix paper field in metafiles (#550) 2021-11-03 15:02:31 +08:00
sdmgr_novisual_60e_wildreceipt.py Fix kie_test_imgs script for novisual (#319) 2021-06-24 07:22:34 +00:00
sdmgr_novisual_60e_wildreceipt_openset.py [Feature] Support openset kie (#498) 2021-11-11 14:47:38 +08:00
sdmgr_unet16_60e_wildreceipt.py [Fix] Fix image export in test.py for KIE models (#486) 2021-09-18 18:24:55 +08:00

README.md

Spatial Dual-Modality Graph Reasoning for Key Information Extraction

Introduction

[ALGORITHM]

@misc{sun2021spatial,
      title={Spatial Dual-Modality Graph Reasoning for Key Information Extraction},
      author={Hongbin Sun and Zhanghui Kuang and Xiaoyu Yue and Chenhao Lin and Wayne Zhang},
      year={2021},
      eprint={2103.14470},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Results and models

WildReceipt

Method Modality Macro F1-Score Download
sdmgr_unet16 Visual + Textual 0.888 model | log
sdmgr_novisual Textual 0.870 model | log

WildReceiptOpenset

Method Modality Edge F1-Score Node Macro F1-Score Node Micro F1-Score Download
sdmgr_novisual Textual 0.786 0.926 0.935 model | log

:::{note}

  1. In the case of openset, the number of node categories is unknown or unfixed, and more node category can be added.
  2. To show that our method can handle openset problem, we modify the ground truth of WildReceipt to WildReceiptOpenset. The nodes are just classified into 4 classes: background, key, value, others, while adding edge labels for each box.
  3. The model is used to predict whether two nodes are a pair connecting by a valid edge.
  4. You can learn more about the key differences between CloseSet and OpenSet annotations in our tutorial. :::