Convert rotations given as Euler angles in radians to rotation matrices. Args: euler_angles: Euler angles in radians as tensor of sha

@brief Convert a Matplotlib figure to a 4D numpy array with RGBA channels and return it @param fig a matplotlib figure @return a numpy 3D

Forward function for `Transformer`. Args: x (Tensor): Input query with shape [bs, c, h, w] where c = embed_dims.

Compute 2D reprojection loss if 2D keypoint annotations are available. The confidence is binary and indicates whether the keypoints e

Compute 3D keypoint loss if 3D keypoint annotations are available.

pred_joints := [B, 21, 3] T_c2m := [BN, 4, 4] K := [BN, 3, 3] gt_joints_2d := [BN, 21, 2]

q: [B, nsample, N, 256] q_merged: [B, nsample, 256]

Perform orthographic projection of 3D points X using the camera parameters Args: X: size = [B, N, 3] camera: size = [B

this is the only loss accepts unnormalized quats

Given a quaternion representing rotation, get the quaternion representing its inverse. Args: quaternion: Quaternions as tensor of

Convert quaternions to rotation matrixs. Args: quaternions (Union[torch.Tensor, numpy.ndarray]): input shape should be (.

Convert rotations given as quaternions to rotation matrices. Args: quat: quaternions in form (w,x,y,z), tensor of shape

Convert rotations given as quaternions to axis/angle. Args: quat: quaternions with real part first, as tensor of shape (.

resume checkpoints from resume_path, all process are synchronized Args: self (T): _description_ model (_type_): _desc

Convert rotations given as rotation matrices to Euler angles in radians. Args: matrix: Rotation matrices as tensor of shape (..., 3,

Convert rotations given as rotation matrices to quaternions. Args: rotmat: tensor of shape (..., 3, 3). Returns: quaterni

Convert rotations given as rotation matrices to axis/angle. Args: rotmat: Rotation matrices as tensor of shape (..., 3, 3). Retur

Convert rotations given as axis/angle to rotation matrices. Args: rotvec: Rotations given as a vector in axis angle form,

Compute per-vertex loss if vertex annotations are available.

Add children for a registry. The ``registry`` will be added as children based on its scope. The parent registry could build objects f

Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value of the angle given. Ar

Return the rotation matrices for one of the rotations about an axis of which Euler angles describe, for each value of the angle given. Ar