mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
e3a8baac4c
* add rotated nms trt plugin * fix ops output shape * rebase * fix lint * add fp16, benchmark * format docs * remove unused definition, add ut * add docs * update docs * add doc
33 lines
774 B
Python
33 lines
774 B
Python
_base_ = ['./rotated-detection_static.py', '../_base_/backends/tensorrt.py']
|
|
|
|
onnx_config = dict(
|
|
output_names=['dets', 'labels'],
|
|
input_shape=None,
|
|
dynamic_axes={
|
|
'input': {
|
|
0: 'batch',
|
|
2: 'height',
|
|
3: 'width'
|
|
},
|
|
'dets': {
|
|
0: 'batch',
|
|
1: 'num_dets',
|
|
},
|
|
'labels': {
|
|
0: 'batch',
|
|
1: 'num_dets',
|
|
},
|
|
},
|
|
)
|
|
|
|
backend_config = dict(
|
|
common_config=dict(max_workspace_size=1 << 30),
|
|
model_inputs=[
|
|
dict(
|
|
input_shapes=dict(
|
|
input=dict(
|
|
min_shape=[1, 3, 320, 320],
|
|
opt_shape=[1, 3, 1024, 1024],
|
|
max_shape=[1, 3, 1024, 1024])))
|
|
])
|