diff --git a/clients/python/dd_bench.py b/clients/python/dd_bench.py index deff0c910..f75ea43cc 100644 --- a/clients/python/dd_bench.py +++ b/clients/python/dd_bench.py @@ -36,6 +36,7 @@ parser.add_argument('--sname',help='service name') parser.add_argument('--img-width',help='image width',type=int,default=224) parser.add_argument('--img-height',help='image height',type=int,default=224) +parser.add_argument('--bw',help='whether images are bw',action='store_true') parser.add_argument('--gpu',help='whether to bench GPU',action='store_true') parser.add_argument('--gpuid',help='gpu id to use',type=int,default=0) parser.add_argument('--cpu',help='whether to bench CPU',action='store_true') @@ -74,7 +75,7 @@ def service_create(bs): description = 'image classification service' mllib = args.mllib model = {'repository':args.create} - parameters_input = {'connector':'image','width':args.img_width,'height':args.img_height} + parameters_input = {'connector':'image','width':args.img_width,'height':args.img_height,'bw':args.bw} if args.segmentation: parameters_input['segmentation'] = True if args.dla: