├── .gitignore ├── data ├── EPHOIE_dataset.py ├── FUNSD_dataset.py └── SROIE_dataset.py ├── deployment ├── inference_EPHOIE.py ├── inference_SROIE.py ├── inference_preporcessing.py ├── main_EPHOIE.py ├── main_SROIE.py └── module_load.py ├── eval_EPHOIE.py ├── eval_FUNSD.py ├── eval_SROIE.py ├── example_config.yaml ├── model ├── BERTgrid_generator.py ├── ResNetFPN_ViBERTgrid.py ├── ViBERTgrid_net.py ├── crf.py ├── field_type_classification_head.py ├── grid_roi_align.py └── semantic_segmentation_head.py ├── pipeline ├── criteria.py ├── custom_loss.py ├── distributed_utils.py ├── ephoie_data_preprocessing.py ├── funsd_data_preprocessing.py ├── sroie_data_preprocessing.py ├── train_val_utils.py └── transform.py ├── readme.md ├── requirements.txt ├── train_EPHOIE.py ├── train_FUNSD.py ├── train_SROIE.py └── utils ├── ViBERTgrid_visualize.py ├── data_de_spilt.py ├── data_train_val_spilt.py └── see_modules.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /data/EPHOIE_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/data/EPHOIE_dataset.py -------------------------------------------------------------------------------- /data/FUNSD_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/data/FUNSD_dataset.py -------------------------------------------------------------------------------- /data/SROIE_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/data/SROIE_dataset.py -------------------------------------------------------------------------------- /deployment/inference_EPHOIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/deployment/inference_EPHOIE.py -------------------------------------------------------------------------------- /deployment/inference_SROIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/deployment/inference_SROIE.py -------------------------------------------------------------------------------- /deployment/inference_preporcessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/deployment/inference_preporcessing.py -------------------------------------------------------------------------------- /deployment/main_EPHOIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/deployment/main_EPHOIE.py -------------------------------------------------------------------------------- /deployment/main_SROIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/deployment/main_SROIE.py -------------------------------------------------------------------------------- /deployment/module_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/deployment/module_load.py -------------------------------------------------------------------------------- /eval_EPHOIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/eval_EPHOIE.py -------------------------------------------------------------------------------- /eval_FUNSD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/eval_FUNSD.py -------------------------------------------------------------------------------- /eval_SROIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/eval_SROIE.py -------------------------------------------------------------------------------- /example_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/example_config.yaml -------------------------------------------------------------------------------- /model/BERTgrid_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/model/BERTgrid_generator.py -------------------------------------------------------------------------------- /model/ResNetFPN_ViBERTgrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/model/ResNetFPN_ViBERTgrid.py -------------------------------------------------------------------------------- /model/ViBERTgrid_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/model/ViBERTgrid_net.py -------------------------------------------------------------------------------- /model/crf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/model/crf.py -------------------------------------------------------------------------------- /model/field_type_classification_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/model/field_type_classification_head.py -------------------------------------------------------------------------------- /model/grid_roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/model/grid_roi_align.py -------------------------------------------------------------------------------- /model/semantic_segmentation_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/model/semantic_segmentation_head.py -------------------------------------------------------------------------------- /pipeline/criteria.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/pipeline/criteria.py -------------------------------------------------------------------------------- /pipeline/custom_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/pipeline/custom_loss.py -------------------------------------------------------------------------------- /pipeline/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/pipeline/distributed_utils.py -------------------------------------------------------------------------------- /pipeline/ephoie_data_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/pipeline/ephoie_data_preprocessing.py -------------------------------------------------------------------------------- /pipeline/funsd_data_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/pipeline/funsd_data_preprocessing.py -------------------------------------------------------------------------------- /pipeline/sroie_data_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/pipeline/sroie_data_preprocessing.py -------------------------------------------------------------------------------- /pipeline/train_val_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/pipeline/train_val_utils.py -------------------------------------------------------------------------------- /pipeline/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/pipeline/transform.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_EPHOIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/train_EPHOIE.py -------------------------------------------------------------------------------- /train_FUNSD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/train_FUNSD.py -------------------------------------------------------------------------------- /train_SROIE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/train_SROIE.py -------------------------------------------------------------------------------- /utils/ViBERTgrid_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/utils/ViBERTgrid_visualize.py -------------------------------------------------------------------------------- /utils/data_de_spilt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/utils/data_de_spilt.py -------------------------------------------------------------------------------- /utils/data_train_val_spilt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/utils/data_train_val_spilt.py -------------------------------------------------------------------------------- /utils/see_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZeningLin/ViBERTgrid-PyTorch/HEAD/utils/see_modules.py --------------------------------------------------------------------------------