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)