You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
PaddleOCR/configs/rec/rec_icdar15_train.yml

50 lines
1.1 KiB

5 years ago
Global:
algorithm: CRNN
5 years ago
use_gpu: true
5 years ago
epoch_num: 1000
5 years ago
log_smooth_window: 20
print_batch_step: 10
5 years ago
save_model_dir: ./output/rec_CRNN
save_epoch_step: 300
eval_batch_step: 500
5 years ago
train_batch_size_per_card: 256
test_batch_size_per_card: 256
5 years ago
image_shape: [3, 32, 100]
max_text_length: 25
5 years ago
character_type: en
5 years ago
loss_type: ctc
distort: true
debug: false
reader_yml: ./configs/rec/rec_icdar15_reader.yml
5 years ago
pretrain_weights: ./pretrain_models/rec_mv3_none_bilstm_ctc/best_accuracy
checkpoints:
save_inference_dir:
infer_img:
5 years ago
Architecture:
function: ppocr.modeling.architectures.rec_model,RecModel
Backbone:
function: ppocr.modeling.backbones.rec_mobilenet_v3,MobileNetV3
scale: 0.5
5 years ago
model_name: large
5 years ago
Head:
function: ppocr.modeling.heads.rec_ctc_head,CTCPredict
encoder_type: rnn
SeqRNN:
5 years ago
hidden_size: 96
5 years ago
Loss:
function: ppocr.modeling.losses.rec_ctc_loss,CTCLoss
Optimizer:
function: ppocr.optimizer,AdamDecay
5 years ago
base_lr: 0.0005
5 years ago
beta1: 0.9
beta2: 0.999
5 years ago
decay:
function: cosine_decay
step_each_epoch: 20
total_epoch: 1000