BreezeWu commited on
Commit
e36d0b4
·
verified ·
1 Parent(s): 46a8750

publish pear_parameter_network bundle (schema v3, ascend_310p)

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ artifacts/ascend_310p/pear_parameter_network_bs1.om filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,201 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: image-to-3d
4
+ tags:
5
+ - pear
6
+ - human-mesh-recovery
7
+ - smplx
8
+ - robotics
9
+ - IB-Robot
10
+ - ascend
11
+ - edge-deployment
12
+ base_model:
13
+ - BestWJH/PEAR_models
14
+ ---
15
+
16
+ # Model Card for PEAR Parameter Network (IB-Robot)
17
+
18
+ The parameter-regression network of PEAR (Pixel-aligned Expressive humAn mesh Recovery),
19
+ packaged for the [IB-Robot](https://atomgit.com/openeuler/IB_Robot) framework. From a
20
+ person crop it regresses SMPL-X / FLAME / camera parameters for expressive 3D human
21
+ understanding in HRI.
22
+
23
+ Person crops come from
24
+ [openEuler/yolox_x_640](https://huggingface.co/openEuler/yolox_x_640).
25
+
26
+ **Scope:** this bundle contains the PEAR image encoder and parameter regression heads only.
27
+ EHM / SMPL-X LBS, mesh generation and rendering stay on the host CPU and are **not** part of
28
+ the OM.
29
+
30
+ ## Repository Structure
31
+
32
+ - `inference_manifest.json` — deployment routing (schema v3)
33
+ - `assets/adapter.json` — adapter identity (`pear_parameter_network` / `predict_parameters`)
34
+ - `assets/pear_model.pt` — original PyTorch checkpoint the OM was converted from
35
+ - `artifacts/ascend_310p/pear_parameter_network_bs1.om` — Ascend 310P1 OM (batch 1)
36
+
37
+ ## Deployment Backends
38
+
39
+ | Target | Backend | Runtime | Hardware |
40
+ |--------|---------|---------|----------|
41
+ | `ascend_310p` | ascend | ACL | Ascend 310P1 |
42
+
43
+ **Input:** `pear.input` float32 [1,3,256,256] NCHW (`input`)
44
+
45
+ **Outputs:** eight tensors, fixed order
46
+
47
+ | # | semantic | shape |
48
+ |---|----------|-------|
49
+ | 0 | `smplx_pose_raw` | [1,312] |
50
+ | 1 | `smplx_scale` | [1,6] |
51
+ | 2 | `smplx_shape` | [1,200] |
52
+ | 3 | `smplx_expression` | [1,50] |
53
+ | 4 | `flame_pose` | [1,14] |
54
+ | 5 | `flame_shape` | [1,300] |
55
+ | 6 | `flame_expression` | [1,50] |
56
+ | 7 | `camera_raw` | [1,3] |
57
+
58
+ `smplx_pose_raw` splits as:
59
+
60
+ ```text
61
+ 0:6 global_orient
62
+ 6:132 body_pose, 21 × 6D
63
+ 132:222 left_hand_pose, 15 × 6D
64
+ 222:312 right_hand_pose, 15 × 6D
65
+ ```
66
+
67
+ The 6D values are **not** Euler angles: decode to rotation matrices first, then to
68
+ axis-angle radians if needed.
69
+
70
+ ### Preprocessing contract (`pear-rgb-crop256-bgr-imagenet-v1`)
71
+
72
+ ```text
73
+ person bbox xyxy in source-image coordinates
74
+ → centre (cx, cy), side = max(w, h) × 1.25
75
+ → square affine crop (cv2.INTER_LINEAR, BORDER_CONSTANT 0)
76
+ → 256×256 BGR
77
+ → NCHW float32, divided by 255
78
+ → ImageNet normalize (mean 0.485/0.456/0.406, std 0.229/0.224/0.225)
79
+ → width slice [:, :, :, 32:-32]
80
+ ```
81
+
82
+ The backbone therefore sees 256×192 content. Crops must be taken in source-video
83
+ coordinates, not in YOLOX's 640×640 letterbox space.
84
+
85
+ ## Source Model
86
+
87
+ This bundle's torch weights (`assets/pear_model.pt`) are the upstream PEAR checkpoint,
88
+ unmodified:
89
+
90
+ - **Model weights (HuggingFace):** [BestWJH/PEAR_models](https://huggingface.co/BestWJH/PEAR_models) — `pear_model.pt`, fetched by upstream code via `hf_hub_download(repo_id="BestWJH/PEAR_models", filename="pear_model.pt")`
91
+ - **Project page:** <https://wujh2001.github.io/PEAR/>
92
+ - `pear_model.pt` — 2,685,908,343 bytes, sha256 `be82dfa06e7b0608c6440058dfa0794d9b2ceee69f6e5b09bf41bb0076abeb18`
93
+
94
+ Upstream states this is the *initial* release of the PEAR model rather than the final version
95
+ presented in the paper; it may underperform on complex poses.
96
+
97
+ ### Source code
98
+
99
+ ```bash
100
+ git clone https://github.com/Pixel-Talk/PEAR.git
101
+ git -C PEAR checkout 230fa1534367c9f357c1c192a328cdc87ab4491c
102
+ ```
103
+
104
+ - **Repository:** <https://github.com/Pixel-Talk/PEAR> (Apache-2.0)
105
+ - **Commit:** `230fa1534367c9f357c1c192a328cdc87ab4491c` on `main` — 2026-08-01, *"Update app.py"*
106
+ - The clone used for export carries no submodules (upstream has no `.gitmodules`) and no
107
+ local patches; the working tree differs from that commit only in file permission bits.
108
+
109
+ The Ascend OM was converted from those weights via ONNX with external data
110
+ (`pear_parameter_network_bs1.onnx` + `.data`, consolidated `.data` sha256
111
+ `76d0b08fea2a17133aa62b718e1faaf329e0167a42638ffe63c177951b5ab766`), with ATC
112
+ `--soc_version=Ascend310P1`. The OM (`pear_parameter_network_bs1.om`, sha256
113
+ `67798d9f1da61fba4e5b706b20acf82cf2daf030a3239f89709e655421c84e81`) is not re-trained.
114
+
115
+ ## Assets Not Included
116
+
117
+ Consuming the parameter outputs (rotation decode, parameter bookkeeping) needs `numpy` only.
118
+
119
+ Reconstructing meshes / 3D joints additionally requires the SMPL-X, FLAME and MANO body
120
+ models, which are **not** redistributed here because their licenses do not permit it.
121
+ Obtain them yourself from the original sites and accept their terms:
122
+
123
+ - SMPL-X — <https://smpl-x.is.tue.mpg.de/> (`SMPLX_NEUTRAL_2020.npz`)
124
+ - FLAME 2020 — <https://flame.is.tue.mpg.de/> (`generic_model.pkl`)
125
+ - MANO — <https://mano.is.tue.mpg.de/> (`MANO_LEFT.pkl`, `MANO_RIGHT.pkl`)
126
+
127
+ These are research-licensed assets and are generally not usable for commercial deployment
128
+ without a separate agreement.
129
+
130
+ ## Validation
131
+
132
+ Board evidence recorded on a real Ascend 310P1 over a 368-frame, 30 FPS clip
133
+ (`npu-smi info` SoC = `Ascend310P1`; `Ascend310P3` is not a valid target for this device).
134
+
135
+ Numerical alignment vs. the PyTorch reference (strict gate passed):
136
+
137
+ ```text
138
+ raw parameter max abs diff: 0.036261 (gate 0.05)
139
+ rotation worst P95: 0.068421° (gate 1.0°)
140
+ body joint max: 0.332952° (gate 5.0°)
141
+ left hand joint max: 0.211992° (gate 5.0°)
142
+ right hand joint max: 0.211410° (gate 5.0°)
143
+ left/right swap: none (direct mean 0.027°, swapped mean 45.964°)
144
+ NaN/Inf: none; rotation matrices orthonormal to ~1e-7
145
+ ```
146
+
147
+ Latency, same run:
148
+
149
+ ```text
150
+ OM only: mean 26.298 ms, P95 27.559 ms, max 31.641 ms
151
+ full ACL: mean 27.784 ms, P95 29.602 ms, max 33.094 ms
152
+ ACL + CPU parameter processing: mean 28.542 ms, P95 30.542 ms, max 33.929 ms
153
+ ```
154
+
155
+ **Status: engineering `GO`, strict real-time `CONDITIONAL`.** Against a 33.33 ms budget at
156
+ 30 Hz, mean/P50/P95 fit, but the worst frame exceeds it by ~0.60 ms. Systems with a hard
157
+ per-frame deadline must budget for that overrun.
158
+
159
+ ### Semantics caveats
160
+
161
+ - Pose outputs are SMPL-X **local joint rotations**, not robot motor angles. Driving a robot
162
+ additionally requires a SMPL-X→joint mapping, axis transforms, zero offsets, sign and unit
163
+ conversion, joint limits and velocity/acceleration limits — none of which are in this bundle.
164
+ - `camera_raw` and the recovered body are in relative/model coordinates. Without real camera
165
+ intrinsics and root depth they must not be presented as absolute camera XYZ.
166
+
167
+ ## Usage
168
+
169
+ Select the `ascend_310p` deployment through the IB-Robot unified inference runtime; the
170
+ bundle is consumed as an external model bundle (it is not stored in the IB-Robot Git
171
+ repository).
172
+
173
+ ```python
174
+ from inference_manifest import load_inference_manifest
175
+
176
+ validated = load_inference_manifest("models/pear_parameter_network", "ascend_310p")
177
+ ```
178
+
179
+ ## License
180
+
181
+ Code and packaging: Apache-2.0. The PEAR weights are redistributed under the upstream
182
+ Apache-2.0 license of [BestWJH/PEAR_models](https://huggingface.co/BestWJH/PEAR_models) /
183
+ [Pixel-Talk/PEAR](https://github.com/Pixel-Talk/PEAR). The SMPL-X / FLAME / MANO body models
184
+ needed for mesh reconstruction are **not** included and carry their own restrictive licenses.
185
+
186
+ ## Citation
187
+
188
+ @misc{wu2026pear,
189
+ title = {PEAR: Pixel-aligned Expressive humAn mesh Recovery},
190
+ author = {Jiahao Wu and Yunfei Liu and Lijian Lin and Ye Zhu and Lei Zhu and Jingyi Li and Yu Li},
191
+ year = {2026},
192
+ eprint = {2601.22693},
193
+ archivePrefix = {arXiv},
194
+ primaryClass = {cs.CV},
195
+ url = {https://arxiv.org/abs/2601.22693}
196
+ }
197
+ @software{ib_robot,
198
+ title = {IB-Robot: Intelligence Boom Robot},
199
+ url = {https://atomgit.com/openeuler/IB_Robot},
200
+ license = {Apache-2.0}
201
+ }
artifacts/ascend_310p/pear_parameter_network_bs1.om ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67798d9f1da61fba4e5b706b20acf82cf2daf030a3239f89709e655421c84e81
3
+ size 1358182285
assets/adapter.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"interface":"tensor_model","family":"pear_parameter_network","model_type":"pear_parameter_network","operation":"predict_parameters","preprocessing":"pear-rgb-crop256-bgr-imagenet-v1","postprocessing":"pear-parameter-v1"}
assets/pear_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be82dfa06e7b0608c6440058dfa0794d9b2ceee69f6e5b09bf41bb0076abeb18
3
+ size 2685908343
inference_manifest.json ADDED
@@ -0,0 +1,242 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "schema_version": 3,
3
+ "bundle": {
4
+ "uuid": "ec439a1d-a65d-41fa-9d7c-47f81d07b09e",
5
+ "revision": 2,
6
+ "name": "pear_parameter_network",
7
+ "files": [
8
+ {
9
+ "path": "assets/adapter.json"
10
+ },
11
+ {
12
+ "path": "assets/pear_model.pt"
13
+ }
14
+ ],
15
+ "digest": {
16
+ "algorithm": "sha256",
17
+ "scope": "structure",
18
+ "value": "dd0291b4c5c6e734ac27aee60d4a0768bbd05c6109aefebfa33034b1fe095ca5"
19
+ }
20
+ },
21
+ "model": {
22
+ "interface": "tensor_model",
23
+ "model_type": "pear_parameter_network",
24
+ "operation": "predict_parameters",
25
+ "inputs": [
26
+ {
27
+ "semantic": "pear.input",
28
+ "dtype": "float32",
29
+ "layout": "NCHW",
30
+ "shape": [
31
+ 1,
32
+ 3,
33
+ 256,
34
+ 256
35
+ ]
36
+ }
37
+ ],
38
+ "outputs": [
39
+ {
40
+ "semantic": "smplx_pose_raw",
41
+ "dtype": "float32",
42
+ "shape": [
43
+ 1,
44
+ 312
45
+ ]
46
+ },
47
+ {
48
+ "semantic": "smplx_scale",
49
+ "dtype": "float32",
50
+ "shape": [
51
+ 1,
52
+ 6
53
+ ]
54
+ },
55
+ {
56
+ "semantic": "smplx_shape",
57
+ "dtype": "float32",
58
+ "shape": [
59
+ 1,
60
+ 200
61
+ ]
62
+ },
63
+ {
64
+ "semantic": "smplx_expression",
65
+ "dtype": "float32",
66
+ "shape": [
67
+ 1,
68
+ 50
69
+ ]
70
+ },
71
+ {
72
+ "semantic": "flame_pose",
73
+ "dtype": "float32",
74
+ "shape": [
75
+ 1,
76
+ 14
77
+ ]
78
+ },
79
+ {
80
+ "semantic": "flame_shape",
81
+ "dtype": "float32",
82
+ "shape": [
83
+ 1,
84
+ 300
85
+ ]
86
+ },
87
+ {
88
+ "semantic": "flame_expression",
89
+ "dtype": "float32",
90
+ "shape": [
91
+ 1,
92
+ 50
93
+ ]
94
+ },
95
+ {
96
+ "semantic": "camera_raw",
97
+ "dtype": "float32",
98
+ "shape": [
99
+ 1,
100
+ 3
101
+ ]
102
+ }
103
+ ],
104
+ "semantic_identity": {
105
+ "logical_model_revision": "pear-parameter-network@handoff",
106
+ "output_semantics": "pear-parameter-v1",
107
+ "preprocessing_contract": "pear-rgb-crop256-bgr-imagenet-v1"
108
+ }
109
+ },
110
+ "deployments": {
111
+ "ascend_310p": {
112
+ "uuid": "76c515a7-7460-4905-b847-562180448d06",
113
+ "revision": 1,
114
+ "artifacts": {
115
+ "model": {
116
+ "format": "om",
117
+ "path": "artifacts/ascend_310p/pear_parameter_network_bs1.om",
118
+ "sha256": "67798d9f1da61fba4e5b706b20acf82cf2daf030a3239f89709e655421c84e81"
119
+ }
120
+ },
121
+ "bindings": {
122
+ "model": {
123
+ "inputs": [
124
+ {
125
+ "semantic": "pear.input",
126
+ "runtime_name": "input",
127
+ "index": 0,
128
+ "dtype": "float32",
129
+ "layout": "NCHW",
130
+ "shape": [
131
+ 1,
132
+ 3,
133
+ 256,
134
+ 256
135
+ ]
136
+ }
137
+ ],
138
+ "outputs": [
139
+ {
140
+ "semantic": "smplx_pose_raw",
141
+ "runtime_name": "/Concat_1:0:smplx_pose_raw",
142
+ "index": 0,
143
+ "dtype": "float32",
144
+ "shape": [
145
+ 1,
146
+ 312
147
+ ]
148
+ },
149
+ {
150
+ "semantic": "smplx_scale",
151
+ "runtime_name": "PartitionedCall_/smplx_scale_decoder/Gemm_add_2114:0:smplx_scale",
152
+ "index": 1,
153
+ "dtype": "float32",
154
+ "shape": [
155
+ 1,
156
+ 6
157
+ ]
158
+ },
159
+ {
160
+ "semantic": "smplx_shape",
161
+ "runtime_name": "PartitionedCall_/smplx_shape_decoder/Gemm_add_2119:0:smplx_shape",
162
+ "index": 2,
163
+ "dtype": "float32",
164
+ "shape": [
165
+ 1,
166
+ 200
167
+ ]
168
+ },
169
+ {
170
+ "semantic": "smplx_expression",
171
+ "runtime_name": "PartitionedCall_/smplx_expression_decoder/Gemm_add_2124:0:smplx_expression",
172
+ "index": 3,
173
+ "dtype": "float32",
174
+ "shape": [
175
+ 1,
176
+ 50
177
+ ]
178
+ },
179
+ {
180
+ "semantic": "flame_pose",
181
+ "runtime_name": "PartitionedCall_/flame_poses_decoder/Gemm_add_2129:0:flame_pose",
182
+ "index": 4,
183
+ "dtype": "float32",
184
+ "shape": [
185
+ 1,
186
+ 14
187
+ ]
188
+ },
189
+ {
190
+ "semantic": "flame_shape",
191
+ "runtime_name": "PartitionedCall_/flame_shape_decoder/Gemm_add_2134:0:flame_shape",
192
+ "index": 5,
193
+ "dtype": "float32",
194
+ "shape": [
195
+ 1,
196
+ 300
197
+ ]
198
+ },
199
+ {
200
+ "semantic": "flame_expression",
201
+ "runtime_name": "PartitionedCall_/flame_expression_decoder/Gemm_add_2139:0:flame_expression",
202
+ "index": 6,
203
+ "dtype": "float32",
204
+ "shape": [
205
+ 1,
206
+ 50
207
+ ]
208
+ },
209
+ {
210
+ "semantic": "camera_raw",
211
+ "runtime_name": "PartitionedCall_/cam_decoder/Gemm_add_2144:0:camera_raw",
212
+ "index": 7,
213
+ "dtype": "float32",
214
+ "shape": [
215
+ 1,
216
+ 3
217
+ ]
218
+ }
219
+ ]
220
+ }
221
+ },
222
+ "execution": [
223
+ "model"
224
+ ],
225
+ "execution_contract": {
226
+ "state_scope": "request",
227
+ "execution_structure": "direct",
228
+ "cancellation_granularity": "request_boundary"
229
+ },
230
+ "runtime_profile": {
231
+ "backend": "ascend",
232
+ "target": {
233
+ "runtime": "acl",
234
+ "soc": "Ascend310P1"
235
+ },
236
+ "profile": {
237
+ "device_id": 0
238
+ }
239
+ }
240
+ }
241
+ }
242
+ }