Hugging Face's logo

trojblue
/
rtmpose-x_simcc-body7

ONNX
Model card Files
xet
 Community
trojblue commited on
Commit
e8a11ef
·
verified ·
1 Parent(s): ffdeb76

Add RTMPose ONNX model files

Browse files
Files changed (7) hide show
  1. SOURCE.txt +1 -0
  2. deploy.json +16 -0
  3. detail.json +48 -0
  4. end2end.onnx +3 -0
  5. output_onnxruntime.jpg +0 -0
  6. output_pytorch.jpg +0 -0
  7. pipeline.json +123 -0
SOURCE.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/onnx_sdk/rtmpose-x_simcc-body7_pt-body7_700e-384x288-71d7b7e9_20230629.zip
deploy.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": "1.2.0",
3
+ "task": "PoseDetector",
4
+ "models": [
5
+ {
6
+ "name": "topdownposeestimator",
7
+ "net": "end2end.onnx",
8
+ "weights": "",
9
+ "backend": "onnxruntime",
10
+ "precision": "FP32",
11
+ "batch_size": 1,
12
+ "dynamic_shape": true
13
+ }
14
+ ],
15
+ "customs": []
16
+ }
detail.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": "1.2.0",
3
+ "codebase": {
4
+ "task": "PoseDetection",
5
+ "codebase": "mmpose",
6
+ "version": "1.1.0",
7
+ "pth": "https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/rtmpose-x_simcc-body7_pt-body7_700e-384x288-71d7b7e9_20230629.pth",
8
+ "config": "../mmpose-wb/projects/rtmpose/rtmpose/body_2d_keypoint/rtmpose-x_8xb256-700e_coco-384x288.py"
9
+ },
10
+ "codebase_config": {
11
+ "type": "mmpose",
12
+ "task": "PoseDetection"
13
+ },
14
+ "onnx_config": {
15
+ "type": "onnx",
16
+ "export_params": true,
17
+ "keep_initializers_as_inputs": false,
18
+ "opset_version": 11,
19
+ "save_file": "end2end.onnx",
20
+ "input_names": [
21
+ "input"
22
+ ],
23
+ "output_names": [
24
+ "simcc_x",
25
+ "simcc_y"
26
+ ],
27
+ "input_shape": [
28
+ 192,
29
+ 256
30
+ ],
31
+ "optimize": true,
32
+ "dynamic_axes": {
33
+ "input": {
34
+ "0": "batch"
35
+ },
36
+ "simcc_x": {
37
+ "0": "batch"
38
+ },
39
+ "simcc_y": {
40
+ "0": "batch"
41
+ }
42
+ }
43
+ },
44
+ "backend_config": {
45
+ "type": "onnxruntime"
46
+ },
47
+ "calib_config": {}
48
+ }
end2end.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df0c0fa91e9870b1515dcaff741fd76cc753dcfb12786862f61b243bae81cd52
3
+ size 197660326
output_onnxruntime.jpg ADDED
output_pytorch.jpg ADDED
pipeline.json ADDED
@@ -0,0 +1,123 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "pipeline": {
3
+ "input": [
4
+ "img"
5
+ ],
6
+ "output": [
7
+ "post_output"
8
+ ],
9
+ "tasks": [
10
+ {
11
+ "type": "Task",
12
+ "module": "Transform",
13
+ "name": "Preprocess",
14
+ "input": [
15
+ "img"
16
+ ],
17
+ "output": [
18
+ "prep_output"
19
+ ],
20
+ "transforms": [
21
+ {
22
+ "type": "LoadImageFromFile"
23
+ },
24
+ {
25
+ "type": "TopDownGetBboxCenterScale",
26
+ "padding": 1.25,
27
+ "image_size": [
28
+ 192,
29
+ 256
30
+ ]
31
+ },
32
+ {
33
+ "type": "TopDownAffine",
34
+ "image_size": [
35
+ 192,
36
+ 256
37
+ ]
38
+ },
39
+ {
40
+ "type": "Normalize",
41
+ "mean": [
42
+ 123.675,
43
+ 116.28,
44
+ 103.53
45
+ ],
46
+ "std": [
47
+ 58.395,
48
+ 57.12,
49
+ 57.375
50
+ ],
51
+ "to_rgb": true
52
+ },
53
+ {
54
+ "type": "ImageToTensor",
55
+ "keys": [
56
+ "img"
57
+ ]
58
+ },
59
+ {
60
+ "type": "Collect",
61
+ "keys": [
62
+ "img"
63
+ ],
64
+ "meta_keys": [
65
+ "img_shape",
66
+ "pad_shape",
67
+ "ori_shape",
68
+ "img_norm_cfg",
69
+ "scale_factor",
70
+ "bbox_score",
71
+ "center",
72
+ "scale"
73
+ ]
74
+ }
75
+ ]
76
+ },
77
+ {
78
+ "name": "topdownposeestimator",
79
+ "type": "Task",
80
+ "module": "Net",
81
+ "is_batched": true,
82
+ "input": [
83
+ "prep_output"
84
+ ],
85
+ "output": [
86
+ "infer_output"
87
+ ],
88
+ "input_map": {
89
+ "img": "input"
90
+ },
91
+ "output_map": {}
92
+ },
93
+ {
94
+ "type": "Task",
95
+ "module": "mmpose",
96
+ "name": "postprocess",
97
+ "component": "SimCCLabelDecode",
98
+ "params": {
99
+ "flip_test": false,
100
+ "type": "SimCCLabel",
101
+ "input_size": [
102
+ 288,
103
+ 384
104
+ ],
105
+ "sigma": [
106
+ 6.0,
107
+ 6.93
108
+ ],
109
+ "simcc_split_ratio": 2.0,
110
+ "normalize": false,
111
+ "use_dark": false
112
+ },
113
+ "output": [
114
+ "post_output"
115
+ ],
116
+ "input": [
117
+ "prep_output",
118
+ "infer_output"
119
+ ]
120
+ }
121
+ ]
122
+ }
123
+ }