/doc2graph

Doc2Graph transforms documents into graphs and exploit a GNN to solve several tasks.

Primary LanguagePythonMIT LicenseMIT

Doc2Graph

Installation

  • Python 3.10.10 (venv)
  • Ubuntu 22.04
  • ./install.sh
  • pip install -r requirements.txt
  • pip install -e .

Use

Initialization

  • Download datasets
python main.py --init

Citation

@inproceedings{gemelli2023doc2graph,
  title={Doc2Graph: A Task Agnostic Document Understanding Framework Based on Graph Neural Networks},
  author={Gemelli, Andrea and Biswas, Sanket and Civitelli, Enrico and Llad{\'o}s, Josep and Marinai, Simone},
  booktitle={Computer Vision--ECCV 2022 Workshops: Tel Aviv, Israel, October 23--27, 2022, Proceedings, Part IV},
  pages={329--344},
  year={2023},
  organization={Springer}
}