mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
7dbc12d23f
* fix mask empty result * support fcn exporting to ONNX for ort and trt in whole mode * resolve comments * remove unnecessary code * update prepare_input * rewrite psp_head & aspp_head * test fcn deeplabv3 deeplabv3plus with trt
8 lines
252 B
Python
8 lines
252 B
Python
_base_ = ['./base.py', '../_base_/backends/tensorrt.py']
|
|
tensorrt_params = dict(model_params=[
|
|
dict(
|
|
opt_shape_dict=dict(
|
|
input=[[1, 3, 512, 512], [1, 3, 1024, 2048], [1, 3, 2048, 2048]]),
|
|
max_workspace_size=1 << 30)
|
|
])
|