birth

segmentation/deploy/configs/_base_/backends/tensorrt.py

@@ -0,0 +1,2 @@
+backend_config = dict(
+    type='tensorrt', common_config=dict(fp16_mode=False, max_workspace_size=0))

segmentation/deploy/configs/_base_/onnx_config.py

@@ -0,0 +1,10 @@
+onnx_config = dict(
+    type='onnx',
+    export_params=True,
+    keep_initializers_as_inputs=False,
+    opset_version=11,
+    save_file='end2end.onnx',
+    input_names=['input'],
+    output_names=['output'],
+    input_shape=None,
+    optimize=True)

segmentation/deploy/configs/mmseg/segmentation_static.py

@@ -0,0 +1,2 @@
+_base_ = ['../_base_/onnx_config.py']
+codebase_config = dict(type='mmseg', task='Segmentation', with_argmax=True)

segmentation/deploy/configs/mmseg/segmentation_tensorrt_static-512x512.py

@@ -0,0 +1,13 @@
+_base_ = ['./segmentation_static.py', '../_base_/backends/tensorrt.py']
+
+onnx_config = dict(input_shape=[512, 512])
+backend_config = dict(
+    common_config=dict(max_workspace_size=1 << 30),
+    model_inputs=[
+        dict(
+            input_shapes=dict(
+                input=dict(
+                    min_shape=[1, 3, 512, 512],
+                    opt_shape=[1, 3, 512, 512],
+                    max_shape=[1, 3, 512, 512])))
+    ])