From fa018e57b4a47468a2c13d8da32a79dbb8b277ae Mon Sep 17 00:00:00 2001
From: LDOUBLEV <liuvv0203@outlook.com>
Date: Tue, 30 Nov 2021 11:09:20 +0000
Subject: [PATCH] fix trt

---
 tools/infer/utility.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index bd9e14a657..c288166534 100644
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -211,7 +211,7 @@ def create_predictor(args, mode, logger):
                     "nearest_interp_v2_0.tmp_0": [1, 256, 2, 2]
                 }
                 max_input_shape = {
-                    "x": [1, 3, 1280, 1280],
+                    "x": [1, 3, 1536, 1536],
                     "conv2d_92.tmp_0": [1, 120, 400, 400],
                     "conv2d_91.tmp_0": [1, 24, 200, 200],
                     "conv2d_59.tmp_0": [1, 96, 400, 400],