add drop_score to args

4 years ago · a804a97c92
parent e696c7906d
commit a804a97c92
1 changed files with 5 additions and 4 deletions
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@ -71,6 +71,7 @@ def parse_args():
    parser.add_argument("--use_space_char", type=str2bool, default=True)
    parser.add_argument(
        "--vis_font_path", type=str, default="./doc/simfang.ttf")
+    parser.add_argument("--drop_score", type=float, default=0.5)

    # params for text classifier
    parser.add_argument("--use_angle_cls", type=str2bool, default=False)