├── README.md ├── ckpt └── README.md ├── data ├── News_info_test.txt ├── News_pic_label_train.txt ├── README.md ├── id_to_item_small.pkl ├── item_to_id_small.pkl ├── stopword.txt ├── test_images │ ├── P0000001.JPEG │ └── P0000002.JPEG └── train_images │ ├── P0000001.JPEG │ └── P0000002.JPEG ├── img ├── example.png ├── model.png ├── ocr.png ├── pesudo.png └── stacking.png └── src ├── README.md ├── __init__.py ├── config.py ├── model ├── __init__.py ├── attention_model.py ├── capsule_model.py ├── catboost_model.py ├── covlstm_model.py ├── dpcnn_model.py ├── lightgbm_model.py ├── lstmcov_model.py ├── lstmgru_model.py ├── model_basic.py ├── model_component.py ├── snapshot.py ├── textCNN_model.py └── xgboost_model.py ├── ocr ├── LICENSE ├── README.md ├── ctpn │ ├── README.md │ ├── __init__.py │ ├── checkpoints │ │ └── test │ ├── ctpn │ │ ├── demo.py │ │ ├── text.yml │ │ └── train_net.py │ ├── data │ │ ├── demo │ │ │ └── 010.png │ │ ├── oriented_results │ │ │ └── 009.jpg │ │ └── pretrain_model │ │ │ └── VGG_imagenet.npy │ ├── lib │ │ ├── __init__.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── ds_utils.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 │ │ ├── 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 │ │ │ ├── bbox.so │ │ │ ├── blob.py │ │ │ ├── boxes_grid.py │ │ │ ├── cython_nms.c │ │ │ ├── cython_nms.pyx │ │ │ ├── cython_nms.so │ │ │ ├── gpu_nms.c │ │ │ ├── gpu_nms.cpp │ │ │ ├── gpu_nms.hpp │ │ │ ├── gpu_nms.pyx │ │ │ ├── make.sh │ │ │ ├── make_cpu.sh │ │ │ ├── nms_kernel.cu │ │ │ ├── setup.py │ │ │ ├── setup_cpu.py │ │ │ └── timer.py │ ├── prepare_training_data │ │ ├── ToVoc.py │ │ └── split_label.py │ └── text_detect.py ├── demo.py ├── demo │ ├── demo_detect.jpg │ └── demo_rec.jpg ├── densenet │ ├── __init__.py │ ├── densenet.py │ ├── keys.py │ ├── model.py │ └── models │ │ └── weights_densenet.h5 ├── ocr.py ├── setup.sh ├── test_images │ └── demo.jpg └── train │ ├── char_std_5990.txt │ ├── densenet.py │ ├── images │ └── .gitkeep │ ├── models │ └── pretrain_model │ │ └── .gitkeep │ └── train.py ├── preprocess ├── EDA&Extract.ipynb ├── cut_word.py ├── generate_text_feature.py ├── msyh.ttf ├── ocr_features.py ├── parse_html.py ├── process_features.py └── train_tfidf.py ├── stacking.ipynb └── train&predict.ipynb /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/README.md -------------------------------------------------------------------------------- /ckpt/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/News_info_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/News_info_test.txt -------------------------------------------------------------------------------- /data/News_pic_label_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/News_pic_label_train.txt -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/id_to_item_small.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/id_to_item_small.pkl -------------------------------------------------------------------------------- /data/item_to_id_small.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/item_to_id_small.pkl -------------------------------------------------------------------------------- /data/stopword.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/stopword.txt -------------------------------------------------------------------------------- /data/test_images/P0000001.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/test_images/P0000001.JPEG -------------------------------------------------------------------------------- /data/test_images/P0000002.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/test_images/P0000002.JPEG -------------------------------------------------------------------------------- /data/train_images/P0000001.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/train_images/P0000001.JPEG -------------------------------------------------------------------------------- /data/train_images/P0000002.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/data/train_images/P0000002.JPEG -------------------------------------------------------------------------------- /img/example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/img/example.png -------------------------------------------------------------------------------- /img/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/img/model.png -------------------------------------------------------------------------------- /img/ocr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/img/ocr.png -------------------------------------------------------------------------------- /img/pesudo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/img/pesudo.png -------------------------------------------------------------------------------- /img/stacking.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/img/stacking.png -------------------------------------------------------------------------------- /src/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/config.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/__init__.py -------------------------------------------------------------------------------- /src/model/attention_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/attention_model.py -------------------------------------------------------------------------------- /src/model/capsule_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/capsule_model.py -------------------------------------------------------------------------------- /src/model/catboost_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/catboost_model.py -------------------------------------------------------------------------------- /src/model/covlstm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/covlstm_model.py -------------------------------------------------------------------------------- /src/model/dpcnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/dpcnn_model.py -------------------------------------------------------------------------------- /src/model/lightgbm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/lightgbm_model.py -------------------------------------------------------------------------------- /src/model/lstmcov_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/lstmcov_model.py -------------------------------------------------------------------------------- /src/model/lstmgru_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/lstmgru_model.py -------------------------------------------------------------------------------- /src/model/model_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/model_basic.py -------------------------------------------------------------------------------- /src/model/model_component.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/model_component.py -------------------------------------------------------------------------------- /src/model/snapshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/snapshot.py -------------------------------------------------------------------------------- /src/model/textCNN_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/textCNN_model.py -------------------------------------------------------------------------------- /src/model/xgboost_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/model/xgboost_model.py -------------------------------------------------------------------------------- /src/ocr/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/LICENSE -------------------------------------------------------------------------------- /src/ocr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/README.md -------------------------------------------------------------------------------- /src/ocr/ctpn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/README.md -------------------------------------------------------------------------------- /src/ocr/ctpn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ocr/ctpn/checkpoints/test: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ocr/ctpn/ctpn/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/ctpn/demo.py -------------------------------------------------------------------------------- /src/ocr/ctpn/ctpn/text.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/ctpn/text.yml -------------------------------------------------------------------------------- /src/ocr/ctpn/ctpn/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/ctpn/train_net.py -------------------------------------------------------------------------------- /src/ocr/ctpn/data/demo/010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/data/demo/010.png -------------------------------------------------------------------------------- /src/ocr/ctpn/data/oriented_results/009.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/data/oriented_results/009.jpg -------------------------------------------------------------------------------- /src/ocr/ctpn/data/pretrain_model/VGG_imagenet.npy: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/__init__.py: -------------------------------------------------------------------------------- 1 | from . import fast_rcnn 2 | -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/datasets/__init__.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/datasets/ds_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/datasets/ds_utils.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/datasets/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/datasets/factory.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/datasets/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/datasets/imdb.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/datasets/pascal_voc.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/fast_rcnn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/fast_rcnn/__init__.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/fast_rcnn/bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/fast_rcnn/bbox_transform.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/fast_rcnn/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/fast_rcnn/config.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/fast_rcnn/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/fast_rcnn/nms_wrapper.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/fast_rcnn/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/fast_rcnn/test.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/fast_rcnn/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/fast_rcnn/train.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/networks/VGGnet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/networks/VGGnet_test.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/networks/VGGnet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/networks/VGGnet_train.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/networks/__init__.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/networks/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/networks/factory.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/networks/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/networks/network.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/roi_data_layer/__init__.py: -------------------------------------------------------------------------------- 1 | from . import roidb -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/roi_data_layer/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/roi_data_layer/layer.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/roi_data_layer/minibatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/roi_data_layer/minibatch.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/roi_data_layer/roidb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/roi_data_layer/roidb.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/rpn_msr/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/rpn_msr/anchor_target_layer_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/rpn_msr/anchor_target_layer_tf.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/rpn_msr/generate_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/rpn_msr/generate_anchors.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/rpn_msr/proposal_layer_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/rpn_msr/proposal_layer_tf.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/text_connector/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/text_connector/__init__.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/text_connector/detectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/text_connector/detectors.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/text_connector/other.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/text_connector/other.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/text_connector/text_connect_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/text_connector/text_connect_cfg.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/text_connector/text_proposal_connector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/text_connector/text_proposal_connector.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/text_connector/text_proposal_connector_oriented.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/text_connector/text_proposal_connector_oriented.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/text_connector/text_proposal_graph_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/text_connector/text_proposal_graph_builder.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/__init__.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/bbox.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/bbox.c -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/bbox.pyx -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/bbox.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/bbox.so -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/blob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/blob.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/boxes_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/boxes_grid.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/cython_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/cython_nms.c -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/cython_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/cython_nms.pyx -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/cython_nms.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/cython_nms.so -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/gpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/gpu_nms.c -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/gpu_nms.cpp -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/gpu_nms.hpp -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/gpu_nms.pyx -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/make.sh -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/make_cpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/make_cpu.sh -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/nms_kernel.cu -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/setup.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/setup_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/setup_cpu.py -------------------------------------------------------------------------------- /src/ocr/ctpn/lib/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/lib/utils/timer.py -------------------------------------------------------------------------------- /src/ocr/ctpn/prepare_training_data/ToVoc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/prepare_training_data/ToVoc.py -------------------------------------------------------------------------------- /src/ocr/ctpn/prepare_training_data/split_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/prepare_training_data/split_label.py -------------------------------------------------------------------------------- /src/ocr/ctpn/text_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ctpn/text_detect.py -------------------------------------------------------------------------------- /src/ocr/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/demo.py -------------------------------------------------------------------------------- /src/ocr/demo/demo_detect.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/demo/demo_detect.jpg -------------------------------------------------------------------------------- /src/ocr/demo/demo_rec.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/demo/demo_rec.jpg -------------------------------------------------------------------------------- /src/ocr/densenet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ocr/densenet/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/densenet/densenet.py -------------------------------------------------------------------------------- /src/ocr/densenet/keys.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/densenet/keys.py -------------------------------------------------------------------------------- /src/ocr/densenet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/densenet/model.py -------------------------------------------------------------------------------- /src/ocr/densenet/models/weights_densenet.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/densenet/models/weights_densenet.h5 -------------------------------------------------------------------------------- /src/ocr/ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/ocr.py -------------------------------------------------------------------------------- /src/ocr/setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/setup.sh -------------------------------------------------------------------------------- /src/ocr/test_images/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/test_images/demo.jpg -------------------------------------------------------------------------------- /src/ocr/train/char_std_5990.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/train/char_std_5990.txt -------------------------------------------------------------------------------- /src/ocr/train/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/train/densenet.py -------------------------------------------------------------------------------- /src/ocr/train/images/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ocr/train/models/pretrain_model/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ocr/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/ocr/train/train.py -------------------------------------------------------------------------------- /src/preprocess/EDA&Extract.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/preprocess/EDA&Extract.ipynb -------------------------------------------------------------------------------- /src/preprocess/cut_word.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/preprocess/cut_word.py -------------------------------------------------------------------------------- /src/preprocess/generate_text_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/preprocess/generate_text_feature.py -------------------------------------------------------------------------------- /src/preprocess/msyh.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/preprocess/msyh.ttf -------------------------------------------------------------------------------- /src/preprocess/ocr_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/preprocess/ocr_features.py -------------------------------------------------------------------------------- /src/preprocess/parse_html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/preprocess/parse_html.py -------------------------------------------------------------------------------- /src/preprocess/process_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/preprocess/process_features.py -------------------------------------------------------------------------------- /src/preprocess/train_tfidf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/preprocess/train_tfidf.py -------------------------------------------------------------------------------- /src/stacking.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/stacking.ipynb -------------------------------------------------------------------------------- /src/train&predict.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanzecheng/SOHU_competition/HEAD/src/train&predict.ipynb --------------------------------------------------------------------------------