From 8d849cfa8cd4fc728ee1f4cc73305654cbbbbd9a Mon Sep 17 00:00:00 2001 From: tink2123 Date: Mon, 6 Jul 2020 14:08:15 +0800 Subject: [PATCH] update logger --- ppocr/data/rec/dataset_traversal.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/ppocr/data/rec/dataset_traversal.py b/ppocr/data/rec/dataset_traversal.py index 23852db6..7083eb75 100755 --- a/ppocr/data/rec/dataset_traversal.py +++ b/ppocr/data/rec/dataset_traversal.py @@ -46,7 +46,11 @@ class LMDBReader(object): if "tps" in params: self.ues_tps = True 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': self.batch_size = params['train_batch_size_per_card'] self.drop_last = True @@ -189,7 +193,11 @@ class SimpleReader(object): if "tps" in params: self.use_tps = True 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': self.batch_size = params['train_batch_size_per_card'] self.drop_last = True -- GitLab