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