Spaces:
Running
Running
File size: 1,902 Bytes
510bcce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
+++ venv/lib/python3.10/site-packages/smplx/body_models.py
@@ -366,7 +366,7 @@
num_repeats = int(batch_size / betas.shape[0])
betas = betas.expand(num_repeats, -1)
- vertices, joints = lbs(betas, full_pose, self.v_template,
+ vertices, joints, _ = lbs(betas, full_pose, self.v_template,
self.shapedirs, self.posedirs,
self.J_regressor, self.parents,
self.lbs_weights, pose2rot=pose2rot)
@@ -1228,7 +1228,7 @@
shapedirs = torch.cat([self.shapedirs, self.expr_dirs], dim=-1)
- vertices, joints = lbs(shape_components, full_pose, self.v_template,
+ vertices, joints, A = lbs(shape_components, full_pose, self.v_template,
shapedirs, self.posedirs,
self.J_regressor, self.parents,
self.lbs_weights, pose2rot=pose2rot,
@@ -1283,7 +1283,9 @@
right_hand_pose=right_hand_pose,
jaw_pose=jaw_pose,
v_shaped=v_shaped,
- full_pose=full_pose if return_full_pose else None)
+ full_pose=full_pose if return_full_pose else None,
+ A=A,
+ )
return output
+++ venv/lib/python3.10/site-packages/smplx/lbs.py
@@ -245,7 +245,7 @@
verts = v_homo[:, :, :3, 0]
- return verts, J_transformed
+ return verts, J_transformed, (A, J)
def vertices2joints(J_regressor: Tensor, vertices: Tensor) -> Tensor:
+++ venv/lib/python3.10/site-packages/smplx/utils.py
@@ -71,6 +71,7 @@
class SMPLXOutput(SMPLHOutput):
expression: Optional[Tensor] = None
jaw_pose: Optional[Tensor] = None
+ A: Optional[Tensor] = None
@dataclass
|