2020-05-12 15:21:59 +08:00
|
|
|
Global:
|
|
|
|
|
algorithm: CRNN
|
2020-06-05 11:01:22 +08:00
|
|
|
use_gpu: true
|
2020-05-14 14:33:15 +08:00
|
|
|
epoch_num: 1000
|
2020-05-12 15:21:59 +08:00
|
|
|
log_smooth_window: 20
|
|
|
|
|
print_batch_step: 10
|
2020-05-13 20:27:45 +08:00
|
|
|
save_model_dir: ./output/rec_CRNN
|
2020-05-13 17:03:29 +08:00
|
|
|
save_epoch_step: 300
|
|
|
|
|
eval_batch_step: 500
|
2020-06-05 11:01:22 +08:00
|
|
|
train_batch_size_per_card: 256
|
|
|
|
|
test_batch_size_per_card: 256
|
2020-05-12 15:21:59 +08:00
|
|
|
image_shape: [3, 32, 100]
|
|
|
|
|
max_text_length: 25
|
2020-05-14 00:15:52 +08:00
|
|
|
character_type: en
|
2020-05-12 15:21:59 +08:00
|
|
|
loss_type: ctc
|
2020-07-06 13:53:12 +08:00
|
|
|
distort: true
|
2020-07-11 12:14:05 +08:00
|
|
|
debug: false
|
2020-05-12 19:55:16 +08:00
|
|
|
reader_yml: ./configs/rec/rec_icdar15_reader.yml
|
2020-06-05 11:01:22 +08:00
|
|
|
pretrain_weights: ./pretrain_models/rec_mv3_none_bilstm_ctc/best_accuracy
|
2020-05-12 21:17:09 +08:00
|
|
|
checkpoints:
|
|
|
|
|
save_inference_dir:
|
2020-06-02 19:03:27 +08:00
|
|
|
infer_img:
|
|
|
|
|
|
2020-05-12 15:21:59 +08:00
|
|
|
Architecture:
|
|
|
|
|
function: ppocr.modeling.architectures.rec_model,RecModel
|
|
|
|
|
|
|
|
|
|
Backbone:
|
|
|
|
|
function: ppocr.modeling.backbones.rec_mobilenet_v3,MobileNetV3
|
|
|
|
|
scale: 0.5
|
2020-05-13 23:16:25 +08:00
|
|
|
model_name: large
|
2020-05-12 15:21:59 +08:00
|
|
|
|
|
|
|
|
Head:
|
|
|
|
|
function: ppocr.modeling.heads.rec_ctc_head,CTCPredict
|
|
|
|
|
encoder_type: rnn
|
|
|
|
|
SeqRNN:
|
2020-05-13 23:16:25 +08:00
|
|
|
hidden_size: 96
|
2020-05-12 15:21:59 +08:00
|
|
|
|
|
|
|
|
Loss:
|
|
|
|
|
function: ppocr.modeling.losses.rec_ctc_loss,CTCLoss
|
|
|
|
|
|
|
|
|
|
Optimizer:
|
|
|
|
|
function: ppocr.optimizer,AdamDecay
|
2020-05-13 20:27:45 +08:00
|
|
|
base_lr: 0.0005
|
2020-05-12 15:21:59 +08:00
|
|
|
beta1: 0.9
|
|
|
|
|
beta2: 0.999
|
2020-06-10 15:41:16 +08:00
|
|
|
decay:
|
|
|
|
|
function: cosine_decay
|
|
|
|
|
step_each_epoch: 20
|
|
|
|
|
total_epoch: 1000
|
