├── .gitignore ├── LICENSE ├── __init__.py ├── compile.sh ├── config └── psenet │ ├── psenet_r50_ctw.py │ ├── psenet_r50_ctw_finetune.py │ ├── psenet_r50_ic15_1024.py │ ├── psenet_r50_ic15_1024_finetune.py │ ├── psenet_r50_ic15_736.py │ ├── psenet_r50_ic15_736_finetune.py │ ├── psenet_r50_synth.py │ ├── psenet_r50_tt.py │ └── psenet_r50_tt_finetune.py ├── dataset ├── __init__.py ├── builder.py └── psenet │ ├── __init__.py │ ├── check_dataloader.py │ ├── psenet_ctw.py │ ├── psenet_ic15.py │ ├── psenet_synth.py │ └── psenet_tt.py ├── eval ├── ctw │ ├── eval.py │ └── file_util.py ├── eval_ctw.sh ├── eval_ic15.sh ├── eval_ic15_rec.sh ├── eval_msra.sh ├── eval_tt.sh ├── eval_tt_rec.sh ├── ic15 │ ├── gt.zip │ ├── rrc_evaluation_funcs.py │ ├── rrc_evaluation_funcs_v1.py │ ├── rrc_evaluation_funcs_v2.py │ ├── script.py │ └── script_self_adapt.py ├── ic15_rec │ ├── gt.zip │ ├── readme.txt │ ├── rrc_evaluation_funcs_1_1.py │ ├── script.py │ └── script_self_adapt.py ├── msra │ ├── eval.py │ └── file_util.py ├── tt │ ├── Deteval.py │ ├── Deteval_rec.py │ └── polygon_wrapper.py └── tt_rec │ ├── gt.zip │ ├── readme.txt │ ├── rrc_evaluation_funcs_1_1.py │ └── script.py ├── logo.jpg ├── models ├── __init__.py ├── backbone │ ├── __init__.py │ ├── builder.py │ └── resnet.py ├── builder.py ├── head │ ├── __init__.py │ ├── builder.py │ └── psenet_head.py ├── loss │ ├── __init__.py │ ├── acc.py │ ├── builder.py │ ├── dice_loss.py │ ├── emb_loss_v1.py │ ├── iou.py │ └── ohem.py ├── neck │ ├── __init__.py │ ├── builder.py │ └── fpn.py ├── post_processing │ ├── __init__.py │ └── pse │ │ ├── __init__.py │ │ ├── pse.cpp │ │ ├── pse.pyx │ │ ├── readme.txt │ │ └── setup.py ├── psenet.py ├── pypse.py └── utils │ ├── __init__.py │ ├── conv_bn_relu.py │ └── fuse_conv_bn.py ├── readme.md ├── requirement.txt ├── test.py ├── train.py └── utils ├── __init__.py ├── average_meter.py ├── logger.py └── result_format.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/LICENSE -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | #from .pa import pa 2 | -------------------------------------------------------------------------------- /compile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/compile.sh -------------------------------------------------------------------------------- /config/psenet/psenet_r50_ctw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_ctw.py -------------------------------------------------------------------------------- /config/psenet/psenet_r50_ctw_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_ctw_finetune.py -------------------------------------------------------------------------------- /config/psenet/psenet_r50_ic15_1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_ic15_1024.py -------------------------------------------------------------------------------- /config/psenet/psenet_r50_ic15_1024_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_ic15_1024_finetune.py -------------------------------------------------------------------------------- /config/psenet/psenet_r50_ic15_736.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_ic15_736.py -------------------------------------------------------------------------------- /config/psenet/psenet_r50_ic15_736_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_ic15_736_finetune.py -------------------------------------------------------------------------------- /config/psenet/psenet_r50_synth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_synth.py -------------------------------------------------------------------------------- /config/psenet/psenet_r50_tt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_tt.py -------------------------------------------------------------------------------- /config/psenet/psenet_r50_tt_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/config/psenet/psenet_r50_tt_finetune.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/dataset/builder.py -------------------------------------------------------------------------------- /dataset/psenet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/dataset/psenet/__init__.py -------------------------------------------------------------------------------- /dataset/psenet/check_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/dataset/psenet/check_dataloader.py -------------------------------------------------------------------------------- /dataset/psenet/psenet_ctw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/dataset/psenet/psenet_ctw.py -------------------------------------------------------------------------------- /dataset/psenet/psenet_ic15.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/dataset/psenet/psenet_ic15.py -------------------------------------------------------------------------------- /dataset/psenet/psenet_synth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/dataset/psenet/psenet_synth.py -------------------------------------------------------------------------------- /dataset/psenet/psenet_tt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/dataset/psenet/psenet_tt.py -------------------------------------------------------------------------------- /eval/ctw/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ctw/eval.py -------------------------------------------------------------------------------- /eval/ctw/file_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ctw/file_util.py -------------------------------------------------------------------------------- /eval/eval_ctw.sh: -------------------------------------------------------------------------------- 1 | cd ctw && python2 eval.py && cd .. 2 | -------------------------------------------------------------------------------- /eval/eval_ic15.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/eval_ic15.sh -------------------------------------------------------------------------------- /eval/eval_ic15_rec.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/eval_ic15_rec.sh -------------------------------------------------------------------------------- /eval/eval_msra.sh: -------------------------------------------------------------------------------- 1 | cd msra && python2 eval.py && cd .. 2 | -------------------------------------------------------------------------------- /eval/eval_tt.sh: -------------------------------------------------------------------------------- 1 | cd tt && python2 Deteval.py && cd .. -------------------------------------------------------------------------------- /eval/eval_tt_rec.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/eval_tt_rec.sh -------------------------------------------------------------------------------- /eval/ic15/gt.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15/gt.zip -------------------------------------------------------------------------------- /eval/ic15/rrc_evaluation_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15/rrc_evaluation_funcs.py -------------------------------------------------------------------------------- /eval/ic15/rrc_evaluation_funcs_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15/rrc_evaluation_funcs_v1.py -------------------------------------------------------------------------------- /eval/ic15/rrc_evaluation_funcs_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15/rrc_evaluation_funcs_v2.py -------------------------------------------------------------------------------- /eval/ic15/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15/script.py -------------------------------------------------------------------------------- /eval/ic15/script_self_adapt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15/script_self_adapt.py -------------------------------------------------------------------------------- /eval/ic15_rec/gt.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15_rec/gt.zip -------------------------------------------------------------------------------- /eval/ic15_rec/readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15_rec/readme.txt -------------------------------------------------------------------------------- /eval/ic15_rec/rrc_evaluation_funcs_1_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15_rec/rrc_evaluation_funcs_1_1.py -------------------------------------------------------------------------------- /eval/ic15_rec/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15_rec/script.py -------------------------------------------------------------------------------- /eval/ic15_rec/script_self_adapt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/ic15_rec/script_self_adapt.py -------------------------------------------------------------------------------- /eval/msra/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/msra/eval.py -------------------------------------------------------------------------------- /eval/msra/file_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/msra/file_util.py -------------------------------------------------------------------------------- /eval/tt/Deteval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/tt/Deteval.py -------------------------------------------------------------------------------- /eval/tt/Deteval_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/tt/Deteval_rec.py -------------------------------------------------------------------------------- /eval/tt/polygon_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/tt/polygon_wrapper.py -------------------------------------------------------------------------------- /eval/tt_rec/gt.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/tt_rec/gt.zip -------------------------------------------------------------------------------- /eval/tt_rec/readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/tt_rec/readme.txt -------------------------------------------------------------------------------- /eval/tt_rec/rrc_evaluation_funcs_1_1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/tt_rec/rrc_evaluation_funcs_1_1.py -------------------------------------------------------------------------------- /eval/tt_rec/script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/eval/tt_rec/script.py -------------------------------------------------------------------------------- /logo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/logo.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/backbone/__init__.py -------------------------------------------------------------------------------- /models/backbone/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/backbone/builder.py -------------------------------------------------------------------------------- /models/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/backbone/resnet.py -------------------------------------------------------------------------------- /models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/builder.py -------------------------------------------------------------------------------- /models/head/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/head/__init__.py -------------------------------------------------------------------------------- /models/head/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/head/builder.py -------------------------------------------------------------------------------- /models/head/psenet_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/head/psenet_head.py -------------------------------------------------------------------------------- /models/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/loss/__init__.py -------------------------------------------------------------------------------- /models/loss/acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/loss/acc.py -------------------------------------------------------------------------------- /models/loss/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/loss/builder.py -------------------------------------------------------------------------------- /models/loss/dice_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/loss/dice_loss.py -------------------------------------------------------------------------------- /models/loss/emb_loss_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/loss/emb_loss_v1.py -------------------------------------------------------------------------------- /models/loss/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/loss/iou.py -------------------------------------------------------------------------------- /models/loss/ohem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/loss/ohem.py -------------------------------------------------------------------------------- /models/neck/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/neck/__init__.py -------------------------------------------------------------------------------- /models/neck/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/neck/builder.py -------------------------------------------------------------------------------- /models/neck/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/neck/fpn.py -------------------------------------------------------------------------------- /models/post_processing/__init__.py: -------------------------------------------------------------------------------- 1 | from .pse import pse 2 | -------------------------------------------------------------------------------- /models/post_processing/pse/__init__.py: -------------------------------------------------------------------------------- 1 | from .pse import pse -------------------------------------------------------------------------------- /models/post_processing/pse/pse.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/post_processing/pse/pse.cpp -------------------------------------------------------------------------------- /models/post_processing/pse/pse.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/post_processing/pse/pse.pyx -------------------------------------------------------------------------------- /models/post_processing/pse/readme.txt: -------------------------------------------------------------------------------- 1 | python setup.py build_ext --inplace -------------------------------------------------------------------------------- /models/post_processing/pse/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/post_processing/pse/setup.py -------------------------------------------------------------------------------- /models/psenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/psenet.py -------------------------------------------------------------------------------- /models/pypse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/pypse.py -------------------------------------------------------------------------------- /models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/utils/__init__.py -------------------------------------------------------------------------------- /models/utils/conv_bn_relu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/utils/conv_bn_relu.py -------------------------------------------------------------------------------- /models/utils/fuse_conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/models/utils/fuse_conv_bn.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/readme.md -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/requirement.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/average_meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/utils/average_meter.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/result_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/whai362/PSENet/HEAD/utils/result_format.py --------------------------------------------------------------------------------