mmdeploy/configs/mmseg/segmentation_tensorrt-fp16_static-1024x2048.py
RunningLeon 499d06d844
Add mmseg performance (#254)
* update doc and config

* test with ncnn

* update docs

* rename config

* update test result with openvino

* merge master

* resolve comments

* update dbnet
2021-12-09 20:17:00 +08:00

14 lines
449 B
Python

_base_ = ['./segmentation_static.py', '../_base_/backends/tensorrt-fp16.py']
onnx_config = dict(input_shape=[2048, 1024])
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 1024, 2048],
opt_shape=[1, 3, 1024, 2048],
max_shape=[1, 3, 1024, 2048])))
])