Skip to content

GitLab

"doc/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "42fc271ae29d7aae533c79d41e745efb43d3e6b6"
Unverified Commit 2ba66200 authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #1496 from LDOUBLEV/dyg_db 

adjust cls_batch_num to 6 for speed-up
parents 6bbd58aa a6fd8f80
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/infer/utility.py
View file @ 2ba66200
...@@ -78,7 +78,7 @@ def parse_args(): ...@@ -78,7 +78,7 @@ def parse_args():
parser.add_argument("--cls_model_dir", type=str) parser.add_argument("--cls_model_dir", type=str)
parser.add_argument("--cls_image_shape", type=str, default="3, 48, 192") 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("--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("--cls_thresh", type=float, default=0.9)
parser.add_argument("--enable_mkldnn", type=str2bool, default=False) parser.add_argument("--enable_mkldnn", type=str2bool, default=False)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment