Skip to content

GitLab

"docs/source/api/vscode:/vscode.git/clone" did not exist on "2625fb59dc7a3f03516f0c6c5c0cdda18ef0ef5b"
Commit 7e88895f authored by tink2123's avatar tink2123
Browse files

add default value for distort

parent f6d6249c
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
configs/rec/rec_chinese_lite_train.yml
View file @ 7e88895f
...@@ -14,7 +14,7 @@ Global: ...@@ -14,7 +14,7 @@ Global:
character_type: ch character_type: ch
character_dict_path: ./ppocr/utils/ppocr_keys_v1.txt character_dict_path: ./ppocr/utils/ppocr_keys_v1.txt
loss_type: ctc loss_type: ctc
distort: true distort: false
reader_yml: ./configs/rec/rec_chinese_reader.yml reader_yml: ./configs/rec/rec_chinese_reader.yml
pretrain_weights: pretrain_weights:
checkpoints: checkpoints:
......
ppocr/data/rec/dataset_traversal.py
View file @ 7e88895f
...@@ -45,6 +45,7 @@ class LMDBReader(object): ...@@ -45,6 +45,7 @@ class LMDBReader(object):
self.use_tps = False self.use_tps = False
if "tps" in params: if "tps" in params:
self.ues_tps = True self.ues_tps = True
self.use_distort = False
if "distort" in params: if "distort" in params:
self.use_distort = params['distort'] and params['use_gpu'] self.use_distort = params['distort'] and params['use_gpu']
if not params['use_gpu']: if not params['use_gpu']:
...@@ -192,6 +193,7 @@ class SimpleReader(object): ...@@ -192,6 +193,7 @@ class SimpleReader(object):
self.use_tps = False self.use_tps = False
if "tps" in params: if "tps" in params:
self.use_tps = True self.use_tps = True
self.use_distort = False
if "distort" in params: if "distort" in params:
self.use_distort = params['distort'] and params['use_gpu'] self.use_distort = params['distort'] and params['use_gpu']
if not params['use_gpu']: if not params['use_gpu']:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment