diff --git a/tools/infer/predict_det.py b/tools/infer/predict_det.py
index dec9964264a785300f588737331ce9be3df61a68..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"] = False
+            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 e842e7b050b2f1bb81572e9aa4f5739d9055ad96..3b92c093d8e57154c3298daa6b270529d19211aa 100755
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -47,6 +47,7 @@ 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_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)