diff --git a/tools/infer/predict_e2e.py b/tools/infer/predict_e2e.py
index a5c57914173b7d44c9479f7bb120e4ff409b91e3..8b94f24a9ffa16ca5683795afa6392fa23c24a94 100755
--- a/tools/infer/predict_e2e.py
+++ b/tools/infer/predict_e2e.py
@@ -66,6 +66,7 @@ class TextE2E(object):
             postprocess_params["score_thresh"] = args.e2e_pgnet_score_thresh
             postprocess_params["character_dict_path"] = args.e2e_char_dict_path
             postprocess_params["valid_set"] = args.e2e_pgnet_valid_set
+            postprocess_params["mode"] = args.e2e_pgnet_mode
             self.e2e_pgnet_polygon = args.e2e_pgnet_polygon
         else:
             logger.info("unknown e2e_algorithm:{}".format(self.e2e_algorithm))
diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index b273eaf3258421d5c5c30c132f99e78f9f0999ba..9fa51d80b41ae80abe206ee779accd44d49cebba 100755
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -86,6 +86,7 @@ def parse_args():
         "--e2e_char_dict_path", type=str, default="./ppocr/utils/ic15_dict.txt")
     parser.add_argument("--e2e_pgnet_valid_set", type=str, default='totaltext')
     parser.add_argument("--e2e_pgnet_polygon", type=bool, default=True)
+    parser.add_argument("--e2e_pgnet_mode", type=str, default='fast')
 
     # params for text classifier
     parser.add_argument("--use_angle_cls", type=str2bool, default=False)