├── .github └── workflows │ └── pytest.yml ├── .gitignore ├── README.md ├── assets ├── 18.png ├── 19.png ├── 19_mae.png ├── 19_masked.jpg └── 2.png ├── config ├── config.json ├── docaligner.gin ├── doctr.json ├── doctr_plus.json └── finetune.json ├── demo ├── background_segmentation.ipynb └── doc3d_dataloader.ipynb ├── docmae ├── __init__.py ├── data │ ├── __init__.py │ ├── augmentation │ │ ├── __init__.py │ │ ├── random_resized_crop.py │ │ └── replace_background.py │ ├── doc3d.py │ ├── doc3d_minio.py │ ├── docaligner.py │ └── list_dataset.py ├── datamodule │ ├── __init__.py │ ├── docaligner_module.py │ ├── mixed_module.py │ └── utils.py ├── fine_tune.py ├── inference.py ├── models │ ├── docmae.py │ ├── doctr.py │ ├── doctr_custom.py │ ├── doctr_plus.py │ ├── mae.py │ ├── rectification.py │ ├── transformer.py │ └── upscale.py ├── pretrain.py ├── pretrain_pl.py └── train.py ├── poetry.lock ├── pyproject.toml └── tests ├── __init__.py ├── architecture.py ├── datasets.py ├── doc3d ├── LICENSE ├── bm │ └── 9_6-vc_Page_002-YT40001.mat ├── img │ └── 9_6-vc_Page_002-YT40001.png ├── tiny.txt └── uv │ └── 9_6-vc_Page_002-YT40001.exr └── dtd ├── images └── lacelike_0037.jpg └── labels └── tiny.txt /.github/workflows/pytest.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/.github/workflows/pytest.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/README.md -------------------------------------------------------------------------------- /assets/18.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/assets/18.png -------------------------------------------------------------------------------- /assets/19.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/assets/19.png -------------------------------------------------------------------------------- /assets/19_mae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/assets/19_mae.png -------------------------------------------------------------------------------- /assets/19_masked.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/assets/19_masked.jpg -------------------------------------------------------------------------------- /assets/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/assets/2.png -------------------------------------------------------------------------------- /config/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/config/config.json -------------------------------------------------------------------------------- /config/docaligner.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/config/docaligner.gin -------------------------------------------------------------------------------- /config/doctr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/config/doctr.json -------------------------------------------------------------------------------- /config/doctr_plus.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/config/doctr_plus.json -------------------------------------------------------------------------------- /config/finetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/config/finetune.json -------------------------------------------------------------------------------- /demo/background_segmentation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/demo/background_segmentation.ipynb -------------------------------------------------------------------------------- /demo/doc3d_dataloader.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/demo/doc3d_dataloader.ipynb -------------------------------------------------------------------------------- /docmae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/__init__.py -------------------------------------------------------------------------------- /docmae/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docmae/data/augmentation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docmae/data/augmentation/random_resized_crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/data/augmentation/random_resized_crop.py -------------------------------------------------------------------------------- /docmae/data/augmentation/replace_background.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/data/augmentation/replace_background.py -------------------------------------------------------------------------------- /docmae/data/doc3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/data/doc3d.py -------------------------------------------------------------------------------- /docmae/data/doc3d_minio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/data/doc3d_minio.py -------------------------------------------------------------------------------- /docmae/data/docaligner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/data/docaligner.py -------------------------------------------------------------------------------- /docmae/data/list_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/data/list_dataset.py -------------------------------------------------------------------------------- /docmae/datamodule/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docmae/datamodule/docaligner_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/datamodule/docaligner_module.py -------------------------------------------------------------------------------- /docmae/datamodule/mixed_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/datamodule/mixed_module.py -------------------------------------------------------------------------------- /docmae/datamodule/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/datamodule/utils.py -------------------------------------------------------------------------------- /docmae/fine_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/fine_tune.py -------------------------------------------------------------------------------- /docmae/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/inference.py -------------------------------------------------------------------------------- /docmae/models/docmae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/models/docmae.py -------------------------------------------------------------------------------- /docmae/models/doctr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/models/doctr.py -------------------------------------------------------------------------------- /docmae/models/doctr_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/models/doctr_custom.py -------------------------------------------------------------------------------- /docmae/models/doctr_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/models/doctr_plus.py -------------------------------------------------------------------------------- /docmae/models/mae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/models/mae.py -------------------------------------------------------------------------------- /docmae/models/rectification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/models/rectification.py -------------------------------------------------------------------------------- /docmae/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/models/transformer.py -------------------------------------------------------------------------------- /docmae/models/upscale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/models/upscale.py -------------------------------------------------------------------------------- /docmae/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/pretrain.py -------------------------------------------------------------------------------- /docmae/pretrain_pl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/pretrain_pl.py -------------------------------------------------------------------------------- /docmae/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/docmae/train.py -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/architecture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/tests/architecture.py -------------------------------------------------------------------------------- /tests/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/tests/datasets.py -------------------------------------------------------------------------------- /tests/doc3d/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/tests/doc3d/LICENSE -------------------------------------------------------------------------------- /tests/doc3d/bm/9_6-vc_Page_002-YT40001.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/tests/doc3d/bm/9_6-vc_Page_002-YT40001.mat -------------------------------------------------------------------------------- /tests/doc3d/img/9_6-vc_Page_002-YT40001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/tests/doc3d/img/9_6-vc_Page_002-YT40001.png -------------------------------------------------------------------------------- /tests/doc3d/tiny.txt: -------------------------------------------------------------------------------- 1 | 9_6-vc_Page_002-YT40001 -------------------------------------------------------------------------------- /tests/doc3d/uv/9_6-vc_Page_002-YT40001.exr: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/tests/doc3d/uv/9_6-vc_Page_002-YT40001.exr -------------------------------------------------------------------------------- /tests/dtd/images/lacelike_0037.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dawars/DocMAE/HEAD/tests/dtd/images/lacelike_0037.jpg -------------------------------------------------------------------------------- /tests/dtd/labels/tiny.txt: -------------------------------------------------------------------------------- 1 | lacelike_0037.jpg --------------------------------------------------------------------------------