File size: 461 Bytes
c15cb31 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# _base_ = ["./pose-detection_static.py", "../_base_/backends/onnxruntime.py"]
# onnx_config = dict(input_shape=None)
codebase_config = dict(type="mmpose", task="PoseDetection")
backend_config = dict(type='onnxruntime')
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"},
},
)
|