rec_icdar15_train.yml 1.1 KB
Newer Older
T
tink2123 已提交
1 2 3
Global:
  algorithm: CRNN
  use_gpu: true
T
tink2123 已提交
4
  epoch_num: 3000
T
tink2123 已提交
5 6
  log_smooth_window: 20
  print_batch_step: 10
T
tink2123 已提交
7
  save_model_dir: ./output/rec_CRNN
T
tink2123 已提交
8 9
  save_epoch_step: 300
  eval_batch_step: 500
T
tink2123 已提交
10 11 12 13 14
  train_batch_size_per_card: 256
  test_batch_size_per_card: 256
  image_shape: [3, 32, 100]
  max_text_length: 25
  character_type: ch
T
tink2123 已提交
15 16
  #character_dict_path: ./ppocr/utils/ic15_dict.txt
  character_dict_path: /workspace/PaddleOCR/train_data/ic15_dict.txt
T
tink2123 已提交
17
  loss_type: ctc
T
tink2123 已提交
18
  reader_yml: ./configs/rec/rec_icdar15_reader.yml
T
tink2123 已提交
19
  pretrain_weights: /workspace/PaddleOCR/pretrain_models/rec_mv3_none_bilstm_ctc/best_accuracy 
T
tink2123 已提交
20 21
  checkpoints:
  save_inference_dir:
T
tink2123 已提交
22 23 24 25 26 27
Architecture:
  function: ppocr.modeling.architectures.rec_model,RecModel

Backbone:
  function: ppocr.modeling.backbones.rec_mobilenet_v3,MobileNetV3
  scale: 0.5
T
tink2123 已提交
28
  model_name: large
T
tink2123 已提交
29 30 31 32 33

Head:
  function: ppocr.modeling.heads.rec_ctc_head,CTCPredict
  encoder_type: rnn
  SeqRNN:
T
tink2123 已提交
34
    hidden_size: 96
T
tink2123 已提交
35 36 37 38 39 40
    
Loss:
  function: ppocr.modeling.losses.rec_ctc_loss,CTCLoss

Optimizer:
  function: ppocr.optimizer,AdamDecay
T
tink2123 已提交
41
  base_lr: 0.0005
T
tink2123 已提交
42 43
  beta1: 0.9
  beta2: 0.999