Functions — github.com/Pixel-Talk/EHM-Tracker

hub / github.com/Pixel-Talk/EHM-Tracker / functions

Functions581 in github.com/Pixel-Talk/EHM-Tracker

⨍ Functions 581◇ Types & classes 83

↓ 2 callersFunctioncam2persp_cam_fov_body

scale by changing zdist,unchange fov Returns: w2c R, T: Rotation matrix and translation vector

src/modules/renderer/util.py:117

↓ 2 callersMethodconstruct_vid_table

(self)

src/modules/flame/FLAME.py:618

↓ 2 callersFunctioncontiguous

(obj)

src/utils/io.py:61

↓ 2 callersFunctionconv3x3

3x3 convolution with padding

src/modules/pixie/models/resnet.py:120

↓ 2 callersFunctionconvert_pose

Convert pose parameters to rotation matrix Args: param_dict: smplx parameters param_type: should be one of body/head/hand Ret

src/modules/smplx/utils.py:6

↓ 2 callersMethodconvert_pose

Convert pose parameters to rotation matrix Args: param_dict: smplx parameters param_type: should be one of body/head/

src/modules/pixie/pixie.py:410

↓ 2 callersMethodconvert_pose

Convert pose parameters to rotation matrix Args: param_dict: smplx parameters param_type: should be one of body/head/

src/modules/pixie/pixie_encoder.py:322

↓ 2 callersMethodcrop

(self, image, points, points_scale=None)

src/modules/pixie/utils/array_cropper.py:64

↓ 2 callersFunctionestimate_translation_np

Find camera translation that brings 3D joints S closest to 2D the corresponding joints_2d. Input: S: (25, 3) 3D joint locations jo

src/modules/pixie/utils/util.py:70

↓ 2 callersFunctioneuler_to_quaternion

(r)

src/modules/pixie/utils/rotation_converter.py:58

↓ 2 callersFunctioneuler_to_quaternion

(r)

src/utils/rotation_converter.py:59

↓ 2 callersFunctionextract_base_id

(video_id)

src/split_dataset.py:8

↓ 2 callersFunctionfind_videos_in_path

(path)

tracking_video.py:40

↓ 2 callersFunctionget_gaussian_kernel

r"""Function that returns Gaussian filter coefficients. Args: kernel_size (int): filter size. It should be odd and positive. sigm

src/modules/pixie/utils/util.py:247

↓ 2 callersFunctionget_machine_info

()

src/utils/helper.py:314

↓ 2 callersFunctionget_proj_matrix

( tanfov,device, z_near=0.01, z_far=100, z_sign=1.0,)

src/utils/graphics.py:17

↓ 2 callersMethodget_projection_transform

(self,device)

src/utils/graphics.py:109

↓ 2 callersFunctionimages2video

(images, wfp, **kwargs)

src/utils/video.py:18

↓ 2 callersFunctionload_image_rgb

(image_path: str)

src/utils/io.py:13

↓ 2 callersFunctionload_obj

Ref: https://github.com/facebookresearch/pytorch3d/blob/25c065e9dafa90163e7cec873dbb324a637c68b7/pytorch3d/io/obj_io.py Load a mesh from a file-l

src/modules/pixie/utils/util.py:556

↓ 2 callersFunctionorginaze_body_pose

(coeff_lst_dct:dict, is_numpy=False)

src/modules/smplx/utils.py:88

↓ 2 callersFunctionparse_pt2_from_pt101

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

src/utils/crop.py:40

↓ 2 callersFunctionparse_rect_from_landmark

parsing center, size, angle from 101/68/5/x landmarks vx_ratio: the offset ratio along the pupil axis x-axis, multiplied by size vy_ratio: the

src/utils/crop.py:163

↓ 2 callersFunctionprefix

a.jpg -> a

src/utils/helper.py:42

↓ 2 callersMethodprocess_face_clusters

Construct a lookup table from face id to cluster id. cluster #0: background cluster #1: foreground c

src/modules/flame/FLAME.py:642

↓ 2 callersMethodprocess_face_mask

(self, faces)

src/modules/flame/FLAME.py:624

↓ 2 callersMethodprocess_vt_mask

(self, faces, faces_t)

src/modules/flame/FLAME.py:660

↓ 2 callersFunctionquaternion_to_angle_axis

Convert quaternion vector to angle axis of rotation. TODO: CORRECT Adapted from ceres C++ library: ceres-solver/include/ceres/rotation.h Arg

src/modules/pixie/utils/rotation_converter.py:230

↓ 2 callersFunctionquaternion_to_angle_axis

Convert quaternion vector to angle axis of rotation. TODO: CORRECT Adapted from ceres C++ library: ceres-solver/include/ceres/rotation.h Arg

src/utils/rotation_converter.py:231

↓ 2 callersFunctionquaternion_to_rotation_matrix

Convert quaternion coefficients to rotation matrix. Args: quat: size = [B, 4] 4 <===>(w, x, y, z) Returns: Rotation matrix cor

src/modules/pixie/utils/rotation_converter.py:208

↓ 2 callersFunctionquaternion_to_rotation_matrix

Convert quaternion coefficients to rotation matrix. Args: quat: size = [B, 4] 4 <===>(w, x, y, z) Returns: Rotation matrix cor

src/utils/rotation_converter.py:209

↓ 2 callersMethodrender_mesh

(self, vertices, cameras, faces=None, lights=None)

src/modules/renderer/base_renderer.py:136

↓ 2 callersMethodrender_mesh

(self, vertices,cameras=None,transform_matrix=None, faces=None,lights=None,skin_color=None,smplx2flame_ind=Non

src/utils/graphics.py:394

↓ 2 callersFunctionrot_mat_to_euler

(rot_mats)

src/modules/flame/lbs.py:27

↓ 2 callersFunctionsmplx_to_dwpose

()

src/modules/refiner/smplx_utils.py:178

↓ 2 callersFunctionsplit_train_valid

(videos_info, num_valid=5)

src/split_dataset.py:6

↓ 2 callersMethodtex_flame2smplx

Convert flame texture to smplx texture

src/modules/pixie/visualizer.py:128

↓ 2 callersMethodtransform_head_pts3d_to_image_coord

(self, X, M)

src/modules/refiner/ehm_refiner.py:123

↓ 2 callersMethodtransform_points

(self, points, tform, points_scale=None, normalize = True)

src/modules/pixie/utils/tensor_cropper.py:102

↓ 2 callersMethodtransform_points3d

(self, points3d:torch.Tensor, M:torch.Tensor)

src/modules/refiner/ehm_refiner.py:103

↓ 2 callersFunctionweak_cam2persp_cam

_summary_ Args: wcam (torch.Tensor): In shape Bx3, for each [s, x, y] focal_length (int, optional): perspective camera focal leng

src/modules/renderer/util.py:74

↓ 1 callersMethod__init__

(self, **kwargs)

src/modules/smplx/lbs.py:458

↓ 1 callersMethod__init__

(self, rho=1)

src/modules/refiner/smplx_utils.py:53

↓ 1 callersMethod__init__

(self, config=None, device='cuda:0')

src/modules/pixie/pixie.py:42

↓ 1 callersMethod__init__

(self, cfg, **kwargs)

src/modules/pixie/models/hrnet.py:203

↓ 1 callersMethod__init__

(self, channels = [2048*2, 1024, 1], detach_inputs=False, detach_feature=False)

src/modules/pixie/models/moderators.py:13

↓ 1 callersMethod__init__

Extension of the official MANO implementation to support more joints.

src/modules/mano/MANO.py:65

↓ 1 callersMethod_build_cameras

(self, transform_matrix, focal_length)

src/modules/renderer/base_renderer.py:46

↓ 1 callersMethod_build_cameras

(self, transform_matrix, focal_length, principal_point, image_size)

src/modules/renderer/base_renderer.py:329

↓ 1 callersMethod_build_orth_cameras

(self)

src/modules/renderer/base_renderer.py:64

↓ 1 callersMethod_check_branches

(self, num_branches, blocks, num_blocks, num_inchannels, num_channels)

src/modules/pixie/models/hrnet.py:50

↓ 1 callersFunction_compute_binary_kernel

r"""Creates a binary kernel to extract the patches. If the window size is HxW will create a (H*W)xHxW kernel.

src/modules/pixie/utils/util.py:323

↓ 1 callersMethod_create_model

(self)

src/modules/pixie/pixie.py:71

↓ 1 callersMethod_create_model

(self)

src/modules/pixie/pixie_encoder.py:68

↓ 1 callersFunction_estimate_similar_transform_from_pts

calculate the affine matrix of the cropped image from sparse points, the original image to the cropped image, the inverse is the cropped image to the

src/utils/crop.py:330

↓ 1 callersMethod_find_dynamic_lmk_idx_and_bcoords

Selects the face contour depending on the reletive position of the head Input: vertices: N X num_of_vertices

src/modules/flame/FLAME.py:177

↓ 1 callersFunction_fix_aspect_ratio

Extend the scale to match the given aspect ratio. Args: scale (np.ndarray): The image scale (w, h) in shape (2, ) aspect_ratio (f

src/modules/dwpose/onnxpose.py:148

↓ 1 callersFunction_get_params

(key)

src/modules/pixie/utils/util.py:392

↓ 1 callersMethod_make_branches

(self, num_branches, block, num_blocks, num_channels)

src/modules/pixie/models/hrnet.py:105

↓ 1 callersMethod_make_conv_layer

(self, in_channels=2048, num_layers=3, num_filters=2048, stride=1)

src/modules/pixie/models/hrnet.py:360

↓ 1 callersMethod_make_fuse_layers

(self)

src/modules/pixie/models/hrnet.py:115

↓ 1 callersMethod_make_layer

(self, block, planes, blocks, stride=1)

src/modules/pixie/models/hrnet.py:341

↓ 1 callersMethod_make_one_branch

(self, branch_index, block, num_blocks, num_channels, stride=1)

src/modules/pixie/models/hrnet.py:67

↓ 1 callersMethod_make_stages

(self, in_channels, out_channels, bin_sz)

src/modules/human_matting/stylematte.py:99

↓ 1 callersFunction_resize_to_limit

ajust the size of the image so that the maximum dimension does not exceed max_dim, and the width and the height of the image are multiples of n.

src/utils/io.py:67

↓ 1 callersFunction_rotate_point

Rotate a point by an angle. Args: pt (np.ndarray): 2D point coordinates (x, y) in shape (2, ) angle_rad (float): rotation angle i

src/modules/dwpose/onnxpose.py:166

↓ 1 callersMethod_setup_cropper

(self)

src/modules/pixie/pixie.py:61

↓ 1 callersMethod_setup_cropper

(self)

src/modules/pixie/pixie_encoder.py:58

↓ 1 callersMethod_setup_renderer

(self)

src/modules/pixie/visualizer.py:63

↓ 1 callersMethodadd_SHlight

(self, normal_images, sh_coeff)

src/modules/renderer/base_renderer.py:315

↓ 1 callersMethodadd_pointlight

vertices: [bz, nv, 3] lights: [bz, nlight, 6] returns: shading: [bz, nv, 3]

src/modules/pixie/utils/renderer.py:334

↓ 1 callersFunctionangle_axis_to_quaternion

Convert an angle axis to a quaternion. Adapted from ceres C++ library: ceres-solver/include/ceres/rotation.h Args: angle_axis (torch

src/modules/pixie/utils/rotation_converter.py:158

↓ 1 callersFunctionangle_axis_to_quaternion

Convert an angle axis to a quaternion. Adapted from ceres C++ library: ceres-solver/include/ceres/rotation.h Args: angle_axis (torch

src/utils/rotation_converter.py:159

↓ 1 callersFunctionaugment_bbox

(center, bbox_size, scale=[1.0, 1.0], trans_scale=0.)

src/modules/pixie/utils/tensor_cropper.py:31

↓ 1 callersFunctionaugment_bbox

(center, bbox_size, scale=[1.0, 1.0], trans_scale=0.)

src/modules/pixie/utils/array_cropper.py:26

↓ 1 callersFunctionaverage_bbox_lst

(bbox_lst)

src/utils/crop.py:418

↓ 1 callersFunctionbatch_axis2matrix

(theta)

src/modules/pixie/utils/rotation_converter.py:439

↓ 1 callersFunctionbatch_axis2matrix

(theta)

src/utils/rotation_converter.py:495

↓ 1 callersFunctionbatch_matrix2euler

(rot_mats)

src/modules/pixie/utils/rotation_converter.py:427

↓ 1 callersFunctionbatch_matrix2euler

(rot_mats)

src/utils/rotation_converter.py:483

↓ 1 callersFunctionbatch_rigid_transform

Applies a batch of rigid transformations to the joints Parameters ---------- rot_mats : torch.tensor BxNx3x3 Tensor of rotat

src/modules/smplx/lbs.py:330

↓ 1 callersFunctionbatch_rigid_transform

Applies a batch of rigid transformations to the joints Parameters ---------- rot_mats : torch.tensor BxNx3x3 Tensor of rotat

src/modules/flame/lbs.py:357

↓ 1 callersFunctionbatch_rodrigues

same as batch_matrix2axis Calculates the rotation matrices for a batch of rotation vectors Parameters ---------- rot_vec

src/modules/pixie/utils/rotation_converter.py:452

↓ 1 callersFunctionbatch_rodrigues

same as batch_matrix2axis Calculates the rotation matrices for a batch of rotation vectors Parameters ---------- rot_vec

src/utils/rotation_converter.py:508

↓ 1 callersFunctionbbox_xyxy2cs

Transform the bbox format from (x,y,w,h) into (center, scale) Args: bbox (ndarray): Bounding box(es) in shape (4,) or (n, 4), formatted

src/modules/dwpose/onnxpose.py:114

↓ 1 callersFunctionblend_shapes

Calculates the per vertex displacement due to the blend shapes Parameters ---------- betas : torch.tensor Bx(num_betas) Blend s

src/modules/flame/lbs.py:286

↓ 1 callersFunctionbuild_command

(args, part_lst, gpu_id)

tracking_video.py:16

↓ 1 callersFunctionbuild_lmdb_dataset

(data_folders,save_path)

src/build_lmdb_dataset.py:20

↓ 1 callersFunctioncolor_map_color

(value, cmap_name='cool', vmin=0, vmax=1)

src/modules/pixie/visualizer.py:26

↓ 1 callersFunctionconvert_json

(pose_dict)

src/modules/dwpose/run_dwpose.py:8

↓ 1 callersMethodconvert_traking_params

(self,optimized_result,id_share_params_results,image_key,device)

src/tracking_single_image.py:424

↓ 1 callersMethodconvert_traking_params

(self,optimized_result,id_share_params_results,image_key,device)

src/data_prepare_pipeline.py:427

↓ 1 callersMethodcreate_custom_mask

Add some cutom masks based on the original FLAME masks

src/modules/flame/FLAME.py:443

↓ 1 callersFunctioncrop_array

for single image only Args: image (numpy.Array): the reference array of shape HxWXC. size (Tuple[int, int]): a tuple with the hei

src/modules/pixie/utils/array_cropper.py:33

↓ 1 callersMethodcrop_single_image

(self, obj, fast_ret=False)

src/utils/cropper.py:56

↓ 1 callersFunctioncrop_tensor

for batch image Args: image (torch.Tensor): the reference tensor of shape BXHxWXC. center: [bz, 2] bboxsize: [bz, 1]

src/modules/pixie/utils/tensor_cropper.py:39

↓ 1 callersMethodcvt_cam_to_persp_cam_fov_body

(self, wcam)

src/tracking_single_image.py:105

↓ 1 callersMethodcvt_cam_to_persp_cam_fov_body

(self, wcam)

src/data_prepare_pipeline.py:98

↓ 1 callersFunctiondata_to_device

(data_dict, device='cuda')

src/modules/refiner/flame_refiner.py:229

↓ 1 callersFunctiondata_to_device

(data_dict, device='cuda')

src/modules/refiner/ehm_refiner.py:437

← previous next →101–200 of 581, ranked by callers