| { | |
| "version": "1.3.0", | |
| "codebase": { | |
| "task": "PoseDetection", | |
| "codebase": "mmpose", | |
| "version": "1.2.0", | |
| "pth": "/content/dwpose_ckpts/dw-ss_ucoco.pth", | |
| "config": "mmpose/configs/wholebody_2d_keypoint/rtmpose/ubody/rtmpose-s_8xb64-270e_coco-ubody-wholebody-256x192.py" | |
| }, | |
| "codebase_config": { | |
| "type": "mmpose", | |
| "task": "PoseDetection", | |
| "export_postprocess": false | |
| }, | |
| "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": {} | |
| } |