set dilation as False

4 years ago · cf550d0524
parent bbd7665c47
commit cf550d0524
1 changed files with 1 additions and 1 deletions
--- 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"] = True
+            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