├── LICENSE ├── README.md ├── datasets ├── DatasetBuilder.py ├── __init__.py ├── dad.py ├── dad_gscnn.py ├── publaynet.py └── publaynet_gscnn.py ├── inference.py ├── loss.py ├── models ├── ModelBuilder.py ├── __init__.py ├── deeplabv3plus │ ├── __init__.py │ └── build_deeplabv3plus.py ├── fast_fcn │ ├── __init__.py │ └── build_fast_fcn.py ├── gated_scnn │ ├── __init__.py │ ├── build_gscnn.py │ └── gated_shape_cnn │ │ ├── __init__.py │ │ ├── datasets │ │ ├── __init__.py │ │ ├── dad │ │ │ ├── __init__.py │ │ │ ├── dataset.py │ │ │ └── raw_dataset.py │ │ └── publaynet │ │ │ ├── __init__.py │ │ │ ├── dataset.py │ │ │ └── raw_dataset.py │ │ ├── model │ │ ├── __init__.py │ │ ├── atrous_xception.py │ │ ├── layers.py │ │ └── model_definition.py │ │ ├── training │ │ ├── __init__.py │ │ ├── dataset.py │ │ ├── loss.py │ │ └── utils.py │ │ └── training_scripts │ │ ├── __init__.py │ │ ├── cityscapes_model.py │ │ ├── dad_model.py │ │ └── scene_parsing_model.py └── unet │ ├── __init__.py │ └── build_unet.py ├── publaynet_small_train_subset.txt ├── requirements.txt ├── train.py └── utils ├── AnnotationUtils.py ├── DataLoaderUtils.py ├── MetricsUtils.py └── __init__.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/README.md -------------------------------------------------------------------------------- /datasets/DatasetBuilder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/datasets/DatasetBuilder.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/dad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/datasets/dad.py -------------------------------------------------------------------------------- /datasets/dad_gscnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/datasets/dad_gscnn.py -------------------------------------------------------------------------------- /datasets/publaynet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/datasets/publaynet.py -------------------------------------------------------------------------------- /datasets/publaynet_gscnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/datasets/publaynet_gscnn.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/inference.py -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/loss.py -------------------------------------------------------------------------------- /models/ModelBuilder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/ModelBuilder.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/deeplabv3plus/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/deeplabv3plus/build_deeplabv3plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/deeplabv3plus/build_deeplabv3plus.py -------------------------------------------------------------------------------- /models/fast_fcn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/fast_fcn/build_fast_fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/fast_fcn/build_fast_fcn.py -------------------------------------------------------------------------------- /models/gated_scnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gated_scnn/build_gscnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/build_gscnn.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/datasets/dad/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/datasets/dad/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/datasets/dad/dataset.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/datasets/dad/raw_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/datasets/dad/raw_dataset.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/datasets/publaynet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/datasets/publaynet/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/datasets/publaynet/dataset.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/datasets/publaynet/raw_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/datasets/publaynet/raw_dataset.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/model/__init__.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/model/atrous_xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/model/atrous_xception.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/model/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/model/layers.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/model/model_definition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/model/model_definition.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/training/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/training/dataset.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/training/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/training/loss.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/training/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/training/utils.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/training_scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/training_scripts/cityscapes_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/training_scripts/cityscapes_model.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/training_scripts/dad_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/training_scripts/dad_model.py -------------------------------------------------------------------------------- /models/gated_scnn/gated_shape_cnn/training_scripts/scene_parsing_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/gated_scnn/gated_shape_cnn/training_scripts/scene_parsing_model.py -------------------------------------------------------------------------------- /models/unet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/unet/build_unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/models/unet/build_unet.py -------------------------------------------------------------------------------- /publaynet_small_train_subset.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/publaynet_small_train_subset.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/train.py -------------------------------------------------------------------------------- /utils/AnnotationUtils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/utils/AnnotationUtils.py -------------------------------------------------------------------------------- /utils/DataLoaderUtils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/utils/DataLoaderUtils.py -------------------------------------------------------------------------------- /utils/MetricsUtils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LivingSkyTechnologies/Document_Layout_Segmentation/HEAD/utils/MetricsUtils.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------