rec_r50fpn_vd_none_srn.yml 1.0 KB
Newer Older
T
tink2123 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
Global:
  algorithm: SRN
  use_gpu: true
  epoch_num: 72
  log_smooth_window: 20
  print_batch_step: 10
  save_model_dir: output/rec_pvam_withrotate
  save_epoch_step: 1
  eval_batch_step: 8000
  train_batch_size_per_card: 64
  test_batch_size_per_card: 1
  image_shape: [1, 64, 256]
  max_text_length: 25
  character_type: en
  loss_type: srn
  num_heads: 8
  average_window: 0.15
  max_average_window: 15625
  min_average_window: 10000
T
tink2123 已提交
20
  reader_yml: ./configs/rec/rec_benchmark_reader.yml
T
tink2123 已提交
21 22 23
  pretrain_weights: 
  checkpoints:
  save_inference_dir:
T
tink2123 已提交
24 25
  infer_img:

T
tink2123 已提交
26 27 28 29
Architecture:
  function: ppocr.modeling.architectures.rec_model,RecModel

Backbone:
T
tink2123 已提交
30
  function: ppocr.modeling.backbones.rec_resnet_fpn,ResNet
T
tink2123 已提交
31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
  layers: 50
 
Head:
  function: ppocr.modeling.heads.rec_srn_all_head,SRNPredict
  encoder_type: rnn
  num_encoder_TUs: 2
  num_decoder_TUs: 4
  hidden_dims: 512
  SeqRNN:
    hidden_size: 256
    
Loss:
  function: ppocr.modeling.losses.rec_srn_loss,SRNLoss

Optimizer:
  function: ppocr.optimizer,AdamDecay
  base_lr: 0.0001
  beta1: 0.9
  beta2: 0.999