SYMBOL INDEX (718 symbols across 79 files)

FILE: app.py
  function partial_fields (line 19) | def partial_fields(target_class, kwargs):
  function fast_check_ffmpeg (line 23) | def fast_check_ffmpeg():
  function gpu_wrapped_execute_video (line 59) | def gpu_wrapped_execute_video(*args, **kwargs):
  function gpu_wrapped_execute_image_retargeting (line 63) | def gpu_wrapped_execute_image_retargeting(*args, **kwargs):
  function gpu_wrapped_execute_video_retargeting (line 67) | def gpu_wrapped_execute_video_retargeting(*args, **kwargs):
  function reset_sliders (line 71) | def reset_sliders(*args, **kwargs):

FILE: app_animals.py
  function partial_fields (line 19) | def partial_fields(target_class, kwargs):
  function fast_check_ffmpeg (line 23) | def fast_check_ffmpeg():
  function gpu_wrapped_execute_video (line 57) | def gpu_wrapped_execute_video(*args, **kwargs):

FILE: inference.py
  function partial_fields (line 17) | def partial_fields(target_class, kwargs):
  function fast_check_ffmpeg (line 21) | def fast_check_ffmpeg():
  function fast_check_args (line 29) | def fast_check_args(args: ArgumentConfig):
  function main (line 36) | def main():

FILE: inference_animals.py
  function partial_fields (line 17) | def partial_fields(target_class, kwargs):
  function fast_check_ffmpeg (line 21) | def fast_check_ffmpeg():
  function fast_check_args (line 29) | def fast_check_args(args: ArgumentConfig):
  function main (line 36) | def main():

FILE: speed.py
  function initialize_inputs (line 20) | def initialize_inputs(batch_size=1, device_id=0):
  function load_and_compile_models (line 49) | def load_and_compile_models(cfg, model_config):
  function warm_up_models (line 83) | def warm_up_models(compiled_models, stitching_retargeting_module, inputs):
  function measure_inference_times (line 100) | def measure_inference_times(compiled_models, stitching_retargeting_modul...
  function print_benchmark_results (line 146) | def print_benchmark_results(compiled_models, stitching_retargeting_modul...
  function main (line 168) | def main():

FILE: src/config/argument_config.py
  class ArgumentConfig (line 14) | class ArgumentConfig(PrintableConfig):

FILE: src/config/base_config.py
  function make_abs_path (line 12) | def make_abs_path(fn):
  class PrintableConfig (line 16) | class PrintableConfig:  # pylint: disable=too-few-public-methods
    method __repr__ (line 19) | def __repr__(self):

FILE: src/config/crop_config.py
  class CropConfig (line 13) | class CropConfig(PrintableConfig):

FILE: src/config/inference_config.py
  function load_lip_array (line 14) | def load_lip_array():
  class InferenceConfig (line 19) | class InferenceConfig(PrintableConfig):

FILE: src/gradio_pipeline.py
  function update_args (line 28) | def update_args(args, user_args):
  class GradioPipeline (line 37) | class GradioPipeline(LivePortraitPipeline):
    method __init__ (line 41) | def __init__(self, inference_cfg, crop_cfg, args: ArgumentConfig):
    method update_delta_new_eyeball_direction (line 47) | def update_delta_new_eyeball_direction(self, eyeball_direction_x, eyeb...
    method update_delta_new_smile (line 67) | def update_delta_new_smile(self, smile, delta_new, **kwargs):
    method update_delta_new_wink (line 80) | def update_delta_new_wink(self, wink, delta_new, **kwargs):
    method update_delta_new_eyebrow (line 90) | def update_delta_new_eyebrow(self, eyebrow, delta_new, **kwargs):
    method update_delta_new_lip_variation_zero (line 102) | def update_delta_new_lip_variation_zero(self, lip_variation_zero, delt...
    method update_delta_new_lip_variation_one (line 108) | def update_delta_new_lip_variation_one(self, lip_variation_one, delta_...
    method update_delta_new_lip_variation_two (line 117) | def update_delta_new_lip_variation_two(self, lip_variation_two, delta_...
    method update_delta_new_lip_variation_three (line 125) | def update_delta_new_lip_variation_three(self, lip_variation_three, de...
    method update_delta_new_mov_x (line 133) | def update_delta_new_mov_x(self, mov_x, delta_new, **kwargs):
    method update_delta_new_mov_y (line 139) | def update_delta_new_mov_y(self, mov_y, delta_new, **kwargs):
    method execute_video (line 145) | def execute_video(
    method execute_image_retargeting (line 231) | def execute_image_retargeting(
    method prepare_retargeting_image (line 336) | def prepare_retargeting_image(
    method init_retargeting_image (line 377) | def init_retargeting_image(self, retargeting_source_scale: float, sour...
    method execute_video_retargeting (line 401) | def execute_video_retargeting(self, input_lip_ratio: float, input_vide...
    method prepare_retargeting_video (line 488) | def prepare_retargeting_video(self, input_video, retargeting_source_sc...
    method prepare_video_lip_silence (line 544) | def prepare_video_lip_silence(self, input_video, device, flag_do_crop=...
  class GradioPipelineAnimal (line 596) | class GradioPipelineAnimal(LivePortraitPipelineAnimal):
    method __init__ (line 599) | def __init__(self, inference_cfg, crop_cfg, args: ArgumentConfig):
    method execute_video (line 606) | def execute_video(

FILE: src/live_portrait_pipeline.py
  function make_abs_path (line 31) | def make_abs_path(fn):
  class LivePortraitPipeline (line 35) | class LivePortraitPipeline(object):
    method __init__ (line 37) | def __init__(self, inference_cfg: InferenceConfig, crop_cfg: CropConfig):
    method make_motion_template (line 41) | def make_motion_template(self, I_lst, c_eyes_lst, c_lip_lst, **kwargs):
    method execute (line 77) | def execute(self, args: ArgumentConfig):

FILE: src/live_portrait_pipeline_animal.py
  function make_abs_path (line 35) | def make_abs_path(fn):
  class LivePortraitPipelineAnimal (line 38) | class LivePortraitPipelineAnimal(object):
    method __init__ (line 40) | def __init__(self, inference_cfg: InferenceConfig, crop_cfg: CropConfig):
    method make_motion_template (line 44) | def make_motion_template(self, I_lst, **kwargs):
    method execute (line 69) | def execute(self, args: ArgumentConfig):

FILE: src/live_portrait_wrapper.py
  class LivePortraitWrapper (line 22) | class LivePortraitWrapper(object):
    method __init__ (line 27) | def __init__(self, inference_cfg: InferenceConfig):
    method inference_ctx (line 70) | def inference_ctx(self):
    method update_config (line 78) | def update_config(self, user_args):
    method prepare_source (line 83) | def prepare_source(self, img: np.ndarray) -> torch.Tensor:
    method prepare_videos (line 104) | def prepare_videos(self, imgs) -> torch.Tensor:
    method extract_feature_3d (line 122) | def extract_feature_3d(self, x: torch.Tensor) -> torch.Tensor:
    method get_kp_info (line 131) | def get_kp_info(self, x: torch.Tensor, **kwargs) -> dict:
    method get_pose_dct (line 157) | def get_pose_dct(self, kp_info: dict) -> dict:
    method get_fs_and_kp_info (line 165) | def get_fs_and_kp_info(self, source_prepared, driving_first_frame):
    method transform_keypoint (line 184) | def transform_keypoint(self, kp_info: dict):
    method retarget_eye (line 214) | def retarget_eye(self, kp_source: torch.Tensor, eye_close_ratio: torch...
    method retarget_lip (line 227) | def retarget_lip(self, kp_source: torch.Tensor, lip_close_ratio: torch...
    method stitch (line 240) | def stitch(self, kp_source: torch.Tensor, kp_driving: torch.Tensor) ->...
    method stitching (line 253) | def stitching(self, kp_source: torch.Tensor, kp_driving: torch.Tensor)...
    method warp_decode (line 276) | def warp_decode(self, feature_3d: torch.Tensor, kp_source: torch.Tenso...
    method parse_output (line 300) | def parse_output(self, out: torch.Tensor) -> np.ndarray:
    method calc_ratio (line 310) | def calc_ratio(self, lmk_lst):
    method calc_combined_eye_ratio (line 320) | def calc_combined_eye_ratio(self, c_d_eyes_i, source_lmk):
    method calc_combined_lip_ratio (line 328) | def calc_combined_lip_ratio(self, c_d_lip_i, source_lmk):
  class LivePortraitWrapperAnimal (line 337) | class LivePortraitWrapperAnimal(LivePortraitWrapper):
    method __init__ (line 341) | def __init__(self, inference_cfg: InferenceConfig):

FILE: src/modules/appearance_feature_extractor.py
  class AppearanceFeatureExtractor (line 12) | class AppearanceFeatureExtractor(nn.Module):
    method __init__ (line 14) | def __init__(self, image_channel, block_expansion, num_down_blocks, ma...
    method forward (line 38) | def forward(self, source_image):

FILE: src/modules/convnextv2.py
  class Block (line 15) | class Block(nn.Module):
    method __init__ (line 23) | def __init__(self, dim, drop_path=0.):
    method forward (line 33) | def forward(self, x):
  class ConvNeXtV2 (line 48) | class ConvNeXtV2(nn.Module):
    method __init__ (line 60) | def __init__(
    method _init_weights (line 108) | def _init_weights(self, m):
    method forward_features (line 113) | def forward_features(self, x):
    method forward (line 119) | def forward(self, x):
  function convnextv2_tiny (line 147) | def convnextv2_tiny(**kwargs):

FILE: src/modules/dense_motion.py
  class DenseMotionNetwork (line 13) | class DenseMotionNetwork(nn.Module):
    method __init__ (line 14) | def __init__(self, block_expansion, num_blocks, max_features, num_kp, ...
    method create_sparse_motions (line 29) | def create_sparse_motions(self, feature, kp_driving, kp_source):
    method create_deformed_feature (line 45) | def create_deformed_feature(self, feature, sparse_motions):
    method create_heatmap_representations (line 55) | def create_heatmap_representations(self, feature, kp_driving, kp_source):
    method forward (line 67) | def forward(self, feature, kp_driving, kp_source):

FILE: src/modules/motion_extractor.py
  class MotionExtractor (line 18) | class MotionExtractor(nn.Module):
    method __init__ (line 19) | def __init__(self, **kwargs):
    method load_pretrained (line 26) | def load_pretrained(self, init_path: str):
    method forward (line 33) | def forward(self, x):

FILE: src/modules/spade_generator.py
  class SPADEDecoder (line 13) | class SPADEDecoder(nn.Module):
    method __init__ (line 14) | def __init__(self, upscale=1, max_features=256, block_expansion=64, ou...
    method forward (line 41) | def forward(self, feature):

FILE: src/modules/stitching_retargeting_network.py
  class StitchingRetargetingNetwork (line 18) | class StitchingRetargetingNetwork(nn.Module):
    method __init__ (line 19) | def __init__(self, input_size, hidden_sizes, output_size):
    method initialize_weights_to_zero (line 31) | def initialize_weights_to_zero(self):
    method forward (line 37) | def forward(self, x):

FILE: src/modules/util.py
  function kp2gaussian (line 17) | def kp2gaussian(kp, spatial_size, kp_variance):
  function make_coordinate_grid (line 41) | def make_coordinate_grid(spatial_size, ref, **kwargs):
  class ConvT2d (line 61) | class ConvT2d(nn.Module):
    method __init__ (line 66) | def __init__(self, in_features, out_features, kernel_size=3, stride=2,...
    method forward (line 73) | def forward(self, x):
  class ResBlock3d (line 80) | class ResBlock3d(nn.Module):
    method __init__ (line 85) | def __init__(self, in_features, kernel_size, padding):
    method forward (line 92) | def forward(self, x):
  class UpBlock3d (line 103) | class UpBlock3d(nn.Module):
    method __init__ (line 108) | def __init__(self, in_features, out_features, kernel_size=3, padding=1...
    method forward (line 115) | def forward(self, x):
  class DownBlock2d (line 123) | class DownBlock2d(nn.Module):
    method __init__ (line 128) | def __init__(self, in_features, out_features, kernel_size=3, padding=1...
    method forward (line 134) | def forward(self, x):
  class DownBlock3d (line 142) | class DownBlock3d(nn.Module):
    method __init__ (line 147) | def __init__(self, in_features, out_features, kernel_size=3, padding=1...
    method forward (line 158) | def forward(self, x):
  class SameBlock2d (line 166) | class SameBlock2d(nn.Module):
    method __init__ (line 171) | def __init__(self, in_features, out_features, groups=1, kernel_size=3,...
    method forward (line 180) | def forward(self, x):
  class Encoder (line 187) | class Encoder(nn.Module):
    method __init__ (line 192) | def __init__(self, block_expansion, in_features, num_blocks=3, max_fea...
    method forward (line 200) | def forward(self, x):
  class Decoder (line 207) | class Decoder(nn.Module):
    method __init__ (line 212) | def __init__(self, block_expansion, in_features, num_blocks=3, max_fea...
    method forward (line 228) | def forward(self, x):
  class Hourglass (line 240) | class Hourglass(nn.Module):
    method __init__ (line 245) | def __init__(self, block_expansion, in_features, num_blocks=3, max_fea...
    method forward (line 251) | def forward(self, x):
  class SPADE (line 255) | class SPADE(nn.Module):
    method __init__ (line 256) | def __init__(self, norm_nc, label_nc):
    method forward (line 268) | def forward(self, x, segmap):
  class SPADEResnetBlock (line 278) | class SPADEResnetBlock(nn.Module):
    method __init__ (line 279) | def __init__(self, fin, fout, norm_G, label_nc, use_se=False, dilation...
    method forward (line 302) | def forward(self, x, seg1):
    method shortcut (line 309) | def shortcut(self, x, seg1):
    method actvn (line 316) | def actvn(self, x):
  function filter_state_dict (line 320) | def filter_state_dict(state_dict, remove_name='fc'):
  class GRN (line 329) | class GRN(nn.Module):
    method __init__ (line 333) | def __init__(self, dim):
    method forward (line 338) | def forward(self, x):
  class LayerNorm (line 344) | class LayerNorm(nn.Module):
    method __init__ (line 351) | def __init__(self, normalized_shape, eps=1e-6, data_format="channels_l...
    method forward (line 361) | def forward(self, x):
  function _no_grad_trunc_normal_ (line 372) | def _no_grad_trunc_normal_(tensor, mean, std, a, b):
  function drop_path (line 408) | def drop_path(x, drop_prob=0., training=False, scale_by_keep=True):
  class DropPath (line 428) | class DropPath(nn.Module):
    method __init__ (line 432) | def __init__(self, drop_prob=None, scale_by_keep=True):
    method forward (line 437) | def forward(self, x):
  function trunc_normal_ (line 441) | def trunc_normal_(tensor, mean=0., std=1., a=-2., b=2.):
  function _ntuple (line 445) | def _ntuple(n):

FILE: src/modules/warping_network.py
  class WarpingNetwork (line 14) | class WarpingNetwork(nn.Module):
    method __init__ (line 15) | def __init__(
    method deform_input (line 46) | def deform_input(self, inp, deformation):
    method forward (line 49) | def forward(self, feature_3d, kp_driving, kp_source):

FILE: src/utils/animal_landmark_runner.py
  class XPoseRunner (line 25) | class XPoseRunner(object):
    method __init__ (line 26) | def __init__(self, model_config_path, model_checkpoint_path, embedding...
    method load_animal_model (line 42) | def load_animal_model(self, model_config_path, model_checkpoint_path, ...
    method load_image (line 51) | def load_image(self, input_image):
    method get_unipose_output (line 61) | def get_unipose_output(self, image, instance_text_prompt, keypoint_tex...
    method run (line 106) | def run(self, input_image, instance_text_prompt, keypoint_text_example...
    method warmup (line 131) | def warmup(self):

FILE: src/utils/camera.py
  function headpose_pred_to_degree (line 14) | def headpose_pred_to_degree(pred):
  function get_rotation_matrix (line 31) | def get_rotation_matrix(pitch_, yaw_, roll_):

FILE: src/utils/crop.py
  function make_abs_path (line 16) | def make_abs_path(fn):
  function _transform_img (line 19) | def _transform_img(img, M, dsize, flags=CV2_INTERP, borderMode=None):
  function _transform_pts (line 36) | def _transform_pts(pts, M):
  function parse_pt2_from_pt101 (line 45) | def parse_pt2_from_pt101(pt101, use_lip=True):
  function parse_pt2_from_pt106 (line 63) | def parse_pt2_from_pt106(pt106, use_lip=True):
  function parse_pt2_from_pt203 (line 80) | def parse_pt2_from_pt203(pt203, use_lip=True):
  function parse_pt2_from_pt68 (line 96) | def parse_pt2_from_pt68(pt68, use_lip=True):
  function parse_pt2_from_pt5 (line 123) | def parse_pt2_from_pt5(pt5, use_lip=True):
  function parse_pt2_from_pt9 (line 139) | def parse_pt2_from_pt9(pt9, use_lip=True):
  function parse_pt2_from_pt_x (line 163) | def parse_pt2_from_pt_x(pts, use_lip=True):
  function parse_rect_from_landmark (line 191) | def parse_rect_from_landmark(
  function parse_bbox_from_landmark (line 250) | def parse_bbox_from_landmark(pts, **kwargs):
  function crop_image_by_bbox (line 282) | def crop_image_by_bbox(img, bbox, lmk=None, dsize=512, angle=None, flag_...
  function _estimate_similar_transform_from_pts (line 328) | def _estimate_similar_transform_from_pts(
  function crop_image (line 376) | def crop_image(img, pts: np.ndarray, **kwargs):
  function average_bbox_lst (line 404) | def average_bbox_lst(bbox_lst):
  function prepare_paste_back (line 410) | def prepare_paste_back(mask_crop, crop_M_c2o, dsize):
  function paste_back (line 417) | def paste_back(img_crop, M_c2o, img_ori, mask_ori):

FILE: src/utils/cropper.py
  function make_abs_path (line 24) | def make_abs_path(fn):
  class Trajectory (line 29) | class Trajectory:
  class Cropper (line 41) | class Cropper(object):
    method __init__ (line 42) | def __init__(self, **kwargs) -> None:
    method update_config (line 88) | def update_config(self, user_args):
    method crop_source_image (line 93) | def crop_source_image(self, img_rgb_: np.ndarray, crop_cfg: CropConfig):
    method calc_lmk_from_cropped_image (line 153) | def calc_lmk_from_cropped_image(self, img_rgb_, **kwargs):
    method crop_source_video (line 172) | def crop_source_video(self, source_rgb_lst, crop_cfg: CropConfig, **kw...
    method crop_driving_video (line 225) | def crop_driving_video(self, driving_rgb_lst, **kwargs):
    method calc_lmks_from_cropped_video (line 285) | def calc_lmks_from_cropped_video(self, driving_rgb_crop_lst, **kwargs):

FILE: src/utils/dependencies/XPose/models/UniPose/attention.py
  class MultiheadAttention (line 39) | class MultiheadAttention(Module):
    method __init__ (line 65) | def __init__(self, embed_dim, num_heads, dropout=0., bias=True, add_bi...
    method _reset_parameters (line 91) | def _reset_parameters(self):
    method __setstate__ (line 94) | def __setstate__(self, state):
    method forward (line 101) | def forward(self, query, key, value, key_padding_mask=None,
  function multi_head_attention_forward (line 163) | def multi_head_attention_forward(query: Tensor,

FILE: src/utils/dependencies/XPose/models/UniPose/backbone.py
  class FrozenBatchNorm2d (line 31) | class FrozenBatchNorm2d(torch.nn.Module):
    method __init__ (line 40) | def __init__(self, n):
    method _load_from_state_dict (line 47) | def _load_from_state_dict(
    method forward (line 58) | def forward(self, x):
  class BackboneBase (line 71) | class BackboneBase(nn.Module):
    method __init__ (line 72) | def __init__(
    method forward (line 98) | def forward(self, tensor_list: NestedTensor):
  class Backbone (line 110) | class Backbone(BackboneBase):
    method __init__ (line 113) | def __init__(
  class Joiner (line 137) | class Joiner(nn.Sequential):
    method __init__ (line 138) | def __init__(self, backbone, position_embedding):
    method forward (line 141) | def forward(self, tensor_list: NestedTensor):
  function build_backbone (line 153) | def build_backbone(args):

FILE: src/utils/dependencies/XPose/models/UniPose/deformable_transformer.py
  class DeformableTransformer (line 33) | class DeformableTransformer(nn.Module):
    method __init__ (line 35) | def __init__(self, d_model=256, nhead=8,
    method _reset_parameters (line 280) | def _reset_parameters(self):
    method get_valid_ratio (line 293) | def get_valid_ratio(self, mask):
    method init_ref_points (line 302) | def init_ref_points(self, use_num_queries):
    method forward (line 311) | def forward(self, srcs, masks, refpoint_embed, pos_embeds, tgt, attn_m...
  class TransformerEncoder (line 525) | class TransformerEncoder(nn.Module):
    method __init__ (line 527) | def __init__(self,
    method get_reference_points (line 579) | def get_reference_points(spatial_shapes, valid_ratios, device):
    method forward (line 592) | def forward(self,
  class TransformerDecoder (line 690) | class TransformerDecoder(nn.Module):
    method __init__ (line 692) | def __init__(self, decoder_layer, num_layers, norm=None,
    method forward (line 769) | def forward(self, tgt, memory,
  class DeformableTransformerEncoderLayer (line 938) | class DeformableTransformerEncoderLayer(nn.Module):
    method __init__ (line 939) | def __init__(self,
    method with_pos_embed (line 969) | def with_pos_embed(tensor, pos):
    method forward_ffn (line 972) | def forward_ffn(self, src):
    method forward (line 978) | def forward(self, src, pos, reference_points, spatial_shapes, level_st...
  class DeformableTransformerDecoderLayer (line 996) | class DeformableTransformerDecoderLayer(nn.Module):
    method __init__ (line 997) | def __init__(self, d_model=256, d_ffn=1024,
    method rm_self_attn_modules (line 1041) | def rm_self_attn_modules(self):
    method with_pos_embed (line 1047) | def with_pos_embed(tensor, pos):
    method forward_ffn (line 1050) | def forward_ffn(self, tgt, ipdb_flag=False):
    method forward (line 1059) | def forward(self,
  function _get_clones (line 1135) | def _get_clones(module, N, layer_share=False):
  function build_deformable_transformer (line 1143) | def build_deformable_transformer(args):

FILE: src/utils/dependencies/XPose/models/UniPose/fuse_modules.py
  class FeatureResizer (line 8) | class FeatureResizer(nn.Module):
    method __init__ (line 14) | def __init__(self, input_feat_size, output_feat_size, dropout, do_ln=T...
    method forward (line 22) | def forward(self, encoder_features):
  function l1norm (line 32) | def l1norm(X, dim, eps=1e-8):
  function l2norm (line 40) | def l2norm(X, dim, eps=1e-8):
  function func_attention (line 48) | def func_attention(query, context, smooth=1, raw_feature_norm="softmax",...
  class BiMultiHeadAttention (line 97) | class BiMultiHeadAttention(nn.Module):
    method __init__ (line 98) | def __init__(self, v_dim, l_dim, embed_dim, num_heads, dropout=0.1, cf...
    method _shape (line 127) | def _shape(self, tensor: torch.Tensor, seq_len: int, bsz: int):
    method _reset_parameters (line 130) | def _reset_parameters(self):
    method forward (line 144) | def forward(self, v, l, attention_mask_v=None, attention_mask_l=None):
  class BiAttentionBlock (line 240) | class BiAttentionBlock(nn.Module):
    method __init__ (line 241) | def __init__(self, v_dim, l_dim, embed_dim, num_heads, dropout=0.1,
    method forward (line 267) | def forward(self, v, l, attention_mask_v=None, attention_mask_l=None):

FILE: src/utils/dependencies/XPose/models/UniPose/mask_generate.py
  function prepare_for_mask (line 4) | def prepare_for_mask(kpt_mask):
  function post_process (line 40) | def post_process(outputs_class, outputs_coord, dn_meta, aux_loss, _set_a...

FILE: src/utils/dependencies/XPose/models/UniPose/ops/functions/ms_deform_attn_func.py
  class MSDeformAttnFunction (line 21) | class MSDeformAttnFunction(Function):
    method forward (line 23) | def forward(ctx, value, value_spatial_shapes, value_level_start_index,...
    method backward (line 32) | def backward(ctx, grad_output):
  function ms_deform_attn_core_pytorch (line 41) | def ms_deform_attn_core_pytorch(value, value_spatial_shapes, sampling_lo...

FILE: src/utils/dependencies/XPose/models/UniPose/ops/modules/ms_deform_attn.py
  function _is_power_of_2 (line 26) | def _is_power_of_2(n):
  class MSDeformAttn (line 32) | class MSDeformAttn(nn.Module):
    method __init__ (line 33) | def __init__(self, d_model=256, n_levels=4, n_heads=8, n_points=4, use...
    method _reset_parameters (line 66) | def _reset_parameters(self):
    method forward (line 82) | def forward(self, query, reference_points, input_flatten, input_spatia...

FILE: src/utils/dependencies/XPose/models/UniPose/ops/modules/ms_deform_attn_key_aware.py
  function _is_power_of_2 (line 27) | def _is_power_of_2(n):
  class MSDeformAttn (line 33) | class MSDeformAttn(nn.Module):
    method __init__ (line 34) | def __init__(self, d_model=256, n_levels=4, n_heads=8, n_points=4, use...
    method _reset_parameters (line 67) | def _reset_parameters(self):
    method forward (line 83) | def forward(self, query, key, reference_points, input_flatten, input_s...

FILE: src/utils/dependencies/XPose/models/UniPose/ops/setup.py
  function get_extensions (line 23) | def get_extensions():

FILE: src/utils/dependencies/XPose/models/UniPose/ops/src/cpu/ms_deform_attn_cpu.cpp
  function ms_deform_attn_cpu_forward (line 17) | at::Tensor
  function ms_deform_attn_cpu_backward (line 29) | std::vector<at::Tensor>

FILE: src/utils/dependencies/XPose/models/UniPose/ops/src/ms_deform_attn.h
  function im2col_step (line 27) | int im2col_step)

FILE: src/utils/dependencies/XPose/models/UniPose/ops/src/vision.cpp
  function PYBIND11_MODULE (line 13) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: src/utils/dependencies/XPose/models/UniPose/ops/test.py
  function check_forward_equal_with_pytorch_double (line 32) | def check_forward_equal_with_pytorch_double():
  function check_forward_equal_with_pytorch_float (line 48) | def check_forward_equal_with_pytorch_float():
  function check_gradient_numerical (line 63) | def check_gradient_numerical(channels=4, grad_value=True, grad_sampling_...

FILE: src/utils/dependencies/XPose/models/UniPose/position_encoding.py
  class PositionEmbeddingSine (line 24) | class PositionEmbeddingSine(nn.Module):
    method __init__ (line 29) | def __init__(self, num_pos_feats=64, temperature=10000, normalize=Fals...
    method forward (line 40) | def forward(self, tensor_list: NestedTensor):
  class PositionEmbeddingSineHW (line 66) | class PositionEmbeddingSineHW(nn.Module):
    method __init__ (line 71) | def __init__(self, num_pos_feats=64, temperatureH=10000, temperatureW=...
    method forward (line 83) | def forward(self, tensor_list: NestedTensor):
  class PositionEmbeddingLearned (line 114) | class PositionEmbeddingLearned(nn.Module):
    method __init__ (line 118) | def __init__(self, num_pos_feats=256):
    method reset_parameters (line 124) | def reset_parameters(self):
    method forward (line 128) | def forward(self, tensor_list: NestedTensor):
  function build_position_encoding (line 142) | def build_position_encoding(args):

FILE: src/utils/dependencies/XPose/models/UniPose/swin_transformer.py
  class Mlp (line 14) | class Mlp(nn.Module):
    method __init__ (line 17) | def __init__(self, in_features, hidden_features=None, out_features=Non...
    method forward (line 26) | def forward(self, x):
  function window_partition (line 35) | def window_partition(x, window_size):
  function window_reverse (line 49) | def window_reverse(windows, window_size, H, W):
  class WindowAttention (line 65) | class WindowAttention(nn.Module):
    method __init__ (line 78) | def __init__(self, dim, window_size, num_heads, qkv_bias=True, qk_scal...
    method forward (line 112) | def forward(self, x, mask=None):
  class SwinTransformerBlock (line 146) | class SwinTransformerBlock(nn.Module):
    method __init__ (line 163) | def __init__(self, dim, num_heads, window_size=7, shift_size=0,
    method forward (line 187) | def forward(self, x, mask_matrix):
  class PatchMerging (line 246) | class PatchMerging(nn.Module):
    method __init__ (line 252) | def __init__(self, dim, norm_layer=nn.LayerNorm):
    method forward (line 258) | def forward(self, x, H, W):
  class BasicLayer (line 287) | class BasicLayer(nn.Module):
    method __init__ (line 305) | def __init__(self,
    method forward (line 347) | def forward(self, x, H, W):
  class PatchEmbed (line 389) | class PatchEmbed(nn.Module):
    method __init__ (line 398) | def __init__(self, patch_size=4, in_chans=3, embed_dim=96, norm_layer=...
    method forward (line 412) | def forward(self, x):
  class SwinTransformer (line 431) | class SwinTransformer(nn.Module):
    method __init__ (line 460) | def __init__(self,
    method _freeze_stages (line 553) | def _freeze_stages(self):
    method forward_raw (line 572) | def forward_raw(self, x):
    method forward (line 605) | def forward(self, tensor_list: NestedTensor):
    method train (line 648) | def train(self, mode=True):
  function build_swin_transformer (line 655) | def build_swin_transformer(modelname, pretrain_img_size, **kw):

FILE: src/utils/dependencies/XPose/models/UniPose/transformer_deformable.py
  class DeformableTransformer (line 25) | class DeformableTransformer(nn.Module):
    method __init__ (line 26) | def __init__(self, d_model=256, nhead=8,
    method _reset_parameters (line 68) | def _reset_parameters(self):
    method get_proposal_pos_embed (line 80) | def get_proposal_pos_embed(self, proposals):
    method gen_encoder_output_proposals (line 95) | def gen_encoder_output_proposals(self, memory, memory_padding_mask, sp...
    method get_valid_ratio (line 127) | def get_valid_ratio(self, mask):
    method forward (line 136) | def forward(self, srcs, masks, pos_embeds, query_embed=None):
  class DeformableTransformerEncoderLayer (line 215) | class DeformableTransformerEncoderLayer(nn.Module):
    method __init__ (line 216) | def __init__(self,
    method with_pos_embed (line 249) | def with_pos_embed(tensor, pos):
    method forward_ffn (line 252) | def forward_ffn(self, src):
    method forward (line 258) | def forward(self, src, pos, reference_points, spatial_shapes, level_st...
  class DeformableTransformerEncoder (line 275) | class DeformableTransformerEncoder(nn.Module):
    method __init__ (line 276) | def __init__(self, encoder_layer, num_layers, norm=None):
    method get_reference_points (line 287) | def get_reference_points(spatial_shapes, valid_ratios, device):
    method forward (line 301) | def forward(self, src, spatial_shapes, level_start_index, valid_ratios...
  class DeformableTransformerDecoderLayer (line 327) | class DeformableTransformerDecoderLayer(nn.Module):
    method __init__ (line 328) | def __init__(self, d_model=256, d_ffn=1024,
    method rm_self_attn_modules (line 374) | def rm_self_attn_modules(self):
    method with_pos_embed (line 381) | def with_pos_embed(tensor, pos):
    method forward_ffn (line 384) | def forward_ffn(self, tgt):
    method forward_sa (line 390) | def forward_sa(self,
    method forward_ca (line 436) | def forward_ca(self,
    method forward (line 473) | def forward(self,
  class DeformableTransformerDecoder (line 515) | class DeformableTransformerDecoder(nn.Module):
    method __init__ (line 516) | def __init__(self, decoder_layer, num_layers, return_intermediate=Fals...
    method forward (line 533) | def forward(self, tgt, reference_points, src, src_spatial_shapes,
  function _get_clones (line 574) | def _get_clones(module, N):
  function build_deforamble_transformer (line 578) | def build_deforamble_transformer(args):

FILE: src/utils/dependencies/XPose/models/UniPose/transformer_vanilla.py
  class TextTransformer (line 18) | class TextTransformer(nn.Module):
    method __init__ (line 19) | def __init__(self, num_layers, d_model=256, nheads=8, dim_feedforward=...
    method forward (line 31) | def forward(self, memory_text:torch.Tensor, text_attention_mask:torch....
  class TransformerEncoderLayer (line 58) | class TransformerEncoderLayer(nn.Module):
    method __init__ (line 59) | def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1, ...
    method with_pos_embed (line 76) | def with_pos_embed(self, tensor, pos: Optional[Tensor]):
    method forward (line 79) | def forward(

FILE: src/utils/dependencies/XPose/models/UniPose/unipose.py
  class UniPose (line 26) | class UniPose(nn.Module):
    method __init__ (line 29) | def __init__(self, backbone, transformer, num_classes, num_queries,
    method open_set_transfer_init (line 242) | def open_set_transfer_init(self):
    method _reset_parameters (line 269) | def _reset_parameters(self):
    method init_ref_points (line 275) | def init_ref_points(self, use_num_queries):
    method forward (line 303) | def forward(self, samples: NestedTensor, targets: List = None, **kw):
  function build_unipose (line 492) | def build_unipose(args):
  class ContrastiveAssign (line 573) | class ContrastiveAssign(nn.Module):
    method __init__ (line 574) | def __init__(self, project=False, cal_bias=None, max_text_len=256):
    method forward (line 586) | def forward(self, x, text_dict):

FILE: src/utils/dependencies/XPose/models/UniPose/utils.py
  function _get_clones (line 18) | def _get_clones(module, N, layer_share=False):
  function get_sine_pos_embed (line 26) | def get_sine_pos_embed(
  function gen_encoder_output_proposals (line 58) | def gen_encoder_output_proposals(memory: Tensor, memory_padding_mask: Te...
  class RandomBoxPerturber (line 116) | class RandomBoxPerturber():
    method __init__ (line 117) | def __init__(self, x_noise_scale=0.2, y_noise_scale=0.2, w_noise_scale...
    method __call__ (line 120) | def __call__(self, refanchors: Tensor) -> Tensor:
  function sigmoid_focal_loss (line 131) | def sigmoid_focal_loss(inputs, targets, num_boxes, alpha: float = 0.25, ...
  class MLP (line 162) | class MLP(nn.Module):
    method __init__ (line 165) | def __init__(self, input_dim, hidden_dim, output_dim, num_layers):
    method forward (line 171) | def forward(self, x):
  function _get_activation_fn (line 177) | def _get_activation_fn(activation, d_model=256, batch_dim=0):
  function gen_sineembed_for_position (line 193) | def gen_sineembed_for_position(pos_tensor):
  function oks_overlaps (line 222) | def oks_overlaps(kpt_preds, kpt_gts, kpt_valids, kpt_areas, sigmas):
  function oks_loss (line 243) | def oks_loss(pred,
  class OKSLoss (line 271) | class OKSLoss(nn.Module):
    method __init__ (line 282) | def __init__(self,
    method forward (line 304) | def forward(self,

FILE: src/utils/dependencies/XPose/models/__init__.py
  function build_model (line 9) | def build_model(args):

FILE: src/utils/dependencies/XPose/models/registry.py
  class Registry (line 12) | class Registry(object):
    method __init__ (line 14) | def __init__(self, name):
    method __repr__ (line 18) | def __repr__(self):
    method __len__ (line 23) | def __len__(self):
    method name (line 27) | def name(self):
    method module_dict (line 31) | def module_dict(self):
    method get (line 34) | def get(self, key):
    method registe_with_name (line 37) | def registe_with_name(self, module_name=None, force=False):
    method register (line 40) | def register(self, module_build_function, module_name=None, force=False):

FILE: src/utils/dependencies/XPose/transforms.py
  function crop (line 19) | def crop(image, target, region):
  function hflip (line 83) | def hflip(image, target):
  function resize (line 148) | def resize(image, target, size, max_size=None):
  function pad (line 213) | def pad(image, target, padding):
  class ResizeDebug (line 226) | class ResizeDebug(object):
    method __init__ (line 227) | def __init__(self, size):
    method __call__ (line 230) | def __call__(self, img, target):
  class RandomCrop (line 234) | class RandomCrop(object):
    method __init__ (line 235) | def __init__(self, size):
    method __call__ (line 238) | def __call__(self, img, target):
  class RandomSizeCrop (line 243) | class RandomSizeCrop(object):
    method __init__ (line 244) | def __init__(self, min_size: int, max_size: int, respect_boxes: bool =...
    method __call__ (line 251) | def __call__(self, img: PIL.Image.Image, target: dict):
  class CenterCrop (line 265) | class CenterCrop(object):
    method __init__ (line 266) | def __init__(self, size):
    method __call__ (line 269) | def __call__(self, img, target):
  class RandomHorizontalFlip (line 277) | class RandomHorizontalFlip(object):
    method __init__ (line 278) | def __init__(self, p=0.5):
    method __call__ (line 281) | def __call__(self, img, target):
  class RandomResize (line 287) | class RandomResize(object):
    method __init__ (line 288) | def __init__(self, sizes, max_size=None):
    method __call__ (line 293) | def __call__(self, img, target=None):
  class RandomPad (line 298) | class RandomPad(object):
    method __init__ (line 299) | def __init__(self, max_pad):
    method __call__ (line 302) | def __call__(self, img, target):
  class RandomSelect (line 308) | class RandomSelect(object):
    method __init__ (line 313) | def __init__(self, transforms1, transforms2, p=0.5):
    method __call__ (line 318) | def __call__(self, img, target):
  class ToTensor (line 324) | class ToTensor(object):
    method __call__ (line 325) | def __call__(self, img, target):
  class RandomErasing (line 329) | class RandomErasing(object):
    method __init__ (line 331) | def __init__(self, *args, **kwargs):
    method __call__ (line 334) | def __call__(self, img, target):
  class Normalize (line 338) | class Normalize(object):
    method __init__ (line 339) | def __init__(self, mean, std):
    method __call__ (line 343) | def __call__(self, image, target=None):
  class Compose (line 379) | class Compose(object):
    method __init__ (line 380) | def __init__(self, transforms):
    method __call__ (line 383) | def __call__(self, image, target):
    method __repr__ (line 388) | def __repr__(self):

FILE: src/utils/dependencies/XPose/util/addict.py
  class Dict (line 4) | class Dict(dict):
    method __init__ (line 6) | def __init__(__self, *args, **kwargs):
    method __setattr__ (line 25) | def __setattr__(self, name, value):
    method __setitem__ (line 32) | def __setitem__(self, name, value):
    method __add__ (line 49) | def __add__(self, other):
    method _hook (line 59) | def _hook(cls, item):
    method __getattr__ (line 66) | def __getattr__(self, item):
    method __missing__ (line 69) | def __missing__(self, name):
    method __delattr__ (line 74) | def __delattr__(self, name):
    method to_dict (line 77) | def to_dict(self):
    method copy (line 90) | def copy(self):
    method deepcopy (line 93) | def deepcopy(self):
    method __deepcopy__ (line 96) | def __deepcopy__(self, memo):
    method update (line 103) | def update(self, *args, **kwargs):
    method __getnewargs__ (line 118) | def __getnewargs__(self):
    method __getstate__ (line 121) | def __getstate__(self):
    method __setstate__ (line 124) | def __setstate__(self, state):
    method __or__ (line 127) | def __or__(self, other):
    method __ror__ (line 134) | def __ror__(self, other):
    method __ior__ (line 141) | def __ior__(self, other):
    method setdefault (line 145) | def setdefault(self, key, default=None):
    method freeze (line 152) | def freeze(self, shouldFreeze=True):
    method unfreeze (line 158) | def unfreeze(self):

FILE: src/utils/dependencies/XPose/util/box_ops.py
  function box_cxcywh_to_xyxy (line 9) | def box_cxcywh_to_xyxy(x):
  function box_xyxy_to_cxcywh (line 16) | def box_xyxy_to_cxcywh(x):
  function box_iou (line 24) | def box_iou(boxes1, boxes2):
  function generalized_box_iou (line 41) | def generalized_box_iou(boxes1, boxes2):
  function box_iou_pairwise (line 69) | def box_iou_pairwise(boxes1, boxes2):
  function generalized_box_iou_pairwise (line 85) | def generalized_box_iou_pairwise(boxes1, boxes2):
  function masks_to_boxes (line 109) | def masks_to_boxes(masks):

FILE: src/utils/dependencies/XPose/util/config.py
  function check_file_exist (line 19) | def check_file_exist(filename, msg_tmpl='file "{}" does not exist'):
  class ConfigDict (line 23) | class ConfigDict(Dict):
    method __missing__ (line 25) | def __missing__(self, name):
    method __getattr__ (line 28) | def __getattr__(self, name):
  class Config (line 41) | class Config(object):
    method _validate_py_syntax (line 66) | def _validate_py_syntax(filename):
    method _file2dict (line 76) | def _file2dict(filename):
    method _merge_a_into_b (line 143) | def _merge_a_into_b(a, b):
    method fromfile (line 187) | def fromfile(filename):
    method __init__ (line 192) | def __init__(self, cfg_dict=None, cfg_text=None, filename=None):
    method filename (line 215) | def filename(self):
    method text (line 219) | def text(self):
    method pretty_text (line 223) | def pretty_text(self):
    method __repr__ (line 310) | def __repr__(self):
    method __len__ (line 313) | def __len__(self):
    method __getattr__ (line 316) | def __getattr__(self, name):
    method __getitem__ (line 329) | def __getitem__(self, name):
    method __setattr__ (line 332) | def __setattr__(self, name, value):
    method __setitem__ (line 337) | def __setitem__(self, name, value):
    method __iter__ (line 342) | def __iter__(self):
    method dump (line 345) | def dump(self, file=None):
    method merge_from_dict (line 353) | def merge_from_dict(self, options):
    method __setstate__ (line 385) | def __setstate__(self, state):
    method copy (line 389) | def copy(self):
    method deepcopy (line 392) | def deepcopy(self):
  class DictAction (line 396) | class DictAction(Action):
    method _parse_int_float_bool (line 404) | def _parse_int_float_bool(val):
    method __call__ (line 419) | def __call__(self, parser, namespace, values, option_string=None):

FILE: src/utils/dependencies/XPose/util/keypoint_ops.py
  function keypoint_xyxyzz_to_xyzxyz (line 3) | def keypoint_xyxyzz_to_xyzxyz(keypoints: torch.Tensor):
  function keypoint_xyzxyz_to_xyxyzz (line 18) | def keypoint_xyzxyz_to_xyxyzz(keypoints: torch.Tensor):

FILE: src/utils/dependencies/XPose/util/misc.py
  class SmoothedValue (line 34) | class SmoothedValue(object):
    method __init__ (line 39) | def __init__(self, window_size=20, fmt=None):
    method update (line 47) | def update(self, value, n=1):
    method synchronize_between_processes (line 52) | def synchronize_between_processes(self):
    method median (line 66) | def median(self):
    method avg (line 73) | def avg(self):
    method global_avg (line 78) | def global_avg(self):
    method max (line 86) | def max(self):
    method value (line 90) | def value(self):
    method __str__ (line 93) | def __str__(self):
  function _get_global_gloo_group (line 102) | def _get_global_gloo_group():
  function all_gather_cpu (line 113) | def all_gather_cpu(data):
  function all_gather (line 171) | def all_gather(data):
  function reduce_dict (line 220) | def reduce_dict(input_dict, average=True):
  class MetricLogger (line 247) | class MetricLogger(object):
    method __init__ (line 248) | def __init__(self, delimiter="\t"):
    method update (line 252) | def update(self, **kwargs):
    method __getattr__ (line 259) | def __getattr__(self, attr):
    method __str__ (line 267) | def __str__(self):
    method synchronize_between_processes (line 278) | def synchronize_between_processes(self):
    method add_meter (line 282) | def add_meter(self, name, meter):
    method log_every (line 285) | def log_every(self, iterable, print_freq, header=None, logger=None):
  function get_sha (line 346) | def get_sha():
  function collate_fn (line 366) | def collate_fn(batch):
  function _max_by_axis (line 373) | def _max_by_axis(the_list):
  class NestedTensor (line 382) | class NestedTensor(object):
    method __init__ (line 383) | def __init__(self, tensors, mask: Optional[Tensor]):
    method imgsize (line 395) | def imgsize(self):
    method to (line 404) | def to(self, device):
    method to_img_list_single (line 415) | def to_img_list_single(self, tensor, mask):
    method to_img_list (line 422) | def to_img_list(self):
    method device (line 439) | def device(self):
    method decompose (line 442) | def decompose(self):
    method __repr__ (line 445) | def __repr__(self):
    method shape (line 449) | def shape(self):
  function nested_tensor_from_tensor_list (line 456) | def nested_tensor_from_tensor_list(tensor_list: List[Tensor]):
  function _onnx_nested_tensor_from_tensor_list (line 484) | def _onnx_nested_tensor_from_tensor_list(tensor_list: List[Tensor]) -> N...
  function setup_for_distributed (line 512) | def setup_for_distributed(is_master):
  function is_dist_avail_and_initialized (line 527) | def is_dist_avail_and_initialized():
  function get_world_size (line 535) | def get_world_size():
  function get_rank (line 541) | def get_rank():
  function is_main_process (line 547) | def is_main_process():
  function save_on_master (line 551) | def save_on_master(*args, **kwargs):
  function init_distributed_mode (line 555) | def init_distributed_mode(args):
  function accuracy (line 621) | def accuracy(output, target, topk=(1,)):
  function accuracy_onehot (line 639) | def accuracy_onehot(pred, gt):
  function interpolate (line 654) | def interpolate(input, size=None, scale_factor=None, mode="nearest", ali...
  class color_sys (line 675) | class color_sys():
    method __init__ (line 676) | def __init__(self, num_colors) -> None:
    method __call__ (line 686) | def __call__(self, idx):
  function inverse_sigmoid (line 689) | def inverse_sigmoid(x, eps=1e-3):
  function clean_state_dict (line 695) | def clean_state_dict(state_dict):

FILE: src/utils/dependencies/insightface/app/common.py
  class Face (line 5) | class Face(dict):
    method __init__ (line 7) | def __init__(self, d=None, **kwargs):
    method __setattr__ (line 19) | def __setattr__(self, name, value):
    method __getattr__ (line 30) | def __getattr__(self, name):
    method embedding_norm (line 34) | def embedding_norm(self):
    method normed_embedding (line 40) | def normed_embedding(self):
    method sex (line 46) | def sex(self):

FILE: src/utils/dependencies/insightface/app/face_analysis.py
  class FaceAnalysis (line 25) | class FaceAnalysis:
    method __init__ (line 26) | def __init__(self, name=DEFAULT_MP_NAME, root='~/.insightface', allowe...
    method prepare (line 49) | def prepare(self, ctx_id, det_thresh=0.5, det_size=(640, 640)):
    method get (line 60) | def get(self, img, max_num=0):
    method draw_on (line 81) | def draw_on(self, img, faces):

FILE: src/utils/dependencies/insightface/data/image.py
  class ImageCache (line 6) | class ImageCache:
  function get_image (line 9) | def get_image(name, to_rgb=False):

FILE: src/utils/dependencies/insightface/data/pickle_object.py
  function get_object (line 7) | def get_object(name):

FILE: src/utils/dependencies/insightface/data/rec_builder.py
  class RecBuilder (line 9) | class RecBuilder():
    method __init__ (line 10) | def __init__(self, path, image_size=(112, 112)):
    method add (line 23) | def add(self, imgs):
    method add_image (line 44) | def add_image(self, img, label):
    method close (line 64) | def close(self):

FILE: src/utils/dependencies/insightface/model_zoo/arcface_onnx.py
  class ArcFaceONNX (line 19) | class ArcFaceONNX:
    method __init__ (line 20) | def __init__(self, model_file=None, session=None):
    method prepare (line 61) | def prepare(self, ctx_id, **kwargs):
    method get (line 65) | def get(self, img, face):
    method compute_sim (line 70) | def compute_sim(self, feat1, feat2):
    method get_feat (line 77) | def get_feat(self, imgs):
    method forward (line 87) | def forward(self, batch_data):

FILE: src/utils/dependencies/insightface/model_zoo/attribute.py
  class Attribute (line 19) | class Attribute:
    method __init__ (line 20) | def __init__(self, model_file=None, session=None):
    method prepare (line 68) | def prepare(self, ctx_id, **kwargs):
    method get (line 72) | def get(self, img, face):

FILE: src/utils/dependencies/insightface/model_zoo/inswapper.py
  class INSwapper (line 12) | class INSwapper():
    method __init__ (line 13) | def __init__(self, model_file=None, session=None):
    method forward (line 41) | def forward(self, img, latent):
    method get (line 46) | def get(self, img, target_face, source_face, paste_back=True):

FILE: src/utils/dependencies/insightface/model_zoo/landmark.py
  class Landmark (line 21) | class Landmark:
    method __init__ (line 22) | def __init__(self, model_file=None, session=None):
    method prepare (line 76) | def prepare(self, ctx_id, **kwargs):
    method get (line 80) | def get(self, img, face):

FILE: src/utils/dependencies/insightface/model_zoo/model_store.py
  function short_hash (line 29) | def short_hash(name):
  function find_params_file (line 36) | def find_params_file(dir_path):
  function get_model_file (line 46) | def get_model_file(name, root=os.path.join('~', '.insightface', 'models')):

FILE: src/utils/dependencies/insightface/model_zoo/model_zoo.py
  class PickableInferenceSession (line 22) | class PickableInferenceSession(onnxruntime.InferenceSession):
    method __init__ (line 24) | def __init__(self, model_path, **kwargs):
    method __getstate__ (line 28) | def __getstate__(self):
    method __setstate__ (line 31) | def __setstate__(self, values):
  class ModelRouter (line 35) | class ModelRouter:
    method __init__ (line 36) | def __init__(self, onnx_file):
    method get_model (line 39) | def get_model(self, **kwargs):
  function find_onnx_file (line 61) | def find_onnx_file(dir_path):
  function get_default_providers (line 70) | def get_default_providers():
  function get_default_provider_options (line 73) | def get_default_provider_options():
  function get_model (line 76) | def get_model(name, **kwargs):

FILE: src/utils/dependencies/insightface/model_zoo/retinaface.py
  function softmax (line 17) | def softmax(z):
  function distance2bbox (line 26) | def distance2bbox(points, distance, max_shape=None):
  function distance2kps (line 49) | def distance2kps(points, distance, max_shape=None):
  class RetinaFace (line 72) | class RetinaFace:
    method __init__ (line 73) | def __init__(self, model_file=None, session=None):
    method _init_vars (line 87) | def _init_vars(self):
    method prepare (line 130) | def prepare(self, ctx_id, **kwargs):
    method forward (line 146) | def forward(self, img, threshold):
    method detect (line 207) | def detect(self, img, input_size = None, max_num=0, metric='default'):
    method nms (line 262) | def nms(self, dets):
  function get_retinaface (line 292) | def get_retinaface(name, download=False, root='~/.insightface/models', *...

FILE: src/utils/dependencies/insightface/model_zoo/scrfd.py
  function softmax (line 17) | def softmax(z):
  function distance2bbox (line 26) | def distance2bbox(points, distance, max_shape=None):
  function distance2kps (line 49) | def distance2kps(points, distance, max_shape=None):
  class SCRFD (line 72) | class SCRFD:
    method __init__ (line 73) | def __init__(self, model_file=None, session=None):
    method _init_vars (line 88) | def _init_vars(self):
    method prepare (line 133) | def prepare(self, ctx_id, **kwargs):
    method forward (line 149) | def forward(self, img, threshold):
    method detect (line 220) | def detect(self, img, input_size = None, max_num=0, metric='default'):
    method nms (line 275) | def nms(self, dets):
  function get_scrfd (line 305) | def get_scrfd(name, download=False, root='~/.insightface/models', **kwar...
  function scrfd_2p5gkps (line 315) | def scrfd_2p5gkps(**kwargs):

FILE: src/utils/dependencies/insightface/utils/download.py
  function check_sha1 (line 10) | def check_sha1(filename, sha1_hash):
  function download_file (line 36) | def download_file(url, path=None, overwrite=False, sha1_hash=None):

FILE: src/utils/dependencies/insightface/utils/face_align.py
  function estimate_norm (line 11) | def estimate_norm(lmk, image_size=112,mode='arcface'):
  function norm_crop (line 27) | def norm_crop(img, landmark, image_size=112, mode='arcface'):
  function norm_crop2 (line 32) | def norm_crop2(img, landmark, image_size=112, mode='arcface'):
  function square_crop (line 37) | def square_crop(im, S):
  function transform (line 52) | def transform(data, center, output_size, scale, rotation):
  function trans_points2d (line 71) | def trans_points2d(pts, M):
  function trans_points3d (line 83) | def trans_points3d(pts, M):
  function trans_points (line 98) | def trans_points(pts, M):

FILE: src/utils/dependencies/insightface/utils/filesystem.py
  function get_model_dir (line 9) | def get_model_dir(name, root='~/.insightface'):
  function makedirs (line 14) | def makedirs(path):
  function try_import (line 30) | def try_import(package, message=None):
  function try_import_cv2 (line 54) | def try_import_cv2():
  function try_import_mmcv (line 68) | def try_import_mmcv():
  function try_import_rarfile (line 82) | def try_import_rarfile():
  function import_try_install (line 96) | def import_try_install(package, extern_url=None):
  function try_import_dali (line 142) | def try_import_dali():

FILE: src/utils/dependencies/insightface/utils/storage.py
  function download (line 9) | def download(sub_dir, name, force=False, root='~/.insightface'):
  function ensure_available (line 27) | def ensure_available(sub_dir, name, root='~/.insightface'):
  function download_onnx (line 30) | def download_onnx(sub_dir, model_file, force=False, root='~/.insightface...

FILE: src/utils/dependencies/insightface/utils/transform.py
  function transform (line 7) | def transform(data, center, output_size, scale, rotation):
  function trans_points2d (line 26) | def trans_points2d(pts, M):
  function trans_points3d (line 38) | def trans_points3d(pts, M):
  function trans_points (line 53) | def trans_points(pts, M):
  function estimate_affine_matrix_3d23d (line 59) | def estimate_affine_matrix_3d23d(X, Y):
  function P2sRt (line 71) | def P2sRt(P):
  function matrix2angle (line 91) | def matrix2angle(R):

FILE: src/utils/face_analysis_diy.py
  function sort_by_direction (line 14) | def sort_by_direction(faces, direction: str = 'large-small', face_center...
  class FaceAnalysisDIY (line 35) | class FaceAnalysisDIY(FaceAnalysis):
    method __init__ (line 36) | def __init__(self, name='buffalo_l', root='~/.insightface', allowed_mo...
    method get (line 41) | def get(self, img_bgr, **kwargs):
    method warmup (line 72) | def warmup(self):

FILE: src/utils/filter.py
  function smooth (line 8) | def smooth(x_d_lst, shape, device, observation_variance=3e-7, process_va...

FILE: src/utils/helper.py
  function tensor_to_numpy (line 23) | def tensor_to_numpy(data: Union[np.ndarray, torch.Tensor]) -> np.ndarray:
  function calc_motion_multiplier (line 29) | def calc_motion_multiplier(
  function suffix (line 44) | def suffix(filename):
  function prefix (line 52) | def prefix(filename):
  function basename (line 60) | def basename(filename):
  function remove_suffix (line 65) | def remove_suffix(filepath):
  function is_image (line 70) | def is_image(file_path):
  function is_video (line 75) | def is_video(file_path):
  function is_template (line 81) | def is_template(file_path):
  function mkdir (line 87) | def mkdir(d, log=False):
  function squeeze_tensor_to_numpy (line 96) | def squeeze_tensor_to_numpy(tensor):
  function dct2device (line 101) | def dct2device(dct: dict, device):
  function concat_feat (line 110) | def concat_feat(kp_source: torch.Tensor, kp_driving: torch.Tensor) -> to...
  function remove_ddp_dumplicate_key (line 124) | def remove_ddp_dumplicate_key(state_dict):
  function load_model (line 131) | def load_model(ckpt_path, model_config, device, model_type):
  function load_description (line 175) | def load_description(fp):
  function is_square_video (line 181) | def is_square_video(video_path):
  function clean_state_dict (line 193) | def clean_state_dict(state_dict):

FILE: src/utils/human_landmark_runner.py
  function make_abs_path (line 13) | def make_abs_path(fn):
  function to_ndarray (line 17) | def to_ndarray(obj):
  class LandmarkRunner (line 26) | class LandmarkRunner(object):
    method __init__ (line 29) | def __init__(self, **kwargs):
    method _run (line 56) | def _run(self, inp):
    method run (line 60) | def run(self, img_rgb: np.ndarray, lmk=None):
    method warmup (line 87) | def warmup(self):

FILE: src/utils/io.py
  function load_image_rgb (line 12) | def load_image_rgb(image_path: str):
  function load_video (line 19) | def load_video(video_info, n_frames=-1):
  function contiguous (line 32) | def contiguous(obj):
  function resize_to_limit (line 38) | def resize_to_limit(img: np.ndarray, max_dim=1920, division=2):
  function load_img_online (line 73) | def load_img_online(obj, mode="bgr", **kwargs):
  function load (line 95) | def load(fp):
  function dump (line 106) | def dump(wfp, obj):

FILE: src/utils/retargeting_utils.py
  function calculate_distance_ratio (line 9) | def calculate_distance_ratio(lmk: np.ndarray, idx1: int, idx2: int, idx3...
  function calc_eye_close_ratio (line 14) | def calc_eye_close_ratio(lmk: np.ndarray, target_eye_ratio: np.ndarray =...
  function calc_lip_close_ratio (line 23) | def calc_lip_close_ratio(lmk: np.ndarray) -> np.ndarray:

FILE: src/utils/timer.py
  class Timer (line 9) | class Timer(object):
    method __init__ (line 12) | def __init__(self):
    method tic (line 18) | def tic(self):
    method toc (line 23) | def toc(self, average=True):
    method clear (line 27) | def clear(self):

FILE: src/utils/video.py
  function exec_cmd (line 21) | def exec_cmd(cmd):
  function images2video (line 25) | def images2video(images, wfp, **kwargs):
  function video2gif (line 50) | def video2gif(video_fp, fps=30, size=256):
  function merge_audio_video (line 67) | def merge_audio_video(video_fp, audio_fp, wfp):
  function blend (line 76) | def blend(img: np.ndarray, mask: np.ndarray, background_color=(255, 255,...
  function concat_frames (line 84) | def concat_frames(driving_image_lst, source_image_lst, I_p_lst):
  class VideoWriter (line 105) | class VideoWriter:
    method __init__ (line 106) | def __init__(self, **kwargs):
    method write (line 122) | def write(self, image):
    method close (line 128) | def close(self):
  function change_video_fps (line 133) | def change_video_fps(input_file, output_file, fps=20, codec='libx264', c...
  function get_fps (line 138) | def get_fps(filepath, default_fps=25):
  function has_audio_stream (line 151) | def has_audio_stream(video_path: str) -> bool:
  function add_audio_to_video (line 189) | def add_audio_to_video(silent_video_path: str, audio_video_path: str, ou...
  function bb_intersection_over_union (line 209) | def bb_intersection_over_union(boxA, boxB):

FILE: src/utils/viz.py
  function viz_lmk (line 6) | def viz_lmk(img_, vps, **kwargs):