diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py index 26febf1c0da21f8b53114e2e236a5ea246e823a8..76c6a4478ae37e650901c8b8704e15e4a94911c4 100755 --- a/tools/infer/predict_det.py +++ b/tools/infer/predict_det.py @@ -64,7 +64,7 @@ class TextDetector(object): postprocess_params["box_thresh"] = args.det_db_box_thresh postprocess_params["max_candidates"] = 1000 postprocess_params["unclip_ratio"] = args.det_db_unclip_ratio - postprocess_params["use_dilation"] = args.use_dialtion + postprocess_params["use_dilation"] = args.use_dilation elif self.det_algorithm == "EAST": postprocess_params['name'] = 'EASTPostProcess' postprocess_params["score_thresh"] = args.det_east_score_thresh diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 18dedbda99c11d5b877388c1c516b2548b1a8335..cad1a01736ee0d1df98923fb208c2734c964104c 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -47,8 +47,8 @@ def parse_args(): parser.add_argument("--det_db_box_thresh", type=float, default=0.5) parser.add_argument("--det_db_unclip_ratio", type=float, default=1.6) parser.add_argument("--max_batch_size", type=int, default=10) - parser.add_argument("--use_dialtion", type=bool, default=False) - + parser.add_argument("--use_dilation", type=bool, default=False) + # EAST parmas parser.add_argument("--det_east_score_thresh", type=float, default=0.8) parser.add_argument("--det_east_cover_thresh", type=float, default=0.1)