├── .gitignore ├── LICENSE ├── README.md ├── adet ├── __init__.py ├── checkpoint │ ├── __init__.py │ └── adet_checkpoint.py ├── config │ ├── __init__.py │ ├── config.py │ ├── defaults.py │ └── defaults_custom.py ├── data │ ├── __init__.py │ ├── augmentation.py │ ├── builtin.py │ ├── dataset_mapper.py │ ├── datasets │ │ └── text.py │ └── detection_utils.py ├── evaluation │ ├── __init__.py │ ├── rrc_evaluation_funcs.py │ ├── text_eval_script.py │ └── text_evaluation.py ├── layers │ ├── __init__.py │ ├── bezier_align.py │ ├── conv_with_kaiming_uniform.py │ ├── csrc │ │ ├── BezierAlign │ │ │ ├── BezierAlign.h │ │ │ ├── BezierAlign_cpu.cpp │ │ │ └── BezierAlign_cuda.cu │ │ ├── DefROIAlign │ │ │ ├── DefROIAlign.h │ │ │ └── DefROIAlign_cuda.cu │ │ ├── cuda_version.cu │ │ ├── ml_nms │ │ │ ├── ml_nms.cu │ │ │ └── ml_nms.h │ │ └── vision.cpp │ ├── def_roi_align.py │ ├── deform_conv.py │ ├── gcn.py │ ├── iou_loss.py │ ├── ml_nms.py │ └── naive_group_norm.py ├── modeling │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── bifpn.py │ │ ├── dla.py │ │ ├── fpn.py │ │ ├── lpf.py │ │ ├── mobilenet.py │ │ ├── resnet_interval.py │ │ ├── resnet_lpf.py │ │ └── vovnet.py │ └── sotr │ │ ├── __init__.py │ │ ├── loss.py │ │ ├── reversible.py │ │ ├── sotr.py │ │ ├── transformer.py │ │ └── utils.py ├── structures │ ├── __init__.py │ └── beziers.py └── utils │ ├── __init__.py │ ├── comm.py │ ├── measures.py │ └── visualizer.py ├── configs └── SOTR │ ├── Base-SOTR.yaml │ ├── R101.yaml │ ├── R50.yaml │ ├── R_101_DCN.yaml │ └── R_101_DCN_doc.yaml ├── docs ├── Makefile ├── conf.py ├── index.rst ├── modules │ ├── checkpoint.rst │ ├── config.rst │ ├── data.rst │ ├── index.rst │ ├── layers.rst │ ├── modeling.rst │ └── utils.rst └── requirements.txt ├── environment.yml ├── images ├── Qualitative_analysis.png ├── architecture.png ├── results.png └── transformers.png ├── setup.py └── tools ├── convert_fcos_weight.py ├── remove_optim_from_ckpt.py ├── train_net.py ├── train_net_custom.py ├── visualize_data.py └── visualize_publaynet.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/README.md -------------------------------------------------------------------------------- /adet/__init__.py: -------------------------------------------------------------------------------- 1 | from adet import modeling 2 | 3 | __version__ = "0.1.1" 4 | -------------------------------------------------------------------------------- /adet/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/checkpoint/__init__.py -------------------------------------------------------------------------------- /adet/checkpoint/adet_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/checkpoint/adet_checkpoint.py -------------------------------------------------------------------------------- /adet/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/config/__init__.py -------------------------------------------------------------------------------- /adet/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/config/config.py -------------------------------------------------------------------------------- /adet/config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/config/defaults.py -------------------------------------------------------------------------------- /adet/config/defaults_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/config/defaults_custom.py -------------------------------------------------------------------------------- /adet/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/data/__init__.py -------------------------------------------------------------------------------- /adet/data/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/data/augmentation.py -------------------------------------------------------------------------------- /adet/data/builtin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/data/builtin.py -------------------------------------------------------------------------------- /adet/data/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/data/dataset_mapper.py -------------------------------------------------------------------------------- /adet/data/datasets/text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/data/datasets/text.py -------------------------------------------------------------------------------- /adet/data/detection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/data/detection_utils.py -------------------------------------------------------------------------------- /adet/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/evaluation/__init__.py -------------------------------------------------------------------------------- /adet/evaluation/rrc_evaluation_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/evaluation/rrc_evaluation_funcs.py -------------------------------------------------------------------------------- /adet/evaluation/text_eval_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/evaluation/text_eval_script.py -------------------------------------------------------------------------------- /adet/evaluation/text_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/evaluation/text_evaluation.py -------------------------------------------------------------------------------- /adet/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/__init__.py -------------------------------------------------------------------------------- /adet/layers/bezier_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/bezier_align.py -------------------------------------------------------------------------------- /adet/layers/conv_with_kaiming_uniform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/conv_with_kaiming_uniform.py -------------------------------------------------------------------------------- /adet/layers/csrc/BezierAlign/BezierAlign.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/BezierAlign/BezierAlign.h -------------------------------------------------------------------------------- /adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp -------------------------------------------------------------------------------- /adet/layers/csrc/BezierAlign/BezierAlign_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/BezierAlign/BezierAlign_cuda.cu -------------------------------------------------------------------------------- /adet/layers/csrc/DefROIAlign/DefROIAlign.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/DefROIAlign/DefROIAlign.h -------------------------------------------------------------------------------- /adet/layers/csrc/DefROIAlign/DefROIAlign_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/DefROIAlign/DefROIAlign_cuda.cu -------------------------------------------------------------------------------- /adet/layers/csrc/cuda_version.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/cuda_version.cu -------------------------------------------------------------------------------- /adet/layers/csrc/ml_nms/ml_nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/ml_nms/ml_nms.cu -------------------------------------------------------------------------------- /adet/layers/csrc/ml_nms/ml_nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/ml_nms/ml_nms.h -------------------------------------------------------------------------------- /adet/layers/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/csrc/vision.cpp -------------------------------------------------------------------------------- /adet/layers/def_roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/def_roi_align.py -------------------------------------------------------------------------------- /adet/layers/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/deform_conv.py -------------------------------------------------------------------------------- /adet/layers/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/gcn.py -------------------------------------------------------------------------------- /adet/layers/iou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/iou_loss.py -------------------------------------------------------------------------------- /adet/layers/ml_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/ml_nms.py -------------------------------------------------------------------------------- /adet/layers/naive_group_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/layers/naive_group_norm.py -------------------------------------------------------------------------------- /adet/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/__init__.py -------------------------------------------------------------------------------- /adet/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/__init__.py -------------------------------------------------------------------------------- /adet/modeling/backbone/bifpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/bifpn.py -------------------------------------------------------------------------------- /adet/modeling/backbone/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/dla.py -------------------------------------------------------------------------------- /adet/modeling/backbone/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/fpn.py -------------------------------------------------------------------------------- /adet/modeling/backbone/lpf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/lpf.py -------------------------------------------------------------------------------- /adet/modeling/backbone/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/mobilenet.py -------------------------------------------------------------------------------- /adet/modeling/backbone/resnet_interval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/resnet_interval.py -------------------------------------------------------------------------------- /adet/modeling/backbone/resnet_lpf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/resnet_lpf.py -------------------------------------------------------------------------------- /adet/modeling/backbone/vovnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/backbone/vovnet.py -------------------------------------------------------------------------------- /adet/modeling/sotr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/sotr/__init__.py -------------------------------------------------------------------------------- /adet/modeling/sotr/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/sotr/loss.py -------------------------------------------------------------------------------- /adet/modeling/sotr/reversible.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/sotr/reversible.py -------------------------------------------------------------------------------- /adet/modeling/sotr/sotr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/sotr/sotr.py -------------------------------------------------------------------------------- /adet/modeling/sotr/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/sotr/transformer.py -------------------------------------------------------------------------------- /adet/modeling/sotr/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/modeling/sotr/utils.py -------------------------------------------------------------------------------- /adet/structures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/structures/__init__.py -------------------------------------------------------------------------------- /adet/structures/beziers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/structures/beziers.py -------------------------------------------------------------------------------- /adet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /adet/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/utils/comm.py -------------------------------------------------------------------------------- /adet/utils/measures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/utils/measures.py -------------------------------------------------------------------------------- /adet/utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/adet/utils/visualizer.py -------------------------------------------------------------------------------- /configs/SOTR/Base-SOTR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/configs/SOTR/Base-SOTR.yaml -------------------------------------------------------------------------------- /configs/SOTR/R101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/configs/SOTR/R101.yaml -------------------------------------------------------------------------------- /configs/SOTR/R50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/configs/SOTR/R50.yaml -------------------------------------------------------------------------------- /configs/SOTR/R_101_DCN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/configs/SOTR/R_101_DCN.yaml -------------------------------------------------------------------------------- /configs/SOTR/R_101_DCN_doc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/configs/SOTR/R_101_DCN_doc.yaml -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/modules/checkpoint.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/modules/checkpoint.rst -------------------------------------------------------------------------------- /docs/modules/config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/modules/config.rst -------------------------------------------------------------------------------- /docs/modules/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/modules/data.rst -------------------------------------------------------------------------------- /docs/modules/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/modules/index.rst -------------------------------------------------------------------------------- /docs/modules/layers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/modules/layers.rst -------------------------------------------------------------------------------- /docs/modules/modeling.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/modules/modeling.rst -------------------------------------------------------------------------------- /docs/modules/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/modules/utils.rst -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/environment.yml -------------------------------------------------------------------------------- /images/Qualitative_analysis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/images/Qualitative_analysis.png -------------------------------------------------------------------------------- /images/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/images/architecture.png -------------------------------------------------------------------------------- /images/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/images/results.png -------------------------------------------------------------------------------- /images/transformers.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/images/transformers.png -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/setup.py -------------------------------------------------------------------------------- /tools/convert_fcos_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/tools/convert_fcos_weight.py -------------------------------------------------------------------------------- /tools/remove_optim_from_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/tools/remove_optim_from_ckpt.py -------------------------------------------------------------------------------- /tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/tools/train_net.py -------------------------------------------------------------------------------- /tools/train_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/tools/train_net_custom.py -------------------------------------------------------------------------------- /tools/visualize_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/tools/visualize_data.py -------------------------------------------------------------------------------- /tools/visualize_publaynet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/biswassanket/DocSegTr/HEAD/tools/visualize_publaynet.py --------------------------------------------------------------------------------