CLOSE data augment if use pdserving

This commit is contained in:
wangjiawei04 2020-09-22 21:08:31 +08:00
parent f7dbf2b702
commit 95f820b835
1 changed files with 1 additions and 1 deletions

View File

@ -22,7 +22,7 @@ logger = initial_logger()
import tools.infer.utility as utility
args = utility.parse_args()
if args.use_serving is False:
if args.use_pdserving is False:
from .data_augment import AugmentData
from .random_crop_data import RandomCropData
from .make_shrink_map import MakeShrinkMap