| { | |
| "version": "1.3.0", | |
| "codebase": { | |
| "task": "PoseDetection", | |
| "codebase": "mmpose", | |
| "version": "1.2.0", | |
| "pth": "/content/dwpose_ckpts/dw-mm_ucoco.pth", | |
| "config": "mmpose/configs/wholebody_2d_keypoint/rtmpose/ubody/rtmpose-m_8xb64-270e_coco-ubody-wholebody-256x192.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", | |
| "precision": "fp16", | |
| "common_config": { | |
| "min_positive_val": 1e-07, | |
| "max_finite_val": 10000.0, | |
| "keep_io_types": false, | |
| "disable_shape_infer": false, | |
| "op_block_list": null, | |
| "node_block_list": null | |
| } | |
| }, | |
| "calib_config": {} | |
| } |