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

Method__init__

(self, hid_channels=16)

src/modules/human_matting/stylematte.py:201

Method__init__

(self, r)

src/modules/human_matting/stylematte.py:264

Method__init__

(self, device='cpu', ckpt_fp='')

src/modules/human_matting/matting.py:7

Method__init__

(self, n_shape=300)

src/modules/teaser/smirk_encoder.py:49

Method__init__

(self, n_exp=50)

src/modules/teaser/smirk_encoder.py:77

Method__init__

(self, n_exp=50, n_shape=300)

src/modules/teaser/smirk_encoder.py:114

Method__init__

(self, assets_dir, image_size=1024, device='cuda', focal_length=24)

src/modules/renderer/body_renderer.py:8

Method__init__

(self, assets_dir, image_size=512, device='cuda', focal_length=24)

src/modules/renderer/hand_renderer.py:8

Method__init__

(self, assets_dir, image_size=512, device='cuda', focal_length=24)

src/modules/renderer/head_renderer.py:9

Method__init__

( self, assets_dir, image_size=512, device='cuda', skin_color=[252, 22

src/modules/renderer/base_renderer.py:16

Method__init__

(self, obj_filename=None, tuv=None, flame_mask=None, device='cpu')

src/modules/renderer/base_renderer.py:265

Method__init__

(self, flame_assets_dir, smplx_assets_dir,mano_assets_dir, n_shape=300, n_exp=50, with_

src/modules/ehm/EHM.py:10

Method__init__

(self, ckpt_fp:str, input_keys=[], output_keys=[], use_gpu=True)

src/modules/base/onnx_model.py:5

Method__init__

(self, smplx_assets_dir, n_shape=300, n_exp=50, with_texture=False)

src/modules/smplx/SMPLX_1.py:109

Method__init__

Selects extra joints from vertices

src/modules/smplx/lbs.py:387

Method__init__

(self, smplx_assets_dir, n_shape=200, n_exp=50, with_texture=False, check_pose=True)

src/modules/smplx/SMPLX.py:105

Method__init__

(self, ckpt_dir:str, use_gpu=True)

src/modules/dwpose/wholebody.py:9

Method__init__

(self, ckpt_dir:str, use_gpu=True)

src/modules/dwpose/__init__.py:37

Method__init__

(self, **kwargs)

src/modules/flame/FLAME.py:48

Method__init__

(self)

src/modules/flame/FLAME.py:362

Method__init__

( self, flame_parts_path, faces=None, faces_t=None,

src/modules/flame/FLAME.py:383

Method__init__

(self, flame_assets_dir, device='cuda:0', image_size=512,tanfov=1/12, )

src/modules/refiner/flame_refiner.py:18

Method__init__

Optimize the body pose and shape parameters using two stage optimization. Stage 1: Optimize the body pose and shape parameters for be

src/modules/refiner/ehm_refiner.py:65

Method__init__

(self, joint_maps=None)

src/modules/refiner/smplx_utils.py:37

Method__init__

(self, render_size=1024, config=None, device='cuda:0', part='body', background=None, rasterizer_type='standard

src/modules/pixie/visualizer.py:41

Method__init__

(self, joint_maps=None)

src/modules/pixie/pixie.py:120

Method__init__

(self, config=None, ckpt_fp='')

src/modules/pixie/pixie_encoder.py:36

Method__init__

use fixed raster_settings for rendering faces

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

Method__init__

use fixed raster_settings for rendering faces

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

Method__init__

(self, crop_size, scale=[1,1], trans_scale = 0.)

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

Method__init__

(self, crop_size, scale=[1,1], trans_scale = 0.)

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

Method__init__

(self, append_layers = None)

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

Method__init__

(self, append_layers = None)

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

Method__init__

(self, num_branches, blocks, num_blocks, num_inchannels, num_channels, fuse_method, multi_sca

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

Method__init__

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

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

Method__init__

(self, block, layers, num_classes=1000)

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

Method__init__

(self, inplanes, planes, stride=1, downsample=None)

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

Method__init__

(self, inplanes, planes, stride=1, downsample=None)

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

Method__init__

(self, in_channels, out_channels)

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

Method__init__

(self, in_channels, out_channels)

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

Method__init__

(self, in_channels, out_channels)

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

Method__init__

(self, n_channels, n_classes, bilinear=True)

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

Method__init__

(self, **kwargs)

src/modules/mano/MANO.py:17

Method__init__

(self, vertex_ids=None, **kwargs)

src/modules/mano/MANO.py:39

Method__init__

(self, **kwargs)

src/utils/cropper.py:32

Method__init__

(self, ckpt_fp='pretrained/mediapipe/face_landmarker.task', dsize=224)

src/utils/mediapipe_runner.py:29

Method__init__

(self, **kwargs)

src/utils/landmark_runner.py:26

Method__init__

(self, **kwargs)

src/utils/fan_landmark_runner.py:105

Method__init__

(self, name='buffalo_l', root='~/.insightface', allowed_modules=None, **kwargs)

src/utils/face_analysis_diy.py:34

Method__init__

(self)

src/utils/timer.py:10

Method__init__

Args: cameras: A cameras object which has a `transform_points` method which returns the transformed points after

src/utils/graphics.py:212

Method__init__

(self,image_size=512, skin_color=[252, 224, 203], bg_color=[0, 0, 0], focal_length=24,inverse_light=False)

src/utils/graphics.py:317

Method__init__

(self, **kwargs)

src/utils/video.py:92

Method__init__

(self, lmdb_path, write=False)

src/utils/lmdb.py:11

Method__init__

(self)

src/losses/pose_loss.py:7

Method__init__

(self, left_indices=None, right_indices=None, front_indices=None, selected_mp_indices=None,

src/losses/lmk2d_loss.py:6

Method__iter__

(self)

src/modules/flame/FLAME.py:371

Method__repr__

(self)

src/configs/base_config.py:9

Method__repr__

(self)

src/modules/flame/FLAME.py:365

Function_compute_euler_from_matrix

(dcm, seq='xyz', extrinsic=False)

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

Function_compute_euler_from_matrix

(dcm, seq='xyz', extrinsic=False)

src/utils/rotation_converter.py:288

Method_vertices2landmarks

Calculates landmarks by barycentric interpolation Input: vertices: torch.tensor NxVx3, dtype = torch.float32

src/modules/flame/FLAME.py:221

Functionalpha_feathering

(src_img, dest_img, img_mask, use_blur=True)

src/utils/draw.py:261

Functionassign_attributes

(a_from, a_to)

src/utils/helper.py:187

Functionbasename

a/b/c.jpg -> c

src/utils/helper.py:50

Functionbatch_axis2euler

(theta)

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

Functionbatch_axis2euler

(theta)

src/utils/rotation_converter.py:502

Functionbatch_cont2matrix

Decoder for transforming a latent representation to rotation matrices Implements the decoding method described in: "On the Continuit

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

Functionbatch_cont2matrix

Decoder for transforming a latent representation to rotation matrices Implements the decoding method described in: "On the Continuit

src/utils/rotation_converter.py:543

Functionbatch_euler2axis

(r)

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

Functionbatch_euler2axis

(r)

src/utils/rotation_converter.py:477

Functionbatch_euler2matrix

(r)

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

Functionbatch_euler2matrix

(r)

src/utils/rotation_converter.py:480

Functionbatch_matrix2axis

(rot_mats)

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

Functionbatch_matrix2axis

(rot_mats)

src/utils/rotation_converter.py:492

Functionbatch_orth_proj

orthgraphic projection X: 3d vertices, [bz, n_point, 3] camera: scale and translation, [bz, 3], [scale, tx, ty]

src/modules/renderer/util.py:60

Functionbatch_orth_proj

X is N x num_verts x 3

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

Functionbatch_orth_proj

orthgraphic projection X: 3d vertices, [bz, n_point, 3] camera: scale and translation, [bz, 3], [scale, tx, ty]

src/utils/rotation_converter.py:464

Methodbatch_persp_proj

X is B x num_verts x 3 Render camera is B x 6, each stands for [s, tx, ty, dx, dy, dz], dx, dy, dz are correction param

src/modules/smplx/SMPLX.py:249

Methodbatch_week_cam_to_perspective_proj

X is B x num_verts x 3 Render camera is at [0, 0, 2] and look at [0, 0, 0]

src/modules/smplx/SMPLX.py:222

Functionblend

(img: np.ndarray, mask: np.ndarray, background_color=(255, 255, 255))

src/utils/video.py:65

Functioncalculate_transformation

(config, s_kp_info, t_0_kp_info, t_i_kp_info, R_s, R_t_0, R_t_i)

src/utils/helper.py:201

Functionchange_video_fps

(input_file, output_file, fps=25, codec='libx264', crf=13, video_len_limit=180)

src/utils/video.py:119

Methodclear

(self)

src/utils/timer.py:29

Methodclose

(self)

src/utils/video.py:114

Functionconcat_feat

kp_source: (bs, k, 3) kp_driving: (bs, k, 3) Return: (bs, 2k*3)

src/utils/helper.py:119

Functionconcat_frames

(I_p_lst, driving_rgb_lst, img_rgb)

src/utils/video.py:73

Functionconv2d_relu

(input_filters, output_filters, kernel_size=3, bias=True)

src/modules/human_matting/stylematte.py:37

Functioncopy_state_dict

(cur_state_dict, pre_state_dict, prefix='', load_name=None)

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

Functioncreate_multi_dirs

(*dir_lst)

src/utils/helper.py:37

Methodcrop

(self, image, points, points_scale=None)

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

Methodcvt_weak_cam_to_persp_cam

(self, wcam)

src/tracking_single_image.py:99

Methodcvt_weak_cam_to_persp_cam

(self, wcam)

src/data_prepare_pipeline.py:92

Functiondct2cuda

(dct: dict, device_id: int = 0)

src/utils/helper.py:106

Functiondct2numpy

(dct: dict)

src/utils/helper.py:112

Methoddecode_Tpose

return body mesh in T pose, support body and head param dict only

src/modules/pixie/pixie.py:804

Functiondeg2rad

Function that converts angles from degrees to radians. See :class:`~torchgeometry.DegToRad` for details. Args: tensor (Tensor): Tens

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

Functiondeg2rad

Function that converts angles from degrees to radians. See :class:`~torchgeometry.DegToRad` for details. Args: tensor (Tensor): Tens

src/utils/rotation_converter.py:36

Methoddel_extra_params_values

(optim_frame_params)

src/tracking_single_image.py:441

Methoddel_extra_params_values

(optim_frame_params)

src/data_prepare_pipeline.py:439

← previous next →301–400 of 581, ranked by callers