forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #140 from tink2123/fix_typo 

fix typo
This commit is contained in:
xiaoting 2020-06-05 18:01:49 +08:00 committed by GitHub
parent 4fbcc57db2 aab46214fd
commit ce2674dcbe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
configs/rec/rec_r34_vd_tps_bilstm_attn.yml
View File

@ -13,6 +13,7 @@ Global:
max_text_length: 25 max_text_length: 25
character_type: en character_type: en
loss_type: attention loss_type: attention
tps: true
reader_yml: ./configs/rec/rec_benchmark_reader.yml reader_yml: ./configs/rec/rec_benchmark_reader.yml
pretrain_weights: pretrain_weights:
checkpoints: checkpoints:

1
configs/rec/rec_r34_vd_tps_bilstm_ctc.yml
View File

@ -13,6 +13,7 @@ Global:
max_text_length: 25 max_text_length: 25
character_type: en character_type: en
loss_type: ctc loss_type: ctc
tps: true
reader_yml: ./configs/rec/rec_benchmark_reader.yml reader_yml: ./configs/rec/rec_benchmark_reader.yml
pretrain_weights: pretrain_weights:
checkpoints: checkpoints:

2
ppocr/data/rec/dataset_traversal.py
View File

@ -184,7 +184,7 @@ class SimpleReader(object):
self.infer_img = params['infer_img'] self.infer_img = params['infer_img']
self.use_tps = False self.use_tps = False
if "tps" in params: if "tps" in params:
self.ues_tps = True self.use_tps = True
if params['mode'] == 'train': if params['mode'] == 'train':
self.batch_size = params['train_batch_size_per_card'] self.batch_size = params['train_batch_size_per_card']
self.drop_last = True self.drop_last = True