From 403bb20d53cb9cb00df05f31061d79d1f5e75d83 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=27=C3=A9lectron=20rare?=
 <108685187+electron-rare@users.noreply.github.com>
Date: Wed, 13 May 2026 11:01:17 +0200
Subject: [PATCH] feat(smplx): add decoder wrapper + neutral test

Adds SMPLXDecoder around smplx.SMPLXLayer with two modes : full decode
from Multi-HMR params (betas+pose+expression+transl) and decode_neutral
for T-pose smoke tests.

Two implementation notes from the test bring-up :
- model_path file -> use file.parent as model_folder (smplx tacks on
  SMPLX_<GENDER>.<ext> internally, single level not double).
- pose tensors are rotation matrices (B, N, 3, 3) ; passing zeros
  collapses the mesh. decode_neutral uses identity matrices.
---
 data_only_viz/smplx_decoder.py            | 78 +++++++++++++++++++++++
 data_only_viz/tests/test_smplx_decoder.py | 23 +++++++
 2 files changed, 101 insertions(+)
 create mode 100644 data_only_viz/smplx_decoder.py
 create mode 100644 data_only_viz/tests/test_smplx_decoder.py

diff --git a/data_only_viz/smplx_decoder.py b/data_only_viz/smplx_decoder.py
new file mode 100644
index 0000000..18b65c6
--- /dev/null
+++ b/data_only_viz/smplx_decoder.py
@@ -0,0 +1,78 @@
+"""Wrapper minimal autour de smplx.SMPLXLayer pour decoder les params
+de Multi-HMR (betas + thetas + expression) en vertices 3D."""
+from __future__ import annotations
+
+import logging
+from pathlib import Path
+
+import numpy as np
+import torch
+
+LOG = logging.getLogger("smplx_decoder")
+
+
+class SMPLXDecoder:
+    """Charge SMPL-X NEUTRAL et expose decode(params) -> (verts, joints)."""
+
+    def __init__(self, model_path: str, device: str = "mps") -> None:
+        import smplx
+        self.device = device
+        model_path_p = Path(model_path)
+        if model_path_p.is_file():
+            # smplx.SMPLXLayer attend le dossier contenant SMPLX_<GENDER>.<ext>
+            model_folder = str(model_path_p.parent)
+            ext = "npz" if model_path_p.suffix == ".npz" else "pkl"
+        else:
+            model_folder = str(model_path_p)
+            ext = "npz"
+        self.layer = smplx.SMPLXLayer(
+            model_path=model_folder,
+            gender="neutral",
+            num_betas=10,
+            num_expression_coeffs=10,
+            ext=ext,
+        ).to(device).eval()
+        LOG.info("SMPL-X loaded from %s (device=%s)", model_folder, device)
+
+    @torch.no_grad()
+    def decode(
+        self,
+        betas: torch.Tensor,
+        body_pose: torch.Tensor,
+        global_orient: torch.Tensor,
+        left_hand_pose: torch.Tensor,
+        right_hand_pose: torch.Tensor,
+        jaw_pose: torch.Tensor,
+        expression: torch.Tensor,
+        transl: torch.Tensor,
+    ) -> tuple[np.ndarray, np.ndarray]:
+        out = self.layer(
+            betas=betas, body_pose=body_pose, global_orient=global_orient,
+            left_hand_pose=left_hand_pose, right_hand_pose=right_hand_pose,
+            jaw_pose=jaw_pose, expression=expression, transl=transl,
+            return_verts=True,
+        )
+        return out.vertices.cpu().numpy(), out.joints.cpu().numpy()
+
+    @torch.no_grad()
+    def decode_neutral(self) -> tuple[np.ndarray, np.ndarray]:
+        """T-pose neutre. Les poses sont des matrices de rotation : on
+        utilise l'identite (pas zeros, qui collapserait le mesh)."""
+        d = self.device
+        B = 1
+
+        def eye(n: int) -> torch.Tensor:
+            return torch.eye(3, device=d).expand(B, n, 3, 3).contiguous()
+
+        out = self.layer(
+            betas=torch.zeros((B, 10), device=d),
+            body_pose=eye(21),
+            global_orient=eye(1),
+            left_hand_pose=eye(15),
+            right_hand_pose=eye(15),
+            jaw_pose=eye(1),
+            expression=torch.zeros((B, 10), device=d),
+            transl=torch.zeros((B, 3), device=d),
+        )
+        return (out.vertices[0].cpu().numpy(),
+                out.joints[0].cpu().numpy())
diff --git a/data_only_viz/tests/test_smplx_decoder.py b/data_only_viz/tests/test_smplx_decoder.py
new file mode 100644
index 0000000..b2cc44d
--- /dev/null
+++ b/data_only_viz/tests/test_smplx_decoder.py
@@ -0,0 +1,23 @@
+"""Test minimal : le decoder doit produire (10475, 3) vertices depuis
+les params canoniques (T-pose, shape neutre)."""
+from pathlib import Path
+
+import pytest
+import numpy as np
+
+SMPLX = (Path.home() / ".cache" / "av-live-multihmr"
+         / "models" / "smplx" / "SMPLX_NEUTRAL.npz")
+
+
+@pytest.mark.skipif(not SMPLX.exists(), reason="SMPL-X model not installed")
+def test_decode_neutral_tpose():
+    from data_only_viz.smplx_decoder import SMPLXDecoder
+    dec = SMPLXDecoder(str(SMPLX), device="cpu")
+    verts, joints = dec.decode_neutral()
+    assert verts.shape == (10475, 3)
+    assert joints.shape[1] == 3
+    # SMPL-X NEUTRAL T-pose : pelvis ~0.35m sous le bary mesh, mesh plein
+    # taille (1.5-2m sur y, et x comparable car bras tendus).
+    assert np.linalg.norm(joints[0]) < 1.0
+    assert 1.3 < verts[:, 1].ptp() < 2.5, "y extent suspect"
+    assert 1.3 < verts[:, 0].ptp() < 2.5, "x extent (bras tendus) suspect"