Skip to content

Commit 43383e8

Browse files
authored
add rtmpose coreml support (#1902)
1 parent 4117aa5 commit 43383e8

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
_base_ = ['../_base_/torchscript_config.py', '../_base_/backends/coreml.py']
2+
3+
ir_config = dict(input_shape=[192, 256], output_names=['simcc_x', 'simcc_y'])
4+
5+
codebase_config = dict(type='mmpose', task='PoseDetection')
6+
7+
backend_config = dict(model_inputs=[
8+
dict(
9+
input_shapes=dict(
10+
input=dict(
11+
min_shape=[1, 3, 256, 192],
12+
max_shape=[1, 3, 256, 192],
13+
default_shape=[1, 3, 256, 192])))
14+
])

0 commit comments

Comments
 (0)