diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index 103dd9bd8dcf0c8c3fcca1acfc5d01a3134e28a3..7a4e8392c4dad4c63bc161c7c34e8976db0f66ff 100755
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -187,7 +187,7 @@ def create_predictor(args, mode, logger):
                 "nearest_interp_v2_0.tmp_0": [1, 256, 2, 2]
             }
             max_input_shape = {
-                "x": [1, 3, 2000, 2000],
+                "x": [1, 3, 1280, 1280],
                 "conv2d_92.tmp_0": [1, 120, 400, 400],
                 "conv2d_91.tmp_0": [1, 24, 200, 200],
                 "conv2d_59.tmp_0": [1, 96, 400, 400],
@@ -237,16 +237,16 @@ def create_predictor(args, mode, logger):
             opt_input_shape.update(opt_pact_shape)
         elif mode == "rec":
             min_input_shape = {"x": [1, 3, 32, 10]}
-            max_input_shape = {"x": [args.rec_batch_num, 3, 32, 2000]}
+            max_input_shape = {"x": [args.rec_batch_num, 3, 32, 1024]}
             opt_input_shape = {"x": [args.rec_batch_num, 3, 32, 320]}
         elif mode == "cls":
             min_input_shape = {"x": [1, 3, 48, 10]}
-            max_input_shape = {"x": [args.rec_batch_num, 3, 48, 2000]}
+            max_input_shape = {"x": [args.rec_batch_num, 3, 48, 1024]}
             opt_input_shape = {"x": [args.rec_batch_num, 3, 48, 320]}
         else:
             min_input_shape = {"x": [1, 3, 10, 10]}
-            max_input_shape = {"x": [1, 3, 1000, 1000]}
-            opt_input_shape = {"x": [1, 3, 500, 500]}
+            max_input_shape = {"x": [1, 3, 512, 512]}
+            opt_input_shape = {"x": [1, 3, 256, 256]}
         config.set_trt_dynamic_shape_info(min_input_shape, max_input_shape,
                                           opt_input_shape)