diff --git a/data_only_viz/multi_hmr_worker.py b/data_only_viz/multi_hmr_worker.py index ef3257d..2cf7723 100644 --- a/data_only_viz/multi_hmr_worker.py +++ b/data_only_viz/multi_hmr_worker.py @@ -214,7 +214,6 @@ class MultiHMRWorker: continue v3d = hh["v3d"].detach().cpu().numpy() - j3d = hh["j3d"].detach().cpu().numpy() transl = hh.get("transl_pelvis", hh.get("transl")) transl_np = transl.detach().cpu().numpy().flatten() @@ -236,7 +235,6 @@ class MultiHMRWorker: persons.append(SMPLXPerson( pid=int(pid), vertices_3d=np.ascontiguousarray(v3d, dtype=np.float32), - joints_3d=np.ascontiguousarray(j3d, dtype=np.float32), translation=np.ascontiguousarray(transl_np[:3], dtype=np.float32), confidence=float(hh.get("scores", 1.0)) if not hasattr( hh.get("scores", None), "item") else float( diff --git a/data_only_viz/state.py b/data_only_viz/state.py index ffcd9d4..23440de 100644 --- a/data_only_viz/state.py +++ b/data_only_viz/state.py @@ -27,7 +27,6 @@ class SMPLXPerson: decodes en metres. Vertices en repere camera (z > 0 devant).""" pid: int = -1 vertices_3d: np.ndarray = field(default_factory=lambda: np.empty((0, 3), dtype=np.float32)) # (10475, 3) - joints_3d: np.ndarray = field(default_factory=lambda: np.empty((0, 3), dtype=np.float32)) # (127, 3) translation: np.ndarray = field(default_factory=lambda: np.zeros(3, dtype=np.float32)) # (3,) confidence: float = 0.0 betas: np.ndarray = field(default_factory=lambda: np.zeros(10, dtype=np.float32)) # (10,)