diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 5a524516a28bd3fa1f7a9f2800ebace7b9776088..a34e3a71bc80d53a7dbf3a9b122440b1aa3173ed 100755 --- 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)