_base_ = ["./pose-detection_static.py", "../_base_/backends/onnxruntime-fp16.py"]

onnx_config = dict(
    input_shape=[288, 384],
    output_names=["kpts", "scores"],
    dynamic_axes={
        "input": {
            0: "batch",
        },
        "kpts": {
            0: "batch",
        },
        "scores": {
            0: "batch",
        },
    },
)

codebase_config = dict(export_postprocess=True)  # export get_simcc_maximum