Modulate simcc distribution with Gaussian. Args: simcc_x (np.ndarray[K, Wx]): model predicted simcc in x. simcc_y (np.ndarray[K,

Decode model parameters to smplx vertices & joints & texture Args: param_dict: smplx parameters param_type: should be

Convert displacement map into detail normal map

Get a yacs CfgNode object with default values for my_project.

Get face indicies by regions

r"""Function that returns Gaussian filter matrix coefficients. Args: kernel_size (Tuple[int, int]): filter sizes in the x and y direction

r"""Function that returns Gaussian filter matrix coefficients. Args: kernel_size (int): filter size should be odd. Returns:

Obtain (x,y) coordinates given a set of N heatmaps. If the center and the scale is provided the function will return the points also in

Get maximum response location and value from simcc representations. Note: instance number: N num_keypoints: K heatmap hei

Calculate the affine transformation matrix that can warp the bbox area in the input image to the output size. Args: center (np.ndarra

Inference RTMPose model. Args: sess (ort.InferenceSession): ONNXRuntime session. img (np.ndarray): Input image in shape. Ret

Performs Linear Blend Skinning with the given shape and pose parameters Parameters ---------- betas : torch.tensor BxNB

add estimated SH lighting to face albedo

:param expr_dir: :param vp_model: either 'snapshot' to use the experiment folder's code or a VPoser imported module, e.g. from human_body

Multiclass NMS implemented in Numpy. Class-aware version.

Single class NMS implemented in Numpy.

parsing the 2 points according to the 106 points, which cancels the roll

parsing the 2 points according to the 203 points, which cancels the roll

parsing the 2 points according to the 5 points, which cancels the roll

parsing the 2 points according to the 68 points, which cancels the roll

crop part(head/left_hand/right_hand) out from body data, joints also change accordingly

crop part(head/left_hand/right_hand) out from body data, joints also change accordingly

Draw 68 key points Args: image: the input image kpt: (68, 3).

Postprocess for RTMPose model output. Args: outputs (np.ndarray): Output of RTMPose model. model_input_size (tuple): RTMPose mode

Do preprocessing for RTMPose model inference. Args: img (np.ndarray): Input image in shape. input_size (tuple): Input image size

Load the vertex masks from the FLAME model and add custom masks

Convert 3x4 rotation matrix to 4d quaternion vector This algorithm is based on algorithm described in https://github.com/KieranWynn/pyquatern

Convert 3x4 rotation matrix to 4d quaternion vector This algorithm is based on algorithm described in https://github.com/KieranWynn/pyquatern

Obtain (x,y) coordinates given a set of N heatmaps and the coresponding locations of the maximums. If the center and the scale is provided the

Get the bbox image as the model input by affine transform. Args: input_size (dict): The input size of the model. bbox_scale (dict

Creates a batch of transformation matrices Args: - R: Bx3x3 array of a batch of rotation matrices - t: Bx3x1 array of

Creates a batch of transformation matrices Args: - R: Bx3x3 array of a batch of rotation matrices - t: Bx3x1 array of

Generate and affine transformation matrix. Given a set of points, a center, a scale and a targer resolution, the function generates and affin

with_xyflip: obey pytroch 3d coordinate

Calculates the 3D joint locations from the vertices Parameters ---------- J_regressor : torch.tensor JxV The regressor array tha

image range should be [0,1] dim: 2 for horizontal. 1 for vertical