diff --git a/configs/rec/rec_chinese_lite_train.yml b/configs/rec/rec_chinese_lite_train.yml index 3368876d217c141aeae6b6d326b913ed99bb3f51..9749c39955a152120afd4f295fb3e2056d07071f 100755 --- a/configs/rec/rec_chinese_lite_train.yml +++ b/configs/rec/rec_chinese_lite_train.yml @@ -14,7 +14,7 @@ Global: character_type: ch character_dict_path: ./ppocr/utils/ppocr_keys_v1.txt loss_type: ctc - distort: true + distort: false reader_yml: ./configs/rec/rec_chinese_reader.yml pretrain_weights: checkpoints: diff --git a/ppocr/data/rec/dataset_traversal.py b/ppocr/data/rec/dataset_traversal.py index 7083eb75ed642f0ca829ada701e64c45d636877b..f54c1d307ee2545c69a2664b149b69e50b4f8490 100755 --- a/ppocr/data/rec/dataset_traversal.py +++ b/ppocr/data/rec/dataset_traversal.py @@ -45,6 +45,7 @@ class LMDBReader(object): self.use_tps = False if "tps" in params: self.ues_tps = True + self.use_distort = False if "distort" in params: self.use_distort = params['distort'] and params['use_gpu'] if not params['use_gpu']: @@ -192,6 +193,7 @@ class SimpleReader(object): self.use_tps = False if "tps" in params: self.use_tps = True + self.use_distort = False if "distort" in params: self.use_distort = params['distort'] and params['use_gpu'] if not params['use_gpu']: