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

update logger

This commit is contained in:
tink2123 2020-07-06 14:08:15 +08:00
parent 97a3af3bbf
commit 8d849cfa8c
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -46,7 +46,11 @@ class LMDBReader(object):
if "tps" in params: if "tps" in params:
self.ues_tps = True self.ues_tps = True
if "distort" in params: if "distort" in params:
self.use_distort = params['distort'] self.use_distort = params['distort'] and params['use_gpu']
if not params['use_gpu']:
logger.info(
"Distort operation can only support in GPU. Distort will be set to False."
)
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
@ -189,7 +193,11 @@ class SimpleReader(object):
if "tps" in params: if "tps" in params:
self.use_tps = True self.use_tps = True
if "distort" in params: if "distort" in params:
self.use_distort = params['distort'] self.use_distort = params['distort'] and params['use_gpu']
if not params['use_gpu']:
logger.info(
"Distort operation can only support in GPU.Distort will be set to False."
)
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