Skip to content

GitLab

Unverified Commit 77cc1d5c authored by Double_V's avatar Double_V Committed by GitHub
Browse files

Merge pull request #2021 from LDOUBLEV/trt_cpp 

set dilation as False
parents f65a5f6e cf550d05
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tools/infer/predict_det.py
View file @ 77cc1d5c
...@@ -64,7 +64,7 @@ class TextDetector(object): ...@@ -64,7 +64,7 @@ class TextDetector(object):
postprocess_params["box_thresh"] = args.det_db_box_thresh postprocess_params["box_thresh"] = args.det_db_box_thresh
postprocess_params["max_candidates"] = 1000 postprocess_params["max_candidates"] = 1000
postprocess_params["unclip_ratio"] = args.det_db_unclip_ratio postprocess_params["unclip_ratio"] = args.det_db_unclip_ratio
postprocess_params["use_dilation"] = True postprocess_params["use_dilation"] = args.use_dilation
elif self.det_algorithm == "EAST": elif self.det_algorithm == "EAST":
postprocess_params['name'] = 'EASTPostProcess' postprocess_params['name'] = 'EASTPostProcess'
postprocess_params["score_thresh"] = args.det_east_score_thresh postprocess_params["score_thresh"] = args.det_east_score_thresh
......
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