diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 1b5e6b5279d881d50e01a033678069206b6fa1a1..162f6cb757a6725e67a76409b6a50ae25b1a6dc8 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)