├── .flake8 ├── .gitignore ├── FUNSDLines_anchors_noRot_25.json ├── FUNSD_example.png ├── FUNSD_extract_lines.py ├── FUNSD_train_valid_test_split.json ├── LICENSE ├── NAF_example.png ├── README.md ├── anchors_noRot_new_25.json ├── base ├── __init__.py ├── base_data_loader.py ├── base_model.py └── base_trainer.py ├── configs ├── cf_FUNSDLinesAndWords_pair_graph663rv_new.json ├── cf_FUNSDLines_detect_augR_staggerLighter.json ├── cf_FUNSDLines_pair_binary333rv_new.json ├── cf_FUNSDLines_pair_binary33rv_ablate.json ├── cf_FUNSDLines_pair_binary33rv_ablate_new.json ├── cf_FUNSDLines_pair_binary3rv_ablate.json ├── cf_FUNSDLines_pair_binary3rv_ablate_new.json ├── cf_FUNSDLines_pair_graph222rv_ablate.json ├── cf_FUNSDLines_pair_graph663rv_new.json ├── cf_FUNSDLines_pair_graph663rv_noEdgeViz_new.json ├── cf_FUNSDLines_pair_graph77rv_ablate.json ├── cf_FUNSDLines_pair_graph9rv_ablate.json ├── cf_FUNSDWords_pair_graph663rv_new.json ├── cf_NAF_detect_augR_staggerLighter.json ├── cf_NAF_pair_binary333rv_new.json └── cf_NAF_pair_graph663rv_new.json ├── data_loader ├── __init__.py └── data_loaders.py ├── datasets ├── box_detect.py ├── forms_box_detect.py ├── forms_graph_pair.py ├── funsd_box_detect.py ├── funsd_graph_pair.py ├── graph_pair.py ├── testforms_box.py ├── testforms_graph_pair.py ├── testfunsd_box_detect.py └── testfunsd_graph_pair.py ├── do_update_bn.py ├── eval.py ├── evaluators ├── __init__.py ├── draw_graph.py ├── funsdboxdetect_eval.py ├── funsdgraphpair_eval.py └── othergraphpair_eval.py ├── graph.py ├── logger ├── __init__.py └── logger.py ├── logo.png ├── model ├── __init__.py ├── attention.py ├── binary_pair_real.py ├── coordconv.py ├── fudge.py ├── loss.py ├── meta_graph_net.py ├── net_builder.py ├── old_binary_pair_real.py ├── optimize.py ├── pairing_graph.py ├── yolo_box_detector.py └── yolo_loss.py ├── run.py ├── train.py ├── trainer ├── __init__.py ├── box_detect_trainer.py └── graph_pair_trainer.py └── utils ├── __init__.py ├── augmentation.py ├── crop_transform.py ├── forms_annotations.py ├── funsd_annotations.py ├── group_pairing.py ├── img_f.py ├── util.py └── yolo_tools.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | .idea/ 3 | saved/ 4 | data/ 5 | -------------------------------------------------------------------------------- /FUNSDLines_anchors_noRot_25.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/FUNSDLines_anchors_noRot_25.json -------------------------------------------------------------------------------- /FUNSD_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/FUNSD_example.png -------------------------------------------------------------------------------- /FUNSD_extract_lines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/FUNSD_extract_lines.py -------------------------------------------------------------------------------- /FUNSD_train_valid_test_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/FUNSD_train_valid_test_split.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/LICENSE -------------------------------------------------------------------------------- /NAF_example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/NAF_example.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/README.md -------------------------------------------------------------------------------- /anchors_noRot_new_25.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/anchors_noRot_new_25.json -------------------------------------------------------------------------------- /base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/base/__init__.py -------------------------------------------------------------------------------- /base/base_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/base/base_data_loader.py -------------------------------------------------------------------------------- /base/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/base/base_model.py -------------------------------------------------------------------------------- /base/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/base/base_trainer.py -------------------------------------------------------------------------------- /configs/cf_FUNSDLinesAndWords_pair_graph663rv_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLinesAndWords_pair_graph663rv_new.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_detect_augR_staggerLighter.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_detect_augR_staggerLighter.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_binary333rv_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_binary333rv_new.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_binary33rv_ablate.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_binary33rv_ablate.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_binary33rv_ablate_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_binary33rv_ablate_new.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_binary3rv_ablate.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_binary3rv_ablate.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_binary3rv_ablate_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_binary3rv_ablate_new.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_graph222rv_ablate.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_graph222rv_ablate.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_graph663rv_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_graph663rv_new.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_graph663rv_noEdgeViz_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_graph663rv_noEdgeViz_new.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_graph77rv_ablate.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_graph77rv_ablate.json -------------------------------------------------------------------------------- /configs/cf_FUNSDLines_pair_graph9rv_ablate.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDLines_pair_graph9rv_ablate.json -------------------------------------------------------------------------------- /configs/cf_FUNSDWords_pair_graph663rv_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_FUNSDWords_pair_graph663rv_new.json -------------------------------------------------------------------------------- /configs/cf_NAF_detect_augR_staggerLighter.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_NAF_detect_augR_staggerLighter.json -------------------------------------------------------------------------------- /configs/cf_NAF_pair_binary333rv_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_NAF_pair_binary333rv_new.json -------------------------------------------------------------------------------- /configs/cf_NAF_pair_graph663rv_new.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/configs/cf_NAF_pair_graph663rv_new.json -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- 1 | from .data_loaders import * 2 | -------------------------------------------------------------------------------- /data_loader/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/data_loader/data_loaders.py -------------------------------------------------------------------------------- /datasets/box_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/box_detect.py -------------------------------------------------------------------------------- /datasets/forms_box_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/forms_box_detect.py -------------------------------------------------------------------------------- /datasets/forms_graph_pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/forms_graph_pair.py -------------------------------------------------------------------------------- /datasets/funsd_box_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/funsd_box_detect.py -------------------------------------------------------------------------------- /datasets/funsd_graph_pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/funsd_graph_pair.py -------------------------------------------------------------------------------- /datasets/graph_pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/graph_pair.py -------------------------------------------------------------------------------- /datasets/testforms_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/testforms_box.py -------------------------------------------------------------------------------- /datasets/testforms_graph_pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/testforms_graph_pair.py -------------------------------------------------------------------------------- /datasets/testfunsd_box_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/testfunsd_box_detect.py -------------------------------------------------------------------------------- /datasets/testfunsd_graph_pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/datasets/testfunsd_graph_pair.py -------------------------------------------------------------------------------- /do_update_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/do_update_bn.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/eval.py -------------------------------------------------------------------------------- /evaluators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/evaluators/__init__.py -------------------------------------------------------------------------------- /evaluators/draw_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/evaluators/draw_graph.py -------------------------------------------------------------------------------- /evaluators/funsdboxdetect_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/evaluators/funsdboxdetect_eval.py -------------------------------------------------------------------------------- /evaluators/funsdgraphpair_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/evaluators/funsdgraphpair_eval.py -------------------------------------------------------------------------------- /evaluators/othergraphpair_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/evaluators/othergraphpair_eval.py -------------------------------------------------------------------------------- /graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/graph.py -------------------------------------------------------------------------------- /logger/__init__.py: -------------------------------------------------------------------------------- 1 | from .logger import * 2 | -------------------------------------------------------------------------------- /logger/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/logger/logger.py -------------------------------------------------------------------------------- /logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/logo.png -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/attention.py -------------------------------------------------------------------------------- /model/binary_pair_real.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/binary_pair_real.py -------------------------------------------------------------------------------- /model/coordconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/coordconv.py -------------------------------------------------------------------------------- /model/fudge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/fudge.py -------------------------------------------------------------------------------- /model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/loss.py -------------------------------------------------------------------------------- /model/meta_graph_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/meta_graph_net.py -------------------------------------------------------------------------------- /model/net_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/net_builder.py -------------------------------------------------------------------------------- /model/old_binary_pair_real.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/old_binary_pair_real.py -------------------------------------------------------------------------------- /model/optimize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/optimize.py -------------------------------------------------------------------------------- /model/pairing_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/pairing_graph.py -------------------------------------------------------------------------------- /model/yolo_box_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/yolo_box_detector.py -------------------------------------------------------------------------------- /model/yolo_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/model/yolo_loss.py -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/run.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/train.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/trainer/__init__.py -------------------------------------------------------------------------------- /trainer/box_detect_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/trainer/box_detect_trainer.py -------------------------------------------------------------------------------- /trainer/graph_pair_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/trainer/graph_pair_trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/utils/augmentation.py -------------------------------------------------------------------------------- /utils/crop_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/utils/crop_transform.py -------------------------------------------------------------------------------- /utils/forms_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/utils/forms_annotations.py -------------------------------------------------------------------------------- /utils/funsd_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/utils/funsd_annotations.py -------------------------------------------------------------------------------- /utils/group_pairing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/utils/group_pairing.py -------------------------------------------------------------------------------- /utils/img_f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/utils/img_f.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/utils/util.py -------------------------------------------------------------------------------- /utils/yolo_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/herobd/FUDGE/HEAD/utils/yolo_tools.py --------------------------------------------------------------------------------