From a6fd8f80664c458b11e8acea3cd120d30c8834fc Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Fri, 18 Dec 2020 11:11:02 +0800 Subject: [PATCH] adjust cls_batch_num to 6 for speed-up --- tools/infer/utility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 1b5e6b52..162f6cb7 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -78,7 +78,7 @@ def parse_args(): parser.add_argument("--cls_model_dir", type=str) parser.add_argument("--cls_image_shape", type=str, default="3, 48, 192") parser.add_argument("--label_list", type=list, default=['0', '180']) - parser.add_argument("--cls_batch_num", type=int, default=30) + parser.add_argument("--cls_batch_num", type=int, default=6) parser.add_argument("--cls_thresh", type=float, default=0.9) parser.add_argument("--enable_mkldnn", type=str2bool, default=False) -- GitLab