fix smpl trans handling

2023-01-16 16:37:11 +00:00
parent 7dffb02286
commit 4dee8e4106
1 changed files with 2 additions and 2 deletions
@@ -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)