diff --git a/configs/rec/rec_r34_vd_tps_bilstm_attn.yml b/configs/rec/rec_r34_vd_tps_bilstm_attn.yml index 405082bdbec0f4b9ac0c885801963e9261c43e6e..dfcd97fad67081a7ed04ad1d67ff298c9f553b0c 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 517322c374a7faf80d6e2b69b7f3e8b2dbb5b5af..574a088cc024541e086bedc0bc0a52082e9e7eb2 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 71b488f4eca6a3ba45354a62c7af1b65d944df25..e57717d937c9dbbfb64007bdc2ed81458559ca6a 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