diff --git a/configs/rec/rec_chinese_reader.yml b/configs/rec/rec_chinese_reader.yml index 19561dd27e3f1bb0e4de3e2a0cf1c2b3aaf0557f..9b742753e7e6b7ad4cba7539d67cb692a134d052 100755 --- a/configs/rec/rec_chinese_reader.yml +++ b/configs/rec/rec_chinese_reader.yml @@ -2,12 +2,12 @@ TrainReader: reader_function: ppocr.data.rec.dataset_traversal,SimpleReader num_workers: 8 img_set_dir: ./train_data - label_file_path: ./train_data/gt_train.txt + label_file_path: ./train_data/train_label.txt EvalReader: reader_function: ppocr.data.rec.dataset_traversal,SimpleReader img_set_dir: ./train_data - label_file_path: ./train_data/gt_test.txt + label_file_path: ./train_data/test_label.txt TestReader: reader_function: ppocr.data.rec.dataset_traversal,SimpleReader diff --git a/configs/rec/rec_ic15_reader.yml b/configs/rec/rec_ic15_reader.yml new file mode 100755 index 0000000000000000000000000000000000000000..19561dd27e3f1bb0e4de3e2a0cf1c2b3aaf0557f --- /dev/null +++ b/configs/rec/rec_ic15_reader.yml @@ -0,0 +1,13 @@ +TrainReader: + reader_function: ppocr.data.rec.dataset_traversal,SimpleReader + num_workers: 8 + img_set_dir: ./train_data + label_file_path: ./train_data/gt_train.txt + +EvalReader: + reader_function: ppocr.data.rec.dataset_traversal,SimpleReader + img_set_dir: ./train_data + label_file_path: ./train_data/gt_test.txt + +TestReader: + reader_function: ppocr.data.rec.dataset_traversal,SimpleReader diff --git a/configs/rec/rec_ic15_train.yml b/configs/rec/rec_ic15_train.yml new file mode 100755 index 0000000000000000000000000000000000000000..05105f0445853b4ee32b720e1049f3c418f09334 --- /dev/null +++ b/configs/rec/rec_ic15_train.yml @@ -0,0 +1,42 @@ +Global: + algorithm: CRNN + use_gpu: true + epoch_num: 300 + log_smooth_window: 20 + print_batch_step: 10 + save_model_dir: output_ic15 + save_epoch_step: 3 + eval_batch_step: 2000 + train_batch_size_per_card: 256 + test_batch_size_per_card: 256 + image_shape: [3, 32, 100] + max_text_length: 25 + character_type: ch + character_dict_path: ./ppocr/utils/ic15_dict.txt + loss_type: ctc + reader_yml: ./configs/rec/rec_ic15_reader.yml + pretrain_weights: ./pretrain_models/best_accuracy + infer_img: ./infer_img + +Architecture: + function: ppocr.modeling.architectures.rec_model,RecModel + +Backbone: + function: ppocr.modeling.backbones.rec_mobilenet_v3,MobileNetV3 + scale: 0.5 + model_name: small + +Head: + function: ppocr.modeling.heads.rec_ctc_head,CTCPredict + encoder_type: rnn + SeqRNN: + hidden_size: 48 + +Loss: + function: ppocr.modeling.losses.rec_ctc_loss,CTCLoss + +Optimizer: + function: ppocr.optimizer,AdamDecay + base_lr: 0.001 + beta1: 0.9 + beta2: 0.999 diff --git a/tools/eval.py b/tools/eval.py index 3b176648812fff74e9878e0b789b6d6f66fc70f5..949f3a343e6f512995363d60c7b6cd3d2e9535f3 100755 --- a/tools/eval.py +++ b/tools/eval.py @@ -80,7 +80,7 @@ def main(): metrics = eval_det_run(exe, config, eval_info_dict, "test") else: reader_type = config['Global']['reader_yml'] - if "chinese" in reader_type: + if "benchmark" not in reader_type: eval_reader = reader_main(config=config, mode="eval") eval_info_dict = {'program': eval_program, \ 'reader': eval_reader, \