[Enhancement] Add 320x320 static config for tensorrt (#1689)

* add trt static 320 config for rtmdet * only add on detection folder
open-mmlab · Jan 31, 2023 · 2ceabd0 · 2ceabd0
1 parent 8062e4e
commit 2ceabd0
Show file tree

Hide file tree

Showing 3 changed files with 42 additions and 0 deletions.
diff --git a/configs/mmdet/detection/detection_tensorrt-fp16_static-320x320.py b/configs/mmdet/detection/detection_tensorrt-fp16_static-320x320.py
@@ -0,0 +1,14 @@
+_base_ = ['../_base_/base_static.py', '../../_base_/backends/tensorrt-fp16.py']
+
+onnx_config = dict(input_shape=(320, 320))
+
+backend_config = dict(
+    common_config=dict(max_workspace_size=1 << 30),
+    model_inputs=[
+        dict(
+            input_shapes=dict(
+                input=dict(
+                    min_shape=[1, 3, 320, 320],
+                    opt_shape=[1, 3, 320, 320],
+                    max_shape=[1, 3, 320, 320])))
+    ])
diff --git a/configs/mmdet/detection/detection_tensorrt-int8_static-320x320.py b/configs/mmdet/detection/detection_tensorrt-int8_static-320x320.py
@@ -0,0 +1,14 @@
+_base_ = ['../_base_/base_static.py', '../../_base_/backends/tensorrt-int8.py']
+
+onnx_config = dict(input_shape=(320, 320))
+
+backend_config = dict(
+    common_config=dict(max_workspace_size=1 << 30),
+    model_inputs=[
+        dict(
+            input_shapes=dict(
+                input=dict(
+                    min_shape=[1, 3, 320, 320],
+                    opt_shape=[1, 3, 320, 320],
+                    max_shape=[1, 3, 320, 320])))
+    ])
diff --git a/configs/mmdet/detection/detection_tensorrt_static-320x320.py b/configs/mmdet/detection/detection_tensorrt_static-320x320.py
@@ -0,0 +1,14 @@
+_base_ = ['../_base_/base_static.py', '../../_base_/backends/tensorrt.py']
+
+onnx_config = dict(input_shape=(320, 320))
+
+backend_config = dict(
+    common_config=dict(max_workspace_size=1 << 30),
+    model_inputs=[
+        dict(
+            input_shapes=dict(
+                input=dict(
+                    min_shape=[1, 3, 320, 320],
+                    opt_shape=[1, 3, 320, 320],
+                    max_shape=[1, 3, 320, 320])))
+    ])