det_r50_vd_east.yml 1.0 KB
Newer Older
L
LDOUBLEV 已提交
1 2 3 4 5 6
Global:
  algorithm: EAST
  use_gpu: true
  epoch_num: 100000
  log_smooth_window: 20
  print_batch_step: 5
7
  save_model_dir: ./output/det_east/
L
LDOUBLEV 已提交
8 9 10 11 12
  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]
T
tink2123 已提交
13
  drop_last: false
L
LDOUBLEV 已提交
14
  reader_yml: ./configs/det/det_east_icdar15_reader.yml
T
tink2123 已提交
15
  pretrain_weights: ./pretrain_models/ResNet50_vd_ssld_pretrained/
16 17 18 19
  save_res_path: ./output/det_east/predicts_east.txt
  checkpoints:
  save_inference_dir:

L
LDOUBLEV 已提交
20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
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
T
tink2123 已提交
45