From f7dbf2b702247c4011373ad70245360c23732344 Mon Sep 17 00:00:00 2001 From: wangjiawei04 Date: Tue, 22 Sep 2020 21:06:45 +0800 Subject: [PATCH] CLOSE data augment if use pdserving --- ppocr/data/det/db_process.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/ppocr/data/det/db_process.py b/ppocr/data/det/db_process.py index 9534c59e..3136b3ec 100644 --- a/ppocr/data/det/db_process.py +++ b/ppocr/data/det/db_process.py @@ -20,10 +20,13 @@ import sys from ppocr.utils.utility import initial_logger, check_and_read_gif logger = initial_logger() -from .data_augment import AugmentData -from .random_crop_data import RandomCropData -from .make_shrink_map import MakeShrinkMap -from .make_border_map import MakeBorderMap +import tools.infer.utility as utility +args = utility.parse_args() +if args.use_serving is False: + from .data_augment import AugmentData + from .random_crop_data import RandomCropData + from .make_shrink_map import MakeShrinkMap + from .make_border_map import MakeBorderMap class DBProcessTrain(object): -- GitLab