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

Functionlaplacian

(x)

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

Functionload_HRNet

(pretrained=False)

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

Functionload_ResNet101Model

()

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

Functionload_ResNet152Model

()

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

Functionload_ResNet50Model

()

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

Functionload_config

(cfg_file)

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

Functionload_description

(fp)

src/utils/helper.py:214

Functionload_driving_info

(driving_info, n_limit=-1)

src/utils/io.py:50

Functionload_img_online

(obj, mode="bgr", **kwargs)

src/utils/io.py:102

Functionload_model

(model_config, device)

src/utils/helper.py:155

Methodload_weights

(self, pretrained='')

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

Functionmake_abs_path

(fn)

src/configs/base_config.py:4

Functionmake_abs_path

(fn)

src/utils/cropper.py:17

Functionmake_abs_path

(fn)

src/utils/mediapipe_runner.py:14

Functionmake_abs_path

(fn)

src/utils/landmark_runner.py:11

Functionmake_abs_path

(fn)

src/utils/fan_landmark_runner.py:12

Methodmake_video_clip

(self, driving_rgb_lst, output_path, output_fps=30, **kwargs)

src/utils/cropper.py:109

Functionmax_grad_change

(grad_arr)

src/modules/refiner/smplx_utils.py:32

Functionmedian_blur

(x, kernel_size=(3,3))

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

Functionmerge_audio_video

(video_fp, audio_fp, wfp)

src/utils/video.py:56

Functionmerge_lower_face_pints

(pred_lst, pred_ff_lst, ff_only=False)

src/utils/draw.py:138

Functionmkdir

(d, log=False)

src/utils/helper.py:92

Functionmount_model

(model)

src/utils/helper.py:255

Functionmove_dict_to_device

(dict, device, tensor2float=False)

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

Functionnpmax

(array)

src/modules/dwpose/util.py:288

FunctionpadRightDownCorner

(img, stride, padValue)

src/modules/dwpose/util.py:34

Functionparse_args

()

src/modules/pixie/utils/config.py:179

Functionperspective_projection

Computes the perspective projection of a set of 3D points. Args: points (torch.Tensor): Tensor of shape (B, N, 3) containing the inpu

src/utils/rotation_converter.py:422

Functionplot_bbox

Draw bbox Args: image: the input image bbox: [left, top, right, bottom]

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

Methodpose_abs2rel

change absolute pose to relative pose Basic knowledge for SMPLX kinematic tree: absolute pose = parent pose * relative pose

src/modules/smplx/SMPLX_1.py:271

Methodpose_rel2abs

change relative pose to absolute pose Basic knowledge for SMPLX kinematic tree: absolute pose = parent pose * relative pose

src/modules/smplx/SMPLX_1.py:317

Methodpose_rel2abs

change relative pose to absolute pose Basic knowledge for SMPLX kinematic tree: absolute pose = parent pose * relative pose

src/modules/smplx/SMPLX.py:509

Functionrad2deg

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

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

Functionrad2deg

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

src/utils/rotation_converter.py:15

Methodraw_dump

(self, key_name, raw_payload)

src/utils/lmdb.py:135

Methodraw_load

(self, key_name, )

src/utils/lmdb.py:131

Methodrecover_position

transofrm mesh back to original image space

src/modules/pixie/visualizer.py:176

Functionrel_change

(prev_val, curr_val)

src/modules/refiner/smplx_utils.py:28

Functionremove_module

(state_dict)

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

Methodrender_alpha

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

src/modules/renderer/base_renderer.py:176

Methodrender_colors

-- rendering colors: could be rgb color/ normals, etc colors: [bz, num of vertices, 3]

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

Methodrender_depth

-- rendering depth

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

Methodrender_results

rendering pixie predictions Args: opdict: output from pixie that contains smplx vertices inpit_images: test image, fo

src/modules/pixie/visualizer.py:190

Functionresize_to_limit

(img, max_dim=1280, n=2)

src/utils/helper.py:220

Functionrotate_global_pose

(coeff_param)

src/modules/smplx/utils.py:152

Methodrotate_results

view body mesh in different poses, save in gif Args: opdict: output from pixie that contains smplx vertices result_ty

src/modules/pixie/visualizer.py:264

Methodrun

(self, tracked_rlt,id_share_params_result,lmdb_engine=None,frame_interval=1,steps=1001)

src/modules/refiner/flame_refiner.py:210

Methodrun

(self, tracked_rlt,id_share_rlt,lmdb_engine=None,interval=1,steps=1001)

src/modules/refiner/ehm_refiner.py:417

Methodrun

(self, img_rgb: np.ndarray, lmk=None)

src/utils/mediapipe_runner.py:67

Functionsave_crop_image

(t_optim_track_fp_smplx,image)

src/build_lmdb_dataset.py:72

Functionsave_image_rgb

(image_path:str, img_rgb:np.ndarray)

src/utils/io.py:35

Methodsave_obj

(self, objpath, opdict, k=0, Tpose=False)

src/modules/pixie/visualizer.py:299

Functionsave_pkl

(savepath, params, ind=0)

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

Methodseletec_3d68

(self, vertices)

src/modules/flame/FLAME.py:253

Functionshow_face

(datas)

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

Functionshow_tensor_img

(im)

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

Functionsmart_resize

(x, s)

src/modules/dwpose/util.py:6

Functionsmart_resize_k

(x, fx, fy)

src/modules/dwpose/util.py:20

Functionsmpl_to_openpose

Returns the indices of the permutation that maps OpenPose to SMPL Parameters ---------- model_type: str, optional

src/modules/refiner/smplx_utils.py:66

Functionsqueeze_tensor_to_numpy

(tensor)

src/utils/helper.py:101

Functionstandard_rasterize_colors

src/modules/pixie/utils/rasterizer/standard_rasterize_cuda.cpp:32

Functionsuffix

a.jpg -> jpg

src/utils/helper.py:23

Functiontensor2image

(tensor)

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

Functiontensor2image

(tensor)

src/utils/helper.py:240

Functiontensor_vis_landmarks

(images, landmarks, gt_landmarks=None, color = 'g', isScale=True)

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

Functionto_ndarray

(obj)

src/utils/mediapipe_runner.py:18

Functionto_ndarray

(obj)

src/utils/fan_landmark_runner.py:16

Functionto_tensor

(tensor, dtype=torch.float32)

src/modules/refiner/smplx_utils.py:21

Functiontransfer

(model, model_weights)

src/modules/dwpose/util.py:57

Methodtransform_points3d

(self, points3d, M)

src/modules/ehm/EHM.py:220

Functionunmount_model

(model)

src/utils/helper.py:259

Methodupdate_config

(self, user_args)

src/utils/cropper.py:51

Functionvertex_normals

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

src/modules/renderer/util.py:26

Functionvertex_normals

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:174

Functionvideo2gif

(video_fp, fps=30, size=256)

src/utils/video.py:40

Methodwarmup

(self)

src/modules/dwpose/__init__.py:41

Methodwarmup

(self)

src/utils/mediapipe_runner.py:93

Methodwarmup

(self)

src/utils/fan_landmark_runner.py:185

Methodwarmup

(self)

src/utils/face_analysis_diy.py:70

Functionwrite_obj

Save 3D face model with texture. borrowed from https://github.com/YadiraF/PRNet/blob/master/utils/write.py Args: obj_name: str

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

Functionwrite_obj

Save 3D face model with texture. borrowed from https://github.com/YadiraF/PRNet/blob/master/utils/write.py Args: obj_name: str

src/utils/io.py:130

← previous501–581 of 581, ranked by callers