Method forward

util/smplx/smplx/body_models.py:1629–1680

Forward pass for the MANO model.

(self,
                betas: Optional[Tensor] = None,
                global_orient: Optional[Tensor] = None,
                hand_pose: Optional[Tensor] = None,
                transl: Optional[Tensor] = None,
                return_verts: bool = True,
                return_full_pose: bool = False,
                **kwargs)

        return 'MANO'

    def forward(self,
                betas: Optional[Tensor] = None,
                global_orient: Optional[Tensor] = None,
                hand_pose: Optional[Tensor] = None,
                transl: Optional[Tensor] = None,
                return_verts: bool = True,
                return_full_pose: bool = False,
                **kwargs) -> MANOOutput:
        """Forward pass for the MANO model."""
        device, dtype = self.shapedirs.device, self.shapedirs.dtype
        if global_orient is None:
            batch_size = 1
            global_orient = torch.zeros(3, device=device, dtype=dtype).view(
                1, 1, 3).expand(batch_size, -1, -1).contiguous()
        else:
            batch_size = global_orient.shape[0]
        if hand_pose is None:
            hand_pose = torch.zeros(3, device=device, dtype=dtype).view(
                1, 1, 3).expand(batch_size, 15, -1).contiguous()
        if betas is None:
            betas = torch.zeros([batch_size, self.num_betas],
                                dtype=dtype,
                                device=device)
        if transl is None:
            transl = torch.zeros([batch_size, 3], dtype=dtype, device=device)

        full_pose = torch.cat([global_orient, hand_pose], dim=1)
        vertices, joints = lbs(betas,
                               full_pose,
                               self.v_template,
                               self.shapedirs,
                               self.posedirs,
                               self.J_regressor,
                               self.parents,
                               self.lbs_weights,
                               pose2rot=True)

        if self.joint_mapper is not None:
            joints = self.joint_mapper(joints)

        if transl is not None:
            joints = joints + transl.unsqueeze(dim=1)
            vertices = vertices + transl.unsqueeze(dim=1)

        output = MANOOutput(vertices=vertices if return_verts else None,
                            joints=joints if return_verts else None,
                            betas=betas,
                            global_orient=global_orient,
                            hand_pose=hand_pose,
                            full_pose=full_pose if return_full_pose else None)

        return output


class FLAME(SMPL):