├── .gitignore ├── LICENSE ├── README.md ├── ctpn ├── __init__.py ├── demo.py ├── demo_pb.py ├── generate_pb.py ├── text.yml └── train_net.py ├── data ├── VOCdevkit2007 ├── demo │ ├── 006.jpg │ ├── 007.jpg │ ├── 008.jpg │ ├── 009.jpg │ └── 010.png ├── oriented_results │ ├── 001.jpg │ ├── 002.jpg │ ├── 003.jpg │ ├── 004.jpg │ ├── 005.jpg │ ├── 006.jpg │ ├── 007.jpg │ ├── 008.jpg │ ├── 009.jpg │ └── 010.png └── results │ ├── 006.jpg │ ├── 007.jpg │ ├── 008.jpg │ ├── 009.jpg │ ├── 010.png │ ├── res_006.txt │ ├── res_007.txt │ ├── res_008.txt │ ├── res_009.txt │ └── res_010.txt ├── lib ├── __init__.py ├── datasets │ ├── __init__.py │ ├── factory.py │ ├── imdb.py │ └── pascal_voc.py ├── fast_rcnn │ ├── __init__.py │ ├── bbox_transform.py │ ├── config.py │ ├── nms_wrapper.py │ ├── test.py │ └── train.py ├── networks │ ├── VGGnet_test.py │ ├── VGGnet_train.py │ ├── __init__.py │ ├── factory.py │ └── network.py ├── prepare_training_data │ ├── ToVoc.py │ └── split_label.py ├── roi_data_layer │ ├── __init__.py │ ├── layer.py │ ├── minibatch.py │ └── roidb.py ├── rpn_msr │ ├── __init__.py │ ├── anchor_target_layer_tf.py │ ├── generate_anchors.py │ └── proposal_layer_tf.py ├── text_connector │ ├── __init__.py │ ├── detectors.py │ ├── other.py │ ├── text_connect_cfg.py │ ├── text_proposal_connector.py │ ├── text_proposal_connector_oriented.py │ └── text_proposal_graph_builder.py └── utils │ ├── __init__.py │ ├── bbox.c │ ├── bbox.pyx │ ├── blob.py │ ├── boxes_grid.py │ ├── cython_nms.c │ ├── cython_nms.pyx │ ├── gpu_nms.c │ ├── gpu_nms.cpp │ ├── gpu_nms.hpp │ ├── gpu_nms.pyx │ ├── make.sh │ ├── nms_kernel.cu │ ├── setup.py │ └── timer.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/README.md -------------------------------------------------------------------------------- /ctpn/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /ctpn/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/ctpn/demo.py -------------------------------------------------------------------------------- /ctpn/demo_pb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/ctpn/demo_pb.py -------------------------------------------------------------------------------- /ctpn/generate_pb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/ctpn/generate_pb.py -------------------------------------------------------------------------------- /ctpn/text.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/ctpn/text.yml -------------------------------------------------------------------------------- /ctpn/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/ctpn/train_net.py -------------------------------------------------------------------------------- /data/VOCdevkit2007: -------------------------------------------------------------------------------- 1 | /media/D/code/OCR/CTPN_LSTM/data/VOCdevkit -------------------------------------------------------------------------------- /data/demo/006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/demo/006.jpg -------------------------------------------------------------------------------- /data/demo/007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/demo/007.jpg -------------------------------------------------------------------------------- /data/demo/008.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/demo/008.jpg -------------------------------------------------------------------------------- /data/demo/009.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/demo/009.jpg -------------------------------------------------------------------------------- /data/demo/010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/demo/010.png -------------------------------------------------------------------------------- /data/oriented_results/001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/001.jpg -------------------------------------------------------------------------------- /data/oriented_results/002.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/002.jpg -------------------------------------------------------------------------------- /data/oriented_results/003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/003.jpg -------------------------------------------------------------------------------- /data/oriented_results/004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/004.jpg -------------------------------------------------------------------------------- /data/oriented_results/005.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/005.jpg -------------------------------------------------------------------------------- /data/oriented_results/006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/006.jpg -------------------------------------------------------------------------------- /data/oriented_results/007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/007.jpg -------------------------------------------------------------------------------- /data/oriented_results/008.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/008.jpg -------------------------------------------------------------------------------- /data/oriented_results/009.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/009.jpg -------------------------------------------------------------------------------- /data/oriented_results/010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/oriented_results/010.png -------------------------------------------------------------------------------- /data/results/006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/006.jpg -------------------------------------------------------------------------------- /data/results/007.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/007.jpg -------------------------------------------------------------------------------- /data/results/008.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/008.jpg -------------------------------------------------------------------------------- /data/results/009.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/009.jpg -------------------------------------------------------------------------------- /data/results/010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/010.png -------------------------------------------------------------------------------- /data/results/res_006.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/res_006.txt -------------------------------------------------------------------------------- /data/results/res_007.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/res_007.txt -------------------------------------------------------------------------------- /data/results/res_008.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/res_008.txt -------------------------------------------------------------------------------- /data/results/res_009.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/res_009.txt -------------------------------------------------------------------------------- /data/results/res_010.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/data/results/res_010.txt -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/datasets/__init__.py -------------------------------------------------------------------------------- /lib/datasets/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/datasets/factory.py -------------------------------------------------------------------------------- /lib/datasets/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/datasets/imdb.py -------------------------------------------------------------------------------- /lib/datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/datasets/pascal_voc.py -------------------------------------------------------------------------------- /lib/fast_rcnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/fast_rcnn/bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/fast_rcnn/bbox_transform.py -------------------------------------------------------------------------------- /lib/fast_rcnn/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/fast_rcnn/config.py -------------------------------------------------------------------------------- /lib/fast_rcnn/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/fast_rcnn/nms_wrapper.py -------------------------------------------------------------------------------- /lib/fast_rcnn/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/fast_rcnn/test.py -------------------------------------------------------------------------------- /lib/fast_rcnn/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/fast_rcnn/train.py -------------------------------------------------------------------------------- /lib/networks/VGGnet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/networks/VGGnet_test.py -------------------------------------------------------------------------------- /lib/networks/VGGnet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/networks/VGGnet_train.py -------------------------------------------------------------------------------- /lib/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/networks/__init__.py -------------------------------------------------------------------------------- /lib/networks/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/networks/factory.py -------------------------------------------------------------------------------- /lib/networks/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/networks/network.py -------------------------------------------------------------------------------- /lib/prepare_training_data/ToVoc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/prepare_training_data/ToVoc.py -------------------------------------------------------------------------------- /lib/prepare_training_data/split_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/prepare_training_data/split_label.py -------------------------------------------------------------------------------- /lib/roi_data_layer/__init__.py: -------------------------------------------------------------------------------- 1 | from . import roidb -------------------------------------------------------------------------------- /lib/roi_data_layer/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/roi_data_layer/layer.py -------------------------------------------------------------------------------- /lib/roi_data_layer/minibatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/roi_data_layer/minibatch.py -------------------------------------------------------------------------------- /lib/roi_data_layer/roidb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/roi_data_layer/roidb.py -------------------------------------------------------------------------------- /lib/rpn_msr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/rpn_msr/anchor_target_layer_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/rpn_msr/anchor_target_layer_tf.py -------------------------------------------------------------------------------- /lib/rpn_msr/generate_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/rpn_msr/generate_anchors.py -------------------------------------------------------------------------------- /lib/rpn_msr/proposal_layer_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/rpn_msr/proposal_layer_tf.py -------------------------------------------------------------------------------- /lib/text_connector/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/text_connector/__init__.py -------------------------------------------------------------------------------- /lib/text_connector/detectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/text_connector/detectors.py -------------------------------------------------------------------------------- /lib/text_connector/other.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/text_connector/other.py -------------------------------------------------------------------------------- /lib/text_connector/text_connect_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/text_connector/text_connect_cfg.py -------------------------------------------------------------------------------- /lib/text_connector/text_proposal_connector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/text_connector/text_proposal_connector.py -------------------------------------------------------------------------------- /lib/text_connector/text_proposal_connector_oriented.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/text_connector/text_proposal_connector_oriented.py -------------------------------------------------------------------------------- /lib/text_connector/text_proposal_graph_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/text_connector/text_proposal_graph_builder.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/__init__.py -------------------------------------------------------------------------------- /lib/utils/bbox.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/bbox.c -------------------------------------------------------------------------------- /lib/utils/bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/bbox.pyx -------------------------------------------------------------------------------- /lib/utils/blob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/blob.py -------------------------------------------------------------------------------- /lib/utils/boxes_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/boxes_grid.py -------------------------------------------------------------------------------- /lib/utils/cython_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/cython_nms.c -------------------------------------------------------------------------------- /lib/utils/cython_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/cython_nms.pyx -------------------------------------------------------------------------------- /lib/utils/gpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/gpu_nms.c -------------------------------------------------------------------------------- /lib/utils/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/gpu_nms.cpp -------------------------------------------------------------------------------- /lib/utils/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/gpu_nms.hpp -------------------------------------------------------------------------------- /lib/utils/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/gpu_nms.pyx -------------------------------------------------------------------------------- /lib/utils/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/make.sh -------------------------------------------------------------------------------- /lib/utils/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/nms_kernel.cu -------------------------------------------------------------------------------- /lib/utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/setup.py -------------------------------------------------------------------------------- /lib/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/lib/utils/timer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hsddlz/faster-CTPN/HEAD/requirements.txt --------------------------------------------------------------------------------