rtmpose-x_simcc-body7 / detail.json
trojblue's picture
Add RTMPose ONNX model files
e8a11ef verified
{
"version": "1.2.0",
"codebase": {
"task": "PoseDetection",
"codebase": "mmpose",
"version": "1.1.0",
"pth": "https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/rtmpose-x_simcc-body7_pt-body7_700e-384x288-71d7b7e9_20230629.pth",
"config": "../mmpose-wb/projects/rtmpose/rtmpose/body_2d_keypoint/rtmpose-x_8xb256-700e_coco-384x288.py"
},
"codebase_config": {
"type": "mmpose",
"task": "PoseDetection"
},
"onnx_config": {
"type": "onnx",
"export_params": true,
"keep_initializers_as_inputs": false,
"opset_version": 11,
"save_file": "end2end.onnx",
"input_names": [
"input"
],
"output_names": [
"simcc_x",
"simcc_y"
],
"input_shape": [
192,
256
],
"optimize": true,
"dynamic_axes": {
"input": {
"0": "batch"
},
"simcc_x": {
"0": "batch"
},
"simcc_y": {
"0": "batch"
}
}
},
"backend_config": {
"type": "onnxruntime"
},
"calib_config": {}
}