From aab46214fd3e1380475e28e0969cf64e9f5bd25a Mon Sep 17 00:00:00 2001 From: tink2123 Date: Fri, 5 Jun 2020 18:00:52 +0800 Subject: [PATCH] fix typo --- configs/rec/rec_r34_vd_tps_bilstm_attn.yml | 1 + configs/rec/rec_r34_vd_tps_bilstm_ctc.yml | 1 + ppocr/data/rec/dataset_traversal.py | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/configs/rec/rec_r34_vd_tps_bilstm_attn.yml b/configs/rec/rec_r34_vd_tps_bilstm_attn.yml index 405082bd..dfcd97fa 100755 --- a/configs/rec/rec_r34_vd_tps_bilstm_attn.yml +++ b/configs/rec/rec_r34_vd_tps_bilstm_attn.yml @@ -13,6 +13,7 @@ Global: max_text_length: 25 character_type: en loss_type: attention + tps: true reader_yml: ./configs/rec/rec_benchmark_reader.yml pretrain_weights: checkpoints: diff --git a/configs/rec/rec_r34_vd_tps_bilstm_ctc.yml b/configs/rec/rec_r34_vd_tps_bilstm_ctc.yml index 517322c3..574a088c 100755 --- a/configs/rec/rec_r34_vd_tps_bilstm_ctc.yml +++ b/configs/rec/rec_r34_vd_tps_bilstm_ctc.yml @@ -13,6 +13,7 @@ Global: max_text_length: 25 character_type: en loss_type: ctc + tps: true reader_yml: ./configs/rec/rec_benchmark_reader.yml pretrain_weights: checkpoints: diff --git a/ppocr/data/rec/dataset_traversal.py b/ppocr/data/rec/dataset_traversal.py index 71b488f4..e57717d9 100755 --- a/ppocr/data/rec/dataset_traversal.py +++ b/ppocr/data/rec/dataset_traversal.py @@ -184,7 +184,7 @@ class SimpleReader(object): self.infer_img = params['infer_img'] self.use_tps = False if "tps" in params: - self.ues_tps = True + self.use_tps = True if params['mode'] == 'train': self.batch_size = params['train_batch_size_per_card'] self.drop_last = True -- GitLab