Skip to content

P
PaddleOCR

PaddlePaddle / PaddleOCR
大约 1 年 前同步成功

通知 1528
Star 32962
Fork 6643
P
PaddleOCR
提交 8d849cfa 编写于 作者: T tink2123
浏览文件
操作

update logger

上级 97a3af3b
隐藏空白更改
内联 并排
Showing with 10 addition and 2 deletion
ppocr/data/rec/dataset_traversal.py
浏览文件 @ 8d849cfa
...@@ -46,7 +46,11 @@ class LMDBReader(object): ...@@ -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): ...@@ -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
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或