diff --git a/ppocr/data/det/db_process.py b/ppocr/data/det/db_process.py index 9534c59ef69d830a8d991f421539c5e4e5bb3d39..3136b3ec873a40c6d21b62b12a4cf906bd6ce2d9 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):