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)