├── LICENSE.md ├── README.md ├── __init__.py ├── base ├── __init__.py └── base_trainer.py ├── config.json ├── config ├── __init__.py └── default.py ├── data_loader ├── __init__.py ├── augment.py ├── data_utils.py └── dataset.py ├── demo ├── bbox.png ├── bin_map.png ├── contour.png ├── dbnet.png ├── prob_map.png └── thres_map.png ├── eval.py ├── genText.py ├── models ├── __init__.py ├── loss.py ├── model.py └── modules │ ├── __init__.py │ ├── resnet.py │ ├── segmentation_head.py │ └── shufflenetv2.py ├── predict.py ├── train.py ├── trainer ├── __init__.py └── trainer.py └── utils ├── __init__.py ├── cal_recall ├── __init__.py ├── rrc_evaluation_funcs.py └── script.py ├── make_trainfile.py ├── metrics.py ├── schedulers.py └── util.py /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- -------------------------------------------------------------------------------- /base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/base/__init__.py -------------------------------------------------------------------------------- /base/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/base/base_trainer.py -------------------------------------------------------------------------------- /config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/config.json -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | -------------------------------------------------------------------------------- /config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/config/default.py -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/data_loader/__init__.py -------------------------------------------------------------------------------- /data_loader/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/data_loader/augment.py -------------------------------------------------------------------------------- /data_loader/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/data_loader/data_utils.py -------------------------------------------------------------------------------- /data_loader/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/data_loader/dataset.py -------------------------------------------------------------------------------- /demo/bbox.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/demo/bbox.png -------------------------------------------------------------------------------- /demo/bin_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/demo/bin_map.png -------------------------------------------------------------------------------- /demo/contour.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/demo/contour.png -------------------------------------------------------------------------------- /demo/dbnet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/demo/dbnet.png -------------------------------------------------------------------------------- /demo/prob_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/demo/prob_map.png -------------------------------------------------------------------------------- /demo/thres_map.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/demo/thres_map.png -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/eval.py -------------------------------------------------------------------------------- /genText.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/genText.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/models/model.py -------------------------------------------------------------------------------- /models/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/models/modules/__init__.py -------------------------------------------------------------------------------- /models/modules/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/models/modules/resnet.py -------------------------------------------------------------------------------- /models/modules/segmentation_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/models/modules/segmentation_head.py -------------------------------------------------------------------------------- /models/modules/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/models/modules/shufflenetv2.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/predict.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/train.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/trainer/__init__.py -------------------------------------------------------------------------------- /trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/trainer/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/cal_recall/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/utils/cal_recall/__init__.py -------------------------------------------------------------------------------- /utils/cal_recall/rrc_evaluation_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/utils/cal_recall/rrc_evaluation_funcs.py -------------------------------------------------------------------------------- /utils/cal_recall/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/utils/cal_recall/script.py -------------------------------------------------------------------------------- /utils/make_trainfile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/utils/make_trainfile.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/utils/schedulers.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SURFZJY/Real-time-Text-Detection-DBNet/HEAD/utils/util.py --------------------------------------------------------------------------------