state: add NLFPerson dataclass + persons_nlf field

This commit is contained in:
L'électron rare
2026-05-13 10:13:02 +02:00
parent eb8afd5bac
commit 2c858bfc52
+16
View File
@@ -19,6 +19,18 @@ class PoseKp:
c: float = 0.0
@dataclass
class NLFPerson:
"""Resultats NLF pour une personne : vertices 3D SMPL (6890) en metres,
coordonnees camera (z > 0 devant). Le path nonparametrique fournit les
vertices directement sans decodage SMPL explicite."""
pid: int = -1
vertices_3d: tuple = field(default_factory=tuple) # ((x,y,z),) x 6890
joints_3d: tuple = field(default_factory=tuple) # ((x,y,z),) x 24 (SMPL)
translation: tuple = (0.0, 0.0, 0.0)
confidence: float = 0.0
@dataclass
class State:
# Audio sync
@@ -72,6 +84,10 @@ class State:
persons_face_ids: list[int] = field(default_factory=list)
persons_hands_ids: list[int] = field(default_factory=list)
# NLF (SMPL 6890 verts x N personnes, path nonparametrique)
persons_nlf: list = field(default_factory=list) # list[NLFPerson]
nlf_last_t: float = 0.0
# Renderer
width: int = 1280
height: int = 720