diff --git a/configs/rec/rec_chinese_lite_train.yml b/configs/rec/rec_chinese_lite_train.yml index 3368876d21..9749c39955 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 7083eb75ed..f54c1d307e 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']: