├── .figure ├── image_anno.jpg ├── image_only.jpg └── network.jpg ├── .gitignore ├── README.md ├── configs ├── __init__.py ├── cfg_base.py ├── cfg_dataset.py └── cfg_model.py ├── eval_utils.py ├── hrnet_code ├── .gitignore ├── LICENSE ├── README.md ├── lib │ ├── config │ │ ├── __init__.py │ │ ├── default.py │ │ └── models.py │ ├── core │ │ ├── criterion.py │ │ └── function.py │ ├── datasets │ │ ├── __init__.py │ │ ├── ade20k.py │ │ ├── base_dataset.py │ │ ├── cityscapes.py │ │ ├── cocostuff.py │ │ ├── lip.py │ │ └── pascal_ctx.py │ ├── models │ │ ├── __init__.py │ │ ├── bn_helper.py │ │ ├── seg_hrnet.py │ │ ├── seg_hrnet_ocr.py │ │ └── sync_bn │ │ │ ├── LICENSE │ │ │ ├── __init__.py │ │ │ └── inplace_abn │ │ │ ├── __init__.py │ │ │ ├── bn.py │ │ │ ├── functions.py │ │ │ └── src │ │ │ ├── common.h │ │ │ ├── inplace_abn.cpp │ │ │ ├── inplace_abn.h │ │ │ ├── inplace_abn_cpu.cpp │ │ │ └── inplace_abn_cuda.cu │ └── utils │ │ ├── __init__.py │ │ ├── distributed.py │ │ ├── modelsummary.py │ │ └── utils.py ├── requirements.txt └── tools │ ├── _init_paths.py │ ├── test.py │ └── train.py ├── inference.py ├── lib ├── __init__.py ├── cfg_helper.py ├── data_factory │ ├── __init__.py │ ├── ds_base.py │ ├── ds_cocotext.py │ ├── ds_formatter.py │ ├── ds_icdar13.py │ ├── ds_loader.py │ ├── ds_mlt.py │ ├── ds_sampler.py │ ├── ds_textseg.py │ ├── ds_textssc.py │ ├── ds_totaltext.py │ └── ds_transform.py ├── evaluate_service.py ├── log_service.py ├── loss.py ├── model_zoo │ ├── __init__.py │ ├── deeplab.py │ ├── get_model.py │ ├── hrnet.py │ ├── optim_manager.py │ ├── resnet.py │ ├── texrnet.py │ └── utils.py ├── nputils.py ├── optimizer │ ├── __init__.py │ └── get_optimizer.py └── torchutils.py ├── main.py ├── requirement.txt └── train_utils.py /.figure/image_anno.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/.figure/image_anno.jpg -------------------------------------------------------------------------------- /.figure/image_only.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/.figure/image_only.jpg -------------------------------------------------------------------------------- /.figure/network.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/.figure/network.jpg -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/cfg_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/configs/cfg_base.py -------------------------------------------------------------------------------- /configs/cfg_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/configs/cfg_dataset.py -------------------------------------------------------------------------------- /configs/cfg_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/configs/cfg_model.py -------------------------------------------------------------------------------- /eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/eval_utils.py -------------------------------------------------------------------------------- /hrnet_code/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/.gitignore -------------------------------------------------------------------------------- /hrnet_code/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/LICENSE -------------------------------------------------------------------------------- /hrnet_code/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/README.md -------------------------------------------------------------------------------- /hrnet_code/lib/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/config/__init__.py -------------------------------------------------------------------------------- /hrnet_code/lib/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/config/default.py -------------------------------------------------------------------------------- /hrnet_code/lib/config/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/config/models.py -------------------------------------------------------------------------------- /hrnet_code/lib/core/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/core/criterion.py -------------------------------------------------------------------------------- /hrnet_code/lib/core/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/core/function.py -------------------------------------------------------------------------------- /hrnet_code/lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/datasets/__init__.py -------------------------------------------------------------------------------- /hrnet_code/lib/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/datasets/ade20k.py -------------------------------------------------------------------------------- /hrnet_code/lib/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/datasets/base_dataset.py -------------------------------------------------------------------------------- /hrnet_code/lib/datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/datasets/cityscapes.py -------------------------------------------------------------------------------- /hrnet_code/lib/datasets/cocostuff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/datasets/cocostuff.py -------------------------------------------------------------------------------- /hrnet_code/lib/datasets/lip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/datasets/lip.py -------------------------------------------------------------------------------- /hrnet_code/lib/datasets/pascal_ctx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/datasets/pascal_ctx.py -------------------------------------------------------------------------------- /hrnet_code/lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/__init__.py -------------------------------------------------------------------------------- /hrnet_code/lib/models/bn_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/bn_helper.py -------------------------------------------------------------------------------- /hrnet_code/lib/models/seg_hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/seg_hrnet.py -------------------------------------------------------------------------------- /hrnet_code/lib/models/seg_hrnet_ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/seg_hrnet_ocr.py -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/LICENSE -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/__init__.py: -------------------------------------------------------------------------------- 1 | from .inplace_abn import bn -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/inplace_abn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/inplace_abn/__init__.py -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/inplace_abn/bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/inplace_abn/bn.py -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/inplace_abn/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/inplace_abn/functions.py -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/inplace_abn/src/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/inplace_abn/src/common.h -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/inplace_abn/src/inplace_abn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/inplace_abn/src/inplace_abn.cpp -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/inplace_abn/src/inplace_abn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/inplace_abn/src/inplace_abn.h -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/inplace_abn/src/inplace_abn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/inplace_abn/src/inplace_abn_cpu.cpp -------------------------------------------------------------------------------- /hrnet_code/lib/models/sync_bn/inplace_abn/src/inplace_abn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/models/sync_bn/inplace_abn/src/inplace_abn_cuda.cu -------------------------------------------------------------------------------- /hrnet_code/lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hrnet_code/lib/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/utils/distributed.py -------------------------------------------------------------------------------- /hrnet_code/lib/utils/modelsummary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/utils/modelsummary.py -------------------------------------------------------------------------------- /hrnet_code/lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/lib/utils/utils.py -------------------------------------------------------------------------------- /hrnet_code/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/requirements.txt -------------------------------------------------------------------------------- /hrnet_code/tools/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/tools/_init_paths.py -------------------------------------------------------------------------------- /hrnet_code/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/tools/test.py -------------------------------------------------------------------------------- /hrnet_code/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/hrnet_code/tools/train.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/inference.py -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/cfg_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/cfg_helper.py -------------------------------------------------------------------------------- /lib/data_factory/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/__init__.py -------------------------------------------------------------------------------- /lib/data_factory/ds_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_base.py -------------------------------------------------------------------------------- /lib/data_factory/ds_cocotext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_cocotext.py -------------------------------------------------------------------------------- /lib/data_factory/ds_formatter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_formatter.py -------------------------------------------------------------------------------- /lib/data_factory/ds_icdar13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_icdar13.py -------------------------------------------------------------------------------- /lib/data_factory/ds_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_loader.py -------------------------------------------------------------------------------- /lib/data_factory/ds_mlt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_mlt.py -------------------------------------------------------------------------------- /lib/data_factory/ds_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_sampler.py -------------------------------------------------------------------------------- /lib/data_factory/ds_textseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_textseg.py -------------------------------------------------------------------------------- /lib/data_factory/ds_textssc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_textssc.py -------------------------------------------------------------------------------- /lib/data_factory/ds_totaltext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_totaltext.py -------------------------------------------------------------------------------- /lib/data_factory/ds_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/data_factory/ds_transform.py -------------------------------------------------------------------------------- /lib/evaluate_service.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/evaluate_service.py -------------------------------------------------------------------------------- /lib/log_service.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/log_service.py -------------------------------------------------------------------------------- /lib/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/loss.py -------------------------------------------------------------------------------- /lib/model_zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/model_zoo/__init__.py -------------------------------------------------------------------------------- /lib/model_zoo/deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/model_zoo/deeplab.py -------------------------------------------------------------------------------- /lib/model_zoo/get_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/model_zoo/get_model.py -------------------------------------------------------------------------------- /lib/model_zoo/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/model_zoo/hrnet.py -------------------------------------------------------------------------------- /lib/model_zoo/optim_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/model_zoo/optim_manager.py -------------------------------------------------------------------------------- /lib/model_zoo/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/model_zoo/resnet.py -------------------------------------------------------------------------------- /lib/model_zoo/texrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/model_zoo/texrnet.py -------------------------------------------------------------------------------- /lib/model_zoo/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/model_zoo/utils.py -------------------------------------------------------------------------------- /lib/nputils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/nputils.py -------------------------------------------------------------------------------- /lib/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/optimizer/__init__.py -------------------------------------------------------------------------------- /lib/optimizer/get_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/optimizer/get_optimizer.py -------------------------------------------------------------------------------- /lib/torchutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/lib/torchutils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/main.py -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/requirement.txt -------------------------------------------------------------------------------- /train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SHI-Labs/Rethinking-Text-Segmentation/HEAD/train_utils.py --------------------------------------------------------------------------------