Skip to content

Commit 2ceabd0

Browse files
[Enhancement] Add 320x320 static config for tensorrt (#1689)
* add trt static 320 config for rtmdet * only add on detection folder
1 parent 8062e4e commit 2ceabd0

File tree

3 files changed

+42
-0
lines changed

3 files changed

+42
-0
lines changed
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
_base_ = ['../_base_/base_static.py', '../../_base_/backends/tensorrt-fp16.py']
2+
3+
onnx_config = dict(input_shape=(320, 320))
4+
5+
backend_config = dict(
6+
common_config=dict(max_workspace_size=1 << 30),
7+
model_inputs=[
8+
dict(
9+
input_shapes=dict(
10+
input=dict(
11+
min_shape=[1, 3, 320, 320],
12+
opt_shape=[1, 3, 320, 320],
13+
max_shape=[1, 3, 320, 320])))
14+
])
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
_base_ = ['../_base_/base_static.py', '../../_base_/backends/tensorrt-int8.py']
2+
3+
onnx_config = dict(input_shape=(320, 320))
4+
5+
backend_config = dict(
6+
common_config=dict(max_workspace_size=1 << 30),
7+
model_inputs=[
8+
dict(
9+
input_shapes=dict(
10+
input=dict(
11+
min_shape=[1, 3, 320, 320],
12+
opt_shape=[1, 3, 320, 320],
13+
max_shape=[1, 3, 320, 320])))
14+
])
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
_base_ = ['../_base_/base_static.py', '../../_base_/backends/tensorrt.py']
2+
3+
onnx_config = dict(input_shape=(320, 320))
4+
5+
backend_config = dict(
6+
common_config=dict(max_workspace_size=1 << 30),
7+
model_inputs=[
8+
dict(
9+
input_shapes=dict(
10+
input=dict(
11+
min_shape=[1, 3, 320, 320],
12+
opt_shape=[1, 3, 320, 320],
13+
max_shape=[1, 3, 320, 320])))
14+
])

0 commit comments

Comments
 (0)