├── .github └── workflows │ ├── greetings.yml │ └── stale.yml ├── .gitignore ├── README.md ├── README_datasets.md ├── configs ├── activity.toml ├── bottle.toml ├── context.toml ├── test_base.toml ├── test_vit_knowbert_attention.toml ├── train_base.toml ├── train_knowbert_attention_activity.toml ├── train_knowbert_attention_bottle.toml └── train_knowbert_attention_context.toml ├── env.yml ├── exact_requirements.txt ├── main.py ├── model ├── cross_attention.py ├── heads.py └── vit_knowbert_interaction_timm.py ├── train_knowbert.py └── utils ├── __init__.py ├── config.py ├── dataset └── bottle_image_text_dataset.py ├── globals.py ├── lazy_list.py ├── log.py ├── save_n_load.py ├── seed.py └── set_parameters.py /.github/workflows/greetings.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/.github/workflows/greetings.yml -------------------------------------------------------------------------------- /.github/workflows/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/.github/workflows/stale.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/README.md -------------------------------------------------------------------------------- /README_datasets.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/README_datasets.md -------------------------------------------------------------------------------- /configs/activity.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/activity.toml -------------------------------------------------------------------------------- /configs/bottle.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/bottle.toml -------------------------------------------------------------------------------- /configs/context.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/context.toml -------------------------------------------------------------------------------- /configs/test_base.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/test_base.toml -------------------------------------------------------------------------------- /configs/test_vit_knowbert_attention.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/test_vit_knowbert_attention.toml -------------------------------------------------------------------------------- /configs/train_base.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/train_base.toml -------------------------------------------------------------------------------- /configs/train_knowbert_attention_activity.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/train_knowbert_attention_activity.toml -------------------------------------------------------------------------------- /configs/train_knowbert_attention_bottle.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/train_knowbert_attention_bottle.toml -------------------------------------------------------------------------------- /configs/train_knowbert_attention_context.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/configs/train_knowbert_attention_context.toml -------------------------------------------------------------------------------- /env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/env.yml -------------------------------------------------------------------------------- /exact_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/exact_requirements.txt -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/main.py -------------------------------------------------------------------------------- /model/cross_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/model/cross_attention.py -------------------------------------------------------------------------------- /model/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/model/heads.py -------------------------------------------------------------------------------- /model/vit_knowbert_interaction_timm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/model/vit_knowbert_interaction_timm.py -------------------------------------------------------------------------------- /train_knowbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/train_knowbert.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/dataset/bottle_image_text_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/utils/dataset/bottle_image_text_dataset.py -------------------------------------------------------------------------------- /utils/globals.py: -------------------------------------------------------------------------------- 1 | global_dict = {} 2 | -------------------------------------------------------------------------------- /utils/lazy_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/utils/lazy_list.py -------------------------------------------------------------------------------- /utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/utils/log.py -------------------------------------------------------------------------------- /utils/save_n_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/utils/save_n_load.py -------------------------------------------------------------------------------- /utils/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/utils/seed.py -------------------------------------------------------------------------------- /utils/set_parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MCLAB-OCR/KnowledgeMiningWithSceneText/HEAD/utils/set_parameters.py --------------------------------------------------------------------------------