From 4dee8e4106674af063a0627cf827eb60c57f97b8 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Mon, 16 Jan 2023 16:37:11 +0000 Subject: [PATCH] fix smpl trans handling --- cloth3d/read.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cloth3d/read.py b/cloth3d/read.py index 702f5fa..03ab45c 100644 --- a/cloth3d/read.py +++ b/cloth3d/read.py @@ -57,11 +57,11 @@ class DataReader: - V: human mesh vertices - F: mesh faces """ - def read_human(self, sample, frame, trans, zrot): + def read_human(self, sample, frame, zrot, use_trans=True): # SMPL parameters gender, pose, shape, trans = self.read_smpl_params(sample, frame) # Compute SMPL - V, J = self.smpl[gender].set_params(pose=pose, beta=shape, trans=trans) + V, J = self.smpl[gender].set_params(pose=pose, beta=shape, trans=trans if use_trans else None) V -= J[0:1] # Apply rotation on z-axis zRot = zRotMatrix(zrot)