2020-05-10 16:26:57 +08:00
|
|
|
Global:
|
|
|
|
|
algorithm: EAST
|
|
|
|
|
use_gpu: true
|
|
|
|
|
epoch_num: 100000
|
|
|
|
|
log_smooth_window: 20
|
|
|
|
|
print_batch_step: 5
|
2020-05-13 16:05:00 +08:00
|
|
|
save_model_dir: ./output/det_east/
|
2020-05-10 16:26:57 +08:00
|
|
|
save_epoch_step: 200
|
|
|
|
|
eval_batch_step: 5000
|
|
|
|
|
train_batch_size_per_card: 8
|
|
|
|
|
test_batch_size_per_card: 16
|
|
|
|
|
image_shape: [3, 512, 512]
|
|
|
|
|
reader_yml: ./configs/det/det_east_icdar15_reader.yml
|
2020-05-19 10:50:42 +08:00
|
|
|
pretrain_weights: ./pretrain_models/ResNet50_vd_ssld_pretrained/
|
2020-05-13 16:05:00 +08:00
|
|
|
save_res_path: ./output/det_east/predicts_east.txt
|
|
|
|
|
checkpoints:
|
|
|
|
|
save_inference_dir:
|
|
|
|
|
|
2020-05-10 16:26:57 +08:00
|
|
|
Architecture:
|
|
|
|
|
function: ppocr.modeling.architectures.det_model,DetModel
|
|
|
|
|
|
|
|
|
|
Backbone:
|
|
|
|
|
function: ppocr.modeling.backbones.det_resnet_vd,ResNet
|
|
|
|
|
layers: 50
|
|
|
|
|
|
|
|
|
|
Head:
|
|
|
|
|
function: ppocr.modeling.heads.det_east_head,EASTHead
|
|
|
|
|
model_name: large
|
|
|
|
|
|
|
|
|
|
Loss:
|
|
|
|
|
function: ppocr.modeling.losses.det_east_loss,EASTLoss
|
|
|
|
|
|
|
|
|
|
Optimizer:
|
|
|
|
|
function: ppocr.optimizer,AdamDecay
|
|
|
|
|
base_lr: 0.001
|
|
|
|
|
beta1: 0.9
|
|
|
|
|
beta2: 0.999
|
|
|
|
|
|
|
|
|
|
PostProcess:
|
|
|
|
|
function: ppocr.postprocess.east_postprocess,EASTPostPocess
|
|
|
|
|
score_thresh: 0.8
|
|
|
|
|
cover_thresh: 0.1
|
|
|
|
|
nms_thresh: 0.2
|
2020-05-19 10:50:42 +08:00
|
|
|
|
