Functions — github.com/KlingAIResearch/LivePortrait

hub / github.com/KlingAIResearch/LivePortrait / functions

Functions617 in github.com/KlingAIResearch/LivePortrait

⨍ Functions 617◇ Types & classes 112

↓ 2 callersFunctionbox_iou

(boxes1, boxes2)

src/utils/dependencies/XPose/util/box_ops.py:24

↓ 2 callersFunctionbuild_swin_transformer

(modelname, pretrain_img_size, **kw)

src/utils/dependencies/XPose/models/UniPose/swin_transformer.py:655

↓ 2 callersMethodcalc_lmk_from_cropped_image

(self, img_rgb_, **kwargs)

src/utils/cropper.py:153

↓ 2 callersFunctioncalc_motion_multiplier

calculate motion_multiplier based on the source image and the first driving frame

src/utils/helper.py:29

↓ 2 callersMethodclose

(self)

src/utils/video.py:128

↓ 2 callersMethodcrop_driving_video

Tracking based landmarks/alignment and cropping

src/utils/cropper.py:225

↓ 2 callersFunctiondownload

(sub_dir, name, force=False, root='~/.insightface')

src/utils/dependencies/insightface/utils/storage.py:9

↓ 2 callersFunctiondump

(wfp, obj)

src/utils/io.py:106

↓ 2 callersMethoddump

(self, file=None)

src/utils/dependencies/XPose/util/config.py:345

↓ 2 callersFunctionestimate_norm

(lmk, image_size=112,mode='arcface')

src/utils/dependencies/insightface/utils/face_align.py:11

↓ 2 callersMethodexecute_video

for video-driven portrait animation or video editing

src/gradio_pipeline.py:145

↓ 2 callersFunctionfind_params_file

(dir_path)

src/utils/dependencies/insightface/model_zoo/model_store.py:36

↓ 2 callersMethodforward_raw

Forward function.

src/utils/dependencies/XPose/models/UniPose/swin_transformer.py:572

↓ 2 callersFunctiongen_sineembed_for_position

(pos_tensor)

src/utils/dependencies/XPose/models/UniPose/utils.py:193

↓ 2 callersMethodget_model

(self, **kwargs)

src/utils/dependencies/insightface/model_zoo/model_zoo.py:39

↓ 2 callersFunctionget_model_file

r"""Return location for the pretrained on local file system. This function will download from online model zoo when model cannot be found or has

src/utils/dependencies/insightface/model_zoo/model_store.py:46

↓ 2 callersFunctionget_sine_pos_embed

generate sine position embedding from a position tensor Args: pos_tensor (torch.Tensor): shape: [..., n]. num_pos_feats (int): pro

src/utils/dependencies/XPose/models/UniPose/utils.py:26

↓ 2 callersFunctionis_main_process

()

src/utils/dependencies/XPose/util/misc.py:547

↓ 2 callersFunctionis_template

(file_path)

src/utils/helper.py:81

↓ 2 callersFunctionkp2gaussian

Transform a keypoint into gaussian like representation

src/modules/util.py:17

↓ 2 callersFunctionload

(fp)

src/utils/io.py:95

↓ 2 callersFunctionload_img_online

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

src/utils/io.py:73

↓ 2 callersFunctionmake_coordinate_grid

(spatial_size, ref, **kwargs)

src/modules/util.py:41

↓ 2 callersFunctionms_deform_attn_core_pytorch

(value, value_spatial_shapes, sampling_locations, attention_weights)

src/utils/dependencies/XPose/models/UniPose/ops/functions/ms_deform_attn_func.py:41

↓ 2 callersFunctionmulti_head_attention_forward

r""" Args: query, key, value: map a query and a set of key-value pairs to an output. See "Attention Is All You Need" for more

src/utils/dependencies/XPose/models/UniPose/attention.py:163

↓ 2 callersFunctionnested_tensor_from_tensor_list

(tensor_list: List[Tensor])

src/utils/dependencies/XPose/util/misc.py:456

↓ 2 callersFunctionnorm_cdf

(x)

src/modules/util.py:375

↓ 2 callersFunctionparse_pt2_from_pt101

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

src/utils/crop.py:45

↓ 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:191

↓ 2 callersFunctionpartial_fields

(target_class, kwargs)

app_animals.py:19

↓ 2 callersFunctionpartial_fields

(target_class, kwargs)

inference_animals.py:17

↓ 2 callersFunctionpartial_fields

(target_class, kwargs)

app.py:19

↓ 2 callersFunctionpartial_fields

(target_class, kwargs)

inference.py:17

↓ 2 callersFunctionprefix

a.jpg -> a

src/utils/helper.py:52

↓ 2 callersMethodprepare

(self, ctx_id, **kwargs)

src/utils/dependencies/insightface/model_zoo/scrfd.py:133

↓ 2 callersFunctionremove_suffix

a/b/c.jpg -> a/b/c

src/utils/helper.py:65

↓ 2 callersFunctionresize

(image, target, size, max_size=None)

src/utils/dependencies/XPose/transforms.py:148

↓ 2 callersFunctionsuffix

a.jpg -> jpg

src/utils/helper.py:44

↓ 2 callersFunctiontensor_to_numpy

transform torch.Tensor into numpy.ndarray

src/utils/helper.py:23

↓ 2 callersMethodto_img_list_single

(self, tensor, mask)

src/utils/dependencies/XPose/util/misc.py:415

↓ 2 callersMethodupdate

(self, value, n=1)

src/utils/dependencies/XPose/util/misc.py:47

↓ 2 callersFunctionwindow_partition

Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)

src/utils/dependencies/XPose/models/UniPose/swin_transformer.py:35

↓ 1 callersMethod__getattr__

(self, name)

src/utils/dependencies/XPose/util/config.py:316

↓ 1 callersMethod__getitem__

(self, name)

src/utils/dependencies/XPose/util/config.py:329

↓ 1 callersMethod__init__

(self, inference_cfg, crop_cfg, args: ArgumentConfig)

src/gradio_pipeline.py:41

↓ 1 callersMethod__init__

(self, dim, drop_path=0.)

src/modules/convnextv2.py:23

↓ 1 callersMethod__init__

(self, input_dim, hidden_dim, output_dim, num_layers)

src/utils/dependencies/XPose/models/UniPose/utils.py:165

↓ 1 callersMethod__init__

Initializes the model. Parameters: backbone: torch module of the backbone to be used. See backbone.py transformer: to

src/utils/dependencies/XPose/models/UniPose/unipose.py:29

↓ 1 callersMethod__init__

(self, num_layers, d_model=256, nheads=8, dim_feedforward=2048, dropout=0.1)

src/utils/dependencies/XPose/models/UniPose/transformer_vanilla.py:19

↓ 1 callersMethod__init__

(self, cfg_dict=None, cfg_text=None, filename=None)

src/utils/dependencies/XPose/util/config.py:192

↓ 1 callersMethod__init__

(self, model_path, **kwargs)

src/utils/dependencies/insightface/model_zoo/model_zoo.py:24

↓ 1 callersMethod__init__

(self, onnx_file)

src/utils/dependencies/insightface/model_zoo/model_zoo.py:36

↓ 1 callersMethod__setitem__

(self, name, value)

src/utils/dependencies/XPose/util/addict.py:32

↓ 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:328

↓ 1 callersMethod_file2dict

(filename)

src/utils/dependencies/XPose/util/config.py:76

↓ 1 callersFunction_get_clones

(module, N, layer_share=False)

src/utils/dependencies/XPose/models/UniPose/utils.py:18

↓ 1 callersFunction_get_global_gloo_group

Return a process group based on gloo backend, containing all the ranks The result is cached.

src/utils/dependencies/XPose/util/misc.py:102

↓ 1 callersMethod_init_vars

(self)

src/utils/dependencies/insightface/model_zoo/retinaface.py:87

↓ 1 callersMethod_init_vars

(self)

src/utils/dependencies/insightface/model_zoo/scrfd.py:88

↓ 1 callersFunction_is_power_of_2

(n)

src/utils/dependencies/XPose/models/UniPose/ops/modules/ms_deform_attn_key_aware.py:27

↓ 1 callersFunction_is_power_of_2

(n)

src/utils/dependencies/XPose/models/UniPose/ops/modules/ms_deform_attn.py:26

↓ 1 callersFunction_max_by_axis

(the_list)

src/utils/dependencies/XPose/util/misc.py:373

↓ 1 callersFunction_no_grad_trunc_normal_

(tensor, mean, std, a, b)

src/modules/util.py:372

↓ 1 callersFunction_ntuple

(n)

src/modules/util.py:445

↓ 1 callersFunction_onnx_nested_tensor_from_tensor_list

(tensor_list: List[Tensor])

src/utils/dependencies/XPose/util/misc.py:484

↓ 1 callersMethod_parse_int_float_bool

(val)

src/utils/dependencies/XPose/util/config.py:404

↓ 1 callersMethod_reset_parameters

(self)

src/utils/dependencies/XPose/models/UniPose/transformer_deformable.py:68

↓ 1 callersMethod_reset_parameters

(self)

src/utils/dependencies/XPose/models/UniPose/unipose.py:269

↓ 1 callersMethod_reset_parameters

(self)

src/utils/dependencies/XPose/models/UniPose/deformable_transformer.py:280

↓ 1 callersMethod_reset_parameters

(self)

src/utils/dependencies/XPose/models/UniPose/fuse_modules.py:130

↓ 1 callersMethod_reset_parameters

(self)

src/utils/dependencies/XPose/models/UniPose/attention.py:91

↓ 1 callersMethod_reset_parameters

(self)

src/utils/dependencies/XPose/models/UniPose/ops/modules/ms_deform_attn_key_aware.py:67

↓ 1 callersMethod_reset_parameters

(self)

src/utils/dependencies/XPose/models/UniPose/ops/modules/ms_deform_attn.py:66

↓ 1 callersMethod_validate_py_syntax

(filename)

src/utils/dependencies/XPose/util/config.py:66

↓ 1 callersFunctionall_gather_cpu

Run all_gather on arbitrary picklable data (not necessarily tensors) Args: data: any picklable object Returns: list[data]

src/utils/dependencies/XPose/util/misc.py:113

↓ 1 callersFunctionaverage_bbox_lst

(bbox_lst)

src/utils/crop.py:404

↓ 1 callersFunctionbox_iou_pairwise

(boxes1, boxes2)

src/utils/dependencies/XPose/util/box_ops.py:69

↓ 1 callersFunctionbox_xyxy_to_cxcywh

(x)

src/utils/dependencies/XPose/util/box_ops.py:16

↓ 1 callersFunctionbuild_backbone

Useful args: - backbone: backbone name - lr_backbone: - dilation - return_interm_indices: available: [0,1,2,3], [

src/utils/dependencies/XPose/models/UniPose/backbone.py:153

↓ 1 callersFunctionbuild_deformable_transformer

(args)

src/utils/dependencies/XPose/models/UniPose/deformable_transformer.py:1143

↓ 1 callersFunctionbuild_model

(args)

src/utils/dependencies/XPose/models/__init__.py:9

↓ 1 callersFunctionbuild_position_encoding

(args)

src/utils/dependencies/XPose/models/UniPose/position_encoding.py:142

↓ 1 callersFunctioncheck_file_exist

(filename, msg_tmpl='file "{}" does not exist')

src/utils/dependencies/XPose/util/config.py:19

↓ 1 callersFunctioncheck_forward_equal_with_pytorch_double

()

src/utils/dependencies/XPose/models/UniPose/ops/test.py:32

↓ 1 callersFunctioncheck_forward_equal_with_pytorch_float

()

src/utils/dependencies/XPose/models/UniPose/ops/test.py:48

↓ 1 callersFunctioncheck_gradient_numerical

(channels=4, grad_value=True, grad_sampling_loc=True, grad_attn_weight=True)

src/utils/dependencies/XPose/models/UniPose/ops/test.py:63

↓ 1 callersFunctionclean_state_dict

(state_dict)

src/utils/helper.py:193

↓ 1 callersMethodclose

(self)

src/utils/dependencies/insightface/data/rec_builder.py:64

↓ 1 callersMethodcopy

(self)

src/utils/dependencies/XPose/util/config.py:389

↓ 1 callersMethodcreate_deformed_feature

(self, feature, sparse_motions)

src/modules/dense_motion.py:45

↓ 1 callersMethodcreate_heatmap_representations

(self, feature, kp_driving, kp_source)

src/modules/dense_motion.py:55

↓ 1 callersMethodcreate_sparse_motions

(self, feature, kp_driving, kp_source)

src/modules/dense_motion.py:29

↓ 1 callersFunctioncrop_image_by_bbox

(img, bbox, lmk=None, dsize=512, angle=None, flag_rot=False, **kwargs)

src/utils/crop.py:282

↓ 1 callersMethoddecompose

(self)

src/utils/dependencies/XPose/util/misc.py:442

↓ 1 callersMethoddeform_input

(self, inp, deformation)

src/modules/warping_network.py:46

↓ 1 callersMethoddevice

(self)

src/utils/dependencies/XPose/util/misc.py:439

↓ 1 callersFunctiondistance2bbox

Decode distance prediction to bounding box. Args: points (Tensor): Shape (n, 2), [x, y]. distance (Tensor): Distance from the giv

src/utils/dependencies/insightface/model_zoo/retinaface.py:26

↓ 1 callersFunctiondistance2bbox

Decode distance prediction to bounding box. Args: points (Tensor): Shape (n, 2), [x, y]. distance (Tensor): Distance from the giv

src/utils/dependencies/insightface/model_zoo/scrfd.py:26

↓ 1 callersFunctiondistance2kps

Decode distance prediction to bounding box. Args: points (Tensor): Shape (n, 2), [x, y]. distance (Tensor): Distance from the giv

src/utils/dependencies/insightface/model_zoo/retinaface.py:49

↓ 1 callersFunctiondistance2kps

Decode distance prediction to bounding box. Args: points (Tensor): Shape (n, 2), [x, y]. distance (Tensor): Distance from the giv

src/utils/dependencies/insightface/model_zoo/scrfd.py:49

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