Merge pull request #1496 from LDOUBLEV/dyg_db

adjust cls_batch_num to 6 for speed-up
PaddlePaddle · Dec 18, 2020 · 2ba6620 · 2ba6620
2 parents 6bbd58a + a6fd8f8
commit 2ba6620
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/infer/utility.py b/tools/infer/utility.py
@@ -78,7 +78,7 @@ def str2bool(v):
  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)