diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index 75a8b2345d50f09d6548ca6f1dac32105a707ee5..6d9935a70e79bb20c5f6380783911ef141b0be17 100644
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -55,7 +55,7 @@ def init_args():
     parser.add_argument("--max_batch_size", type=int, default=10)
     parser.add_argument("--use_dilation", type=str2bool, default=False)
     parser.add_argument("--det_db_score_mode", type=str, default="fast")
-    parser.add_argument("--visual_output", type=str2bool, default=False)
+    parser.add_argument("--vis_seg_map", type=str2bool, 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)