mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

[Enhancement] check in tensorrt-fp16 deployment config for simcc (#1741)

Browse Source 
* check in tensorrt-fp16 deployment config for simcc

* update
This commit is contained in:
lvhan028 2023-02-09 20:11:53 +08:00 committed by GitHub
parent 19815ae9ae
commit 2efb42d8d8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 35 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
configs/mmpose/pose-detection_simcc_onnxruntime_dynamic.py
View File

@ -7,7 +7,10 @@ onnx_config = dict(
'input': {
0: 'batch',
},
'output': {
'simcc_x': {
0: 'batch'
},
'simcc_y': {
0: 'batch'
}
})

27
configs/mmpose/pose-detection_simcc_tensorrt-fp16_dynamic-256x192.py Normal file
View File

@ -0,0 +1,27 @@
_base_ = ['./pose-detection_static.py', '../_base_/backends/tensorrt-fp16.py']
onnx_config = dict(
input_shape=[192, 256],
output_names=['simcc_x', 'simcc_y'],
dynamic_axes={
'input': {
0: 'batch',
},
'simcc_x': {
0: 'batch'
},
'simcc_y': {
0: 'batch'
}
})
backend_config = dict(
common_config=dict(max_workspace_size=1 << 30),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 256, 192],
opt_shape=[2, 3, 256, 192],
max_shape=[4, 3, 256, 192])))
])

5
configs/mmpose/pose-detection_simcc_tensorrt_dynamic-256x192.py
View File

@ -7,7 +7,10 @@ onnx_config = dict(
'input': {
0: 'batch',
},
'output': {
'simcc_x': {
0: 'batch'
},
'simcc_y': {
0: 'batch'
}
})