Multiply two quaternions. Usual torch rules for broadcasting apply. Args: a: Quaternions as tensor of shape (..., 4), real part

Multiply two quaternions. Usual torch rules for broadcasting apply. Args: a: Quaternions as tensor of shape (..., 4), real part

Multiply two quaternions. Usual torch rules for broadcasting apply. Args: a: Quaternions as tensor of shape (..., 4), real part

Extract the first or third Euler angle from the two members of the matrix which are positive constant times its sine and cosine. Args:

Extract the first or third Euler angle from the two members of the matrix which are positive constant times its sine and cosine. Args:

Extract the first or third Euler angle from the two members of the matrix which are positive constant times its sine and cosine. Args:

Return an activation function given a string

Angle prior that penalizes unnatural bending of the knees and elbows

Loss function for camera optimization.

Mean per-joint position error (i.e. mean Euclidean distance) often referred to as "Protocol #1" in many papers.

Params: -- matrix1: N1 x D -- matrix2: N2 x D Returns: -- dist: N1 x N2 dist[i, j] == distance(matrix1[i], matrix2[j])

Freeze the parameters of this module, i.e. do not update them during training :param module: freeze parameters of this module

Local pose

Geman-McClure error function

Official PyTorch Hessian Penalty implementation. Note: If you want to regularize multiple network activations simultaneously, you need to

This function computes the perspective projection of a set of points. Input: points (bs, N, 3): 3D points rotation (bs, 3, 3)

Uniform Skeleton

Convert quaternion(s) q to Euler angles. Expects a tensor of shape (*, 4), where * denotes any number of dimensions. Returns a tensor of

Enforce quaternion continuity across the time dimension by selecting the representation (q or -q) with minimal distance (or, equivalently, ma

Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as ten

Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as ten

Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as ten

Convert rotations given as quaternions to rotation matrices. Args: quaternions: quaternions with real part first, as tens

returns a list of [pad, conv] => should be += to some list, then apply sequential

load T2M text encoder and motion encoder for evaluating