├── .gitattributes ├── .gitignore ├── LICENSE.md ├── README.MD ├── base ├── __init__.py ├── base_dataset.py └── base_trainer.py ├── config ├── SynthText.yaml ├── SynthText_resnet18_FPN_DBhead_polyLR.yaml ├── icdar2015.yaml ├── icdar2015_dcn_resnet18_FPN_DBhead_polyLR.yaml ├── icdar2015_resnet18_FPN_DBhead_polyLR.yaml ├── icdar2015_resnet18_FPN_DBhead_polyLR_finetune.yaml ├── open_dataset.yaml ├── open_dataset_dcn_resnet50_FPN_DBhead_polyLR.yaml ├── open_dataset_resnest50_FPN_DBhead_polyLR.yaml └── open_dataset_resnet18_FPN_DBhead_polyLR.yaml ├── data_loader ├── __init__.py ├── dataset.py └── modules │ ├── __init__.py │ ├── augment.py │ ├── iaa_augment.py │ ├── make_border_map.py │ ├── make_shrink_map.py │ └── random_crop_data.py ├── datasets ├── test.txt ├── test │ ├── gt │ │ └── README.MD │ └── img │ │ └── README.MD ├── train.txt └── train │ ├── gt │ └── README.MD │ └── img │ └── README.MD ├── environment.yml ├── eval.sh ├── generate_lists.sh ├── imgs └── paper │ └── db.jpg ├── models ├── __init__.py ├── backbone │ ├── MobilenetV3.py │ ├── __init__.py │ ├── resnest │ │ ├── __init__.py │ │ ├── ablation.py │ │ ├── resnest.py │ │ ├── resnet.py │ │ └── splat.py │ ├── resnet.py │ └── shufflenetv2.py ├── basic.py ├── head │ ├── ConvHead.py │ ├── DBHead.py │ └── __init__.py ├── losses │ ├── DB_loss.py │ ├── __init__.py │ └── basic_loss.py ├── model.py └── neck │ ├── FPEM_FFM.py │ ├── FPN.py │ └── __init__.py ├── multi_gpu_train.sh ├── post_processing ├── __init__.py └── seg_detector_representer.py ├── predict.sh ├── requirement.txt ├── singlel_gpu_train.sh ├── test └── README.MD ├── tools ├── __init__.py ├── eval.py ├── predict.py └── train.py ├── trainer ├── __init__.py └── trainer.py └── utils ├── __init__.py ├── cal_recall ├── __init__.py ├── rrc_evaluation_funcs.py └── script.py ├── compute_mean_std.py ├── make_trainfile.py ├── metrics.py ├── ocr_metric ├── __init__.py └── icdar2015 │ ├── __init__.py │ ├── detection │ ├── __init__.py │ ├── deteval.py │ ├── icdar2013.py │ ├── iou.py │ └── mtwi2018.py │ └── quad_metric.py ├── schedulers.py └── util.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/README.MD -------------------------------------------------------------------------------- /base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/base/__init__.py -------------------------------------------------------------------------------- /base/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/base/base_dataset.py -------------------------------------------------------------------------------- /base/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/base/base_trainer.py -------------------------------------------------------------------------------- /config/SynthText.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/SynthText.yaml -------------------------------------------------------------------------------- /config/SynthText_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/SynthText_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/icdar2015.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/icdar2015.yaml -------------------------------------------------------------------------------- /config/icdar2015_dcn_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/icdar2015_dcn_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/icdar2015_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/icdar2015_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/icdar2015_resnet18_FPN_DBhead_polyLR_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/icdar2015_resnet18_FPN_DBhead_polyLR_finetune.yaml -------------------------------------------------------------------------------- /config/open_dataset.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/open_dataset.yaml -------------------------------------------------------------------------------- /config/open_dataset_dcn_resnet50_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/open_dataset_dcn_resnet50_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/open_dataset_resnest50_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/open_dataset_resnest50_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /config/open_dataset_resnet18_FPN_DBhead_polyLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/config/open_dataset_resnet18_FPN_DBhead_polyLR.yaml -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/data_loader/__init__.py -------------------------------------------------------------------------------- /data_loader/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/data_loader/dataset.py -------------------------------------------------------------------------------- /data_loader/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/data_loader/modules/__init__.py -------------------------------------------------------------------------------- /data_loader/modules/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/data_loader/modules/augment.py -------------------------------------------------------------------------------- /data_loader/modules/iaa_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/data_loader/modules/iaa_augment.py -------------------------------------------------------------------------------- /data_loader/modules/make_border_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/data_loader/modules/make_border_map.py -------------------------------------------------------------------------------- /data_loader/modules/make_shrink_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/data_loader/modules/make_shrink_map.py -------------------------------------------------------------------------------- /data_loader/modules/random_crop_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/data_loader/modules/random_crop_data.py -------------------------------------------------------------------------------- /datasets/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/datasets/test.txt -------------------------------------------------------------------------------- /datasets/test/gt/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/datasets/test/gt/README.MD -------------------------------------------------------------------------------- /datasets/test/img/README.MD: -------------------------------------------------------------------------------- 1 | Place the `.jpg` files here. 2 | -------------------------------------------------------------------------------- /datasets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/datasets/train.txt -------------------------------------------------------------------------------- /datasets/train/gt/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/datasets/train/gt/README.MD -------------------------------------------------------------------------------- /datasets/train/img/README.MD: -------------------------------------------------------------------------------- 1 | Place the `.jpg` files here. 2 | -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/environment.yml -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- 1 | CUDA_VISIBLE_DEVICES=0 python3 tools/eval.py --model_path '' -------------------------------------------------------------------------------- /generate_lists.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/generate_lists.sh -------------------------------------------------------------------------------- /imgs/paper/db.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/imgs/paper/db.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone/MobilenetV3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/backbone/MobilenetV3.py -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/resnest/__init__.py: -------------------------------------------------------------------------------- 1 | from .resnest import * -------------------------------------------------------------------------------- /models/backbone/resnest/ablation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/backbone/resnest/ablation.py -------------------------------------------------------------------------------- /models/backbone/resnest/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/backbone/resnest/resnest.py -------------------------------------------------------------------------------- /models/backbone/resnest/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/backbone/resnest/resnet.py -------------------------------------------------------------------------------- /models/backbone/resnest/splat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/backbone/resnest/splat.py -------------------------------------------------------------------------------- /models/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/backbone/resnet.py -------------------------------------------------------------------------------- /models/backbone/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/backbone/shufflenetv2.py -------------------------------------------------------------------------------- /models/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/basic.py -------------------------------------------------------------------------------- /models/head/ConvHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/head/ConvHead.py -------------------------------------------------------------------------------- /models/head/DBHead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/head/DBHead.py -------------------------------------------------------------------------------- /models/head/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/head/__init__.py -------------------------------------------------------------------------------- /models/losses/DB_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/losses/DB_loss.py -------------------------------------------------------------------------------- /models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/losses/__init__.py -------------------------------------------------------------------------------- /models/losses/basic_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/losses/basic_loss.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/model.py -------------------------------------------------------------------------------- /models/neck/FPEM_FFM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/neck/FPEM_FFM.py -------------------------------------------------------------------------------- /models/neck/FPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/neck/FPN.py -------------------------------------------------------------------------------- /models/neck/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/models/neck/__init__.py -------------------------------------------------------------------------------- /multi_gpu_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/multi_gpu_train.sh -------------------------------------------------------------------------------- /post_processing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/post_processing/__init__.py -------------------------------------------------------------------------------- /post_processing/seg_detector_representer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/post_processing/seg_detector_representer.py -------------------------------------------------------------------------------- /predict.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/predict.sh -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/requirement.txt -------------------------------------------------------------------------------- /singlel_gpu_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/singlel_gpu_train.sh -------------------------------------------------------------------------------- /test/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/test/README.MD -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/tools/predict.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/tools/train.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/trainer/__init__.py -------------------------------------------------------------------------------- /trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/trainer/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/cal_recall/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/cal_recall/__init__.py -------------------------------------------------------------------------------- /utils/cal_recall/rrc_evaluation_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/cal_recall/rrc_evaluation_funcs.py -------------------------------------------------------------------------------- /utils/cal_recall/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/cal_recall/script.py -------------------------------------------------------------------------------- /utils/compute_mean_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/compute_mean_std.py -------------------------------------------------------------------------------- /utils/make_trainfile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/make_trainfile.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/ocr_metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/ocr_metric/__init__.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/ocr_metric/icdar2015/__init__.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/deteval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/ocr_metric/icdar2015/detection/deteval.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/icdar2013.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/ocr_metric/icdar2015/detection/icdar2013.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/ocr_metric/icdar2015/detection/iou.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/detection/mtwi2018.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/ocr_metric/icdar2015/detection/mtwi2018.py -------------------------------------------------------------------------------- /utils/ocr_metric/icdar2015/quad_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/ocr_metric/icdar2015/quad_metric.py -------------------------------------------------------------------------------- /utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/schedulers.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BaofengZan/DBNet.pytorch/HEAD/utils/util.py --------------------------------------------------------------------------------