feat: added bw image input support to dd_bench

jolibrain · Sep 16, 2020 · 6e558d6 · 6e558d6
1 parent 8bf52c0
commit 6e558d6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: