pose-detection_simcc_onnxruntime_dynamic.py

# _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"},
    },
)