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

Methoddelete

(self, key_name)

src/utils/lmdb.py:118

Functiondict2obj

(d)

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

Functiondict_tensor2npy

(tensor_dict)

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

Functiondraw_bodypose

(canvas, candidate, subset)

src/modules/dwpose/util.py:64

Functiondraw_facepose

(canvas, all_lmks)

src/modules/dwpose/util.py:136

Functiondraw_flame_lowerface_condition

draw flame lower face condition Args: img_fg (np.ndarray): foreground image in 0~1, HxWxC img_fg_alpha (np.ndarray): foreground i

src/utils/draw.py:211

Functiondraw_fullface_mask

(size, ver, mean_y)

src/utils/draw.py:241

Functiondraw_handpose

(canvas, all_hand_peaks)

src/modules/dwpose/util.py:108

Functiondraw_lowerface_mask

(size, ver, mean_y, is_kp68=False)

src/utils/draw.py:190

Functiondraw_mediapipe_kps

(mp_kps, canvas)

src/utils/draw.py:62

Functiondraw_mouth_mask

(size, ver, is_kp68=False)

src/utils/draw.py:201

Functiondraw_nose_mask

(size, ver)

src/utils/draw.py:185

Functiondraw_pose

(pose, H, W)

src/modules/dwpose/__init__.py:19

Functiondraw_rectangle

Draws the rectangle around the object borrowed from: https://bbox-visualizer.readthedocs.io/en/latest/_modules/bbox_visualizer/bbox_visualizer.htm

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

Functiondraw_teeth_mask

(ver, canvas)

src/utils/draw.py:178

Methoddummy_proj

(self, X, camera)

src/modules/ehm/EHM.py:29

Methodexecute

(self, args: ArgumentConfig)

src/data_prepare_pipeline.py:207

Methodextra_joint_names

Returns the names of the extra joints

src/modules/smplx/lbs.py:428

Methodextra_repr

(self)

src/modules/refiner/smplx_utils.py:57

Functionextract_audio

(video_fp, audio_fp)

src/utils/video.py:151

Methodextract_texture

extract face texture from given cropped face image

src/modules/pixie/visualizer.py:158

FunctionfaceDetect

(candidate, subset, oriImg)

src/modules/dwpose/util.py:217

Functionface_vertices

:param vertices: [batch size, number of vertices, 3] :param faces: [batch size, number of faces, 3] :return: [batch size, number of faces

src/modules/flame/FLAME.py:53

Functionface_vertices

borrowed from https://github.com/daniilidis-group/neural_renderer/blob/master/neural_renderer/vertices_to_faces.py :param vertices: [batch si

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

Functionface_vertices

borrowed from https://github.com/daniilidis-group/neural_renderer/blob/master/neural_renderer/vertices_to_faces.py :param vertices: [batch si

src/utils/helper.py:264

Functionfind_dynamic_lmk_idx_and_bcoords

Compute the faces, barycentric coordinates for the dynamic landmarks To do so, we first compute the rotation of the neck around the y-axis

src/modules/flame/lbs.py:36

Functionflip_pose

(pose_vector, pose_format='rot-mat')

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

Functionflip_pose

(pose_vector, pose_format='rot-mat')

src/utils/rotation_converter.py:570

Methodforward

(self, image, normalize=False)

src/modules/human_matting/stylematte.py:17

Methodforward

(self, features)

src/modules/human_matting/stylematte.py:65

Methodforward

(self, features)

src/modules/human_matting/stylematte.py:106

Methodforward

(self, x, y)

src/modules/human_matting/stylematte.py:123

Methodforward

(self, lr_x, lr_y, hr_x)

src/modules/human_matting/stylematte.py:164

Methodforward

(self, fine_src, base_src, base_fgr, base_pha, base_hid)

src/modules/human_matting/stylematte.py:217

Methodforward

(self, x)

src/modules/human_matting/stylematte.py:269

Methodforward

(self, input_image, return_type='matting', background_rgb=1.0)

src/modules/human_matting/matting.py:22

Methodforward

(self, img)

src/modules/teaser/smirk_encoder.py:34

Methodforward

(self, img)

src/modules/teaser/smirk_encoder.py:66

Methodforward

(self, img)

src/modules/teaser/smirk_encoder.py:95

Methodforward

(self, img, to_list=True)

src/modules/teaser/smirk_encoder.py:123

Methodforward

( self, vertices, faces=None, landmarks={}, cameras=None, tran

src/modules/renderer/body_renderer.py:32

Methodforward

(self, vertices, faces=None, landmarks={}, cameras=None, transform_matrix=None, focal_length=None, is_weak_cam

src/modules/renderer/hand_renderer.py:37

Methodforward

(self, vertices, faces=None, landmarks={}, cameras=None, transform_matrix=None, focal_length=None, is_weak_cam

src/modules/renderer/head_renderer.py:41

Methodforward

( self, vertices, faces=None, landmarks={}, cameras=None, tran

src/modules/renderer/base_renderer.py:77

Methodforward

(self, points, D=3, E=15, A=30, coords=True, ex_points=None)

src/modules/renderer/base_renderer.py:234

Methodforward

( self, vertices_world, texture_images, lights=None, image_size=512,

src/modules/renderer/base_renderer.py:347

Methodforward

(self, body_param_dict:dict, flame_param_dict:dict=None,mano_param_dict:dict=None, zero_expression=False, zero

src/modules/ehm/EHM.py:32

Methodforward

Args: shape_params: [N, number of shape parameters] expression_params: [N, number of expression parameter

src/modules/smplx/SMPLX_1.py:180

Methodforward

(self, vertices, faces)

src/modules/smplx/lbs.py:433

Methodforward

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

src/modules/smplx/SMPLX.py:324

Methodforward

(self, param_dictionary, zero_expression=False, zero_shape=False, zero_pose=False, zero_jaw=False)

src/modules/flame/FLAME.py:259

Methodforward

(self, joints, **kwargs)

src/modules/refiner/smplx_utils.py:45

Methodforward

(self, residual)

src/modules/refiner/smplx_utils.py:60

Methodforward

(self, joints, **kwargs)

src/modules/pixie/pixie.py:128

Methodforward

Encode images to smplx parameters Args: image: [bz, 3, 224, 224], range [0,1] image_hd(needed if key==body): a high r

src/modules/pixie/pixie_encoder.py:194

Methodforward

(self, vertices, faces, attributes=None, h=None, w=None)

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

Methodforward

(self, vertices, faces, attributes=None, h=None, w=None)

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

Methodforward

-- Texture Rendering vertices: [batch_size, V, 3], vertices in world space, for calculating normals, then shading transformed

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

Methodforward

inputs: [bz, 3, h, w], range: [0,1]

src/modules/pixie/models/encoders.py:21

Methodforward

(self, inputs)

src/modules/pixie/models/encoders.py:44

Methodforward

inputs: [bz, 3, h, w], range: [0,1]

src/modules/pixie/models/encoders.py:62

Methodforward

(self, x)

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

Methodforward

(self, x)

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

Methodforward

x: feature from body y: feature from part(head/hand) work: whether to fuse features

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

Methodforward

x: feature from body y: feature from part(head/hand) work: whether to fuse features

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

Methodforward

(self, x)

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

Methodforward

(self, x)

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

Methodforward

(self, x)

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

Methodforward

(self, x)

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

Methodforward

(self, x)

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

Methodforward

(self, x1, x2)

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

Methodforward

(self, x)

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

Methodforward

(self, x)

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

Methodforward

(self, vertices, joints)

src/modules/mano/MANO.py:52

Methodforward

(self, param_dictionary: dict, pose_type='rotmat')

src/modules/mano/MANO.py:147

Methodforward

Args: meshes_world: a Meshes object representing a batch of meshes with coordinates in world space.

src/utils/graphics.py:263

Methodforward

(self, vertices, faces=None, landmarks={}, cameras=None, transform_matrix=None, focal_length=None, ret_image=T

src/utils/graphics.py:358

Methodforward

(self, pose_out, pose_gt)

src/losses/pose_loss.py:10

Methodforward

calc face landmark loss Args: x (torch.Tensor): [B, N, x] y (torch.Tensor): [B, N, x] cam (torch.Tensor,

src/losses/lmk2d_loss.py:22

Functiongaussian_blur

(x, kernel_size=(5,5), sigma=(1.3,1.3))

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

Functiongenerate_triangles

quad layout: 0 1 ... w-1 w w+1 . w*h

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

Functionget_bbox_from_vert

(vert)

src/utils/draw.py:249

Methodget_fid_except_fids

(self, fids)

src/modules/flame/FLAME.py:721

Methodget_fid_except_region

(self, regions)

src/modules/flame/FLAME.py:707

Functionget_fps

(filepath)

src/utils/video.py:133

Methodget_output_dim

(self)

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

Functionget_recurrent_index

get recurrent index. [0, 1, 2, 2, 1, 0, 0, 1, 2, ....] For example: i = 6, length=3, return 0; i = 6, length=7, return 6;

src/utils/helper.py:67

Methodget_retargeting_lmk_info

(self, driving_rgb_lst)

src/utils/cropper.py:101

Functionget_rotation_matrix

the input is in degree

src/utils/camera.py:29

Functionget_semantic_indices

()

src/utils/draw.py:85

Methodget_training_params

(self)

src/modules/human_matting/stylematte.py:33

Methodget_union_human_box_in_video

(self, video_fp,frame_interval=1)

src/tracking_single_image.py:75

Methodget_vid_by_region

Get vertex indicies by regions

src/modules/flame/FLAME.py:671

FunctionhandDetect

(candidate, subset, oriImg)

src/modules/dwpose/util.py:151

Functionheadpose_pred_to_degree

pred: (bs, 66) or (bs, 1) or others

src/utils/camera.py:12

Functioninsert_dict

(adict:dict, key, elem:object,)

src/utils/helper.py:31

Functionis_template

(file_path)

src/utils/helper.py:61

Functionis_video

(file_path)

src/utils/helper.py:55

Functionl1_distance

(x, y)

src/losses/__init__.py:9

Functionl2_distance

(x, y)

src/losses/__init__.py:5

← previous next →401–500 of 581, ranked by callers