MCPcopy Create free account

hub / github.com/MyNiuuu/AniCrafter / functions

Functions5,947 in github.com/MyNiuuu/AniCrafter

↓ 2 callersFunctionget_hanning_kernel1d
r"""Returns Hanning (also known as Hann) kernel, used in signal processing and KCF tracker. .. math:: w(n) = 0.5 - 0.5cos\\left(\\frac{2\\pi{n}}
ProPainter/model/canny/kernels.py:642
↓ 2 callersFunctionget_i3d_activations
Get features from i3d model and flatten them to 1d feature, valid target endpoints are defined in InceptionI3d.VALID_ENDPOINTS VALID_ENDP
ProPainter/core/metrics.py:153
↓ 2 callersMethodget_joint_offset
(self, joint_offset)
LHM/models/rendering/smpl_x.py:296
↓ 2 callersMethodget_joint_offset
(self, joint_offset)
LHM/models/rendering/smpl_x_voxel_dense_sampling.py:357
↓ 2 callersMethodget_loss
Calculate keypoint loss. Note: - batch_size: N - num_keypoints: K Args: output (torch.Tensor[N,
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/temporal_regression_head.py:89
↓ 2 callersMethodget_name_dict
(self, lora_state_dict)
diffsynth/models/lora.py:205
↓ 2 callersFunctionget_normalized_camera_intrinsics
intrinsics: (N, 3, 2), [[fx, fy], [cx, cy], [width, height]] Return batched fx, fy, cx, cy
LHM/datasets/cam_utils.py:90
↓ 2 callersMethodget_num_verts
(self)
LHM/models/rendering/smplx/smplx/body_models.py:306
↓ 2 callersFunctionget_pairs
Return set of symbol pairs in a word. Word is represented as tuple of symbols (symbols being variable-length strings).
diffsynth/extensions/ImageQualityMetric/open_clip/tokenizer.py:51
↓ 2 callersMethodget_patch_error
(self, source, nnf, target)
diffsynth/extensions/FastBlend/patch_match.py:53
↓ 2 callersFunctionget_pretrained_cfg
(model: str, tag: str)
diffsynth/extensions/ImageQualityMetric/open_clip/pretrained.py:271
↓ 2 callersFunctionget_pretrained_url
(model: str, tag: str)
diffsynth/extensions/ImageQualityMetric/open_clip/pretrained.py:278
↓ 2 callersFunctionget_random_shape
There is the initial point and 3 points per cubic bezier curve. Thus, the curve will only pass though n points, which will be the sharp e
ProPainter/core/utils.py:268
↓ 2 callersMethodget_route_info
(self)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/message.py:81
↓ 2 callersMethodget_single_batch_smpl_data
(self, smpl_data, bidx)
LHM/models/rendering/gs_renderer.py:1291
↓ 2 callersMethodget_single_view_smpl_data
(self, smpl_data, vidx)
LHM/models/rendering/gs_renderer.py:1303
↓ 2 callersFunctionget_slurm_partition
(cluster_type: Optional[ClusterType] = None)
engine/pose_estimation/dinov2/dinov2/utils/cluster.py:61
↓ 2 callersFunctionget_spatial_gradient_kernel3d
r"""Function that returns kernel for 1st or 2nd order scale pyramid gradients, using one of the following operators: sobel, diff.
ProPainter/model/canny/kernels.py:318
↓ 2 callersFunctionget_state_dict
Returns a dictionary containing a whole state of the module. Both parameters and persistent buffers (e.g. running averages) are included. Key
engine/pose_estimation/third-party/ViTPose/mmcv_custom/checkpoint.py:446
↓ 2 callersFunctionget_timestep_embedding
This matches the implementation in Denoising Diffusion Probabilistic Models: Create sinusoidal timestep embeddings. :param timesteps: a 1-D
diffsynth/models/svd_unet.py:40
↓ 2 callersFunctionget_torchvision_models
()
engine/pose_estimation/third-party/ViTPose/mmcv_custom/checkpoint.py:178
↓ 2 callersMethodget_transform_mat_vertex
(self, transform_mat_joint, query_points, fix_mask)
LHM/models/rendering/smpl_x_voxel_dense_sampling.py:1057
↓ 2 callersMethodget_value
(self, step)
LHM/models/utils.py:19
↓ 2 callersMethodget_zero_pose_human
( self, shape_param, device, face_offset, joint_offset, return_mesh=False )
LHM/models/rendering/smpl_x_voxel_dense_sampling.py:959
↓ 2 callersFunctionhas_hf_hub
(necessary=False)
diffsynth/extensions/ImageQualityMetric/open_clip/pretrained.py:329
↓ 2 callersFunctionimfrombytes
Read an image from bytes. Args: content (bytes): Image bytes got from files or other streams. flag (str): Flags specifying the co
ProPainter/utils/img_util.py:114
↓ 2 callersFunctionimg_center_padding
(img_np, pad_ratio)
LHM/runners/infer/utils.py:82
↓ 2 callersMethodimg_center_padding
(self, img_np)
engine/pose_estimation/pose_estimator.py:75
↓ 2 callersFunctionimshow_mesh_3d
Render 3D meshes on background image. Args: img(np.ndarray): Background image. vertices (list of np.ndarray): Vetrex coordinates
engine/pose_estimation/third-party/ViTPose/mmpose/core/visualization/image.py:353
↓ 2 callersMethodinfer_single_view
( self, image, source_c2ws, source_intrs, render_c2ws, render_
LHM/models/modeling_human_lrm.py:437
↓ 2 callersMethodinference_model
Inference function. Returns: output_regression (np.ndarray): Output regression. Args: x (torch.Tensor[N, K,
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/deeppose_regression_head.py:97
↓ 2 callersMethodinit_offset
(self)
ProPainter/model/recurrent_flow_completion.py:27
↓ 2 callersMethodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/hourglass_ae.py:175
↓ 2 callersMethodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/hrnet.py:535
↓ 2 callersMethodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/litehrnet.py:935
↓ 2 callersMethodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/cpm.py:149
↓ 2 callersMethodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to No
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/hourglass.py:174
↓ 2 callersMethodinit_weights
Weight initialization for model.
engine/pose_estimation/third-party/ViTPose/mmpose/models/detectors/associative_embedding.py:75
↓ 2 callersMethodinit_weights
Initialize model weights.
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/vipnas_heatmap_simple_head.py:338
↓ 2 callersMethodinit_weights
Initialize model weights.
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/topdown_heatmap_simple_head.py:339
↓ 2 callersMethodinit_weights
Initialize model weights.
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/topdown_heatmap_multi_stage_head.py:253
↓ 2 callersMethodinit_weights
Initialize model weights.
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/hmr_head.py:90
↓ 2 callersMethodinitialize_tracking
(self, )
engine/pose_estimation/blocks/detector.py:48
↓ 2 callersFunctioninteract_with_ipadapter
(hidden_states, q, ip_k, ip_v, scale=1.0)
diffsynth/models/flux_dit.py:7
↓ 2 callersFunctioninterpolate_pos_embed
(pos_embed_checkpoint, visual_encoder)
diffsynth/extensions/ImageQualityMetric/BLIP/vit.py:277
↓ 2 callersFunctioninverse_sigmoid
(x)
LHM/models/rendering/gs_renderer.py:40
↓ 2 callersMethodisOpened
(self)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/misc.py:326
↓ 2 callersMethodis_empty
(self, name)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/buffer.py:88
↓ 2 callersFunctionis_norm
Check if is one of the norms.
engine/pose_estimation/third-party/ViTPose/tests/test_backbones/test_mobilenet_v2.py:18
↓ 2 callersFunctionkeypoint_auc
Calculate the pose accuracy of PCK for each individual keypoint and the averaged accuracy across all keypoints for coordinates. Note:
engine/pose_estimation/third-party/ViTPose/mmpose/core/evaluation/top_down_eval.py:218
↓ 2 callersFunctionkeypoints_from_heatmaps3d
Get final keypoint predictions from 3d heatmaps and transform them back to the image. Note: - batch size: N - num keypoints:
engine/pose_estimation/third-party/ViTPose/mmpose/core/evaluation/top_down_eval.py:625
↓ 2 callersFunctionlets_dance_with_long_video
( unet: SDUNet, motion_modules: SDMotionModel = None, controlnet: MultiControlNetManager = None,
diffsynth/pipelines/sd_video.py:14
↓ 2 callersFunctionlimit_max_fps
(fps: Optional[float])
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/misc.py:17
↓ 2 callersFunctionlinear
Create a linear module.
LHM/models/utils.py:4
↓ 2 callersFunctionload_checkpoint
(model, checkpoint_path, strict=True)
diffsynth/extensions/ImageQualityMetric/open_clip/factory.py:98
↓ 2 callersFunctionload_checkpoint
Load checkpoint from a file or URI. Args: model (Module): Module to load checkpoint. filename (str): Accept local filepath, URL,
engine/pose_estimation/third-party/ViTPose/mmcv_custom/checkpoint.py:312
↓ 2 callersFunctionload_config
(config_name: str)
engine/pose_estimation/dinov2/dinov2/configs/__init__.py:11
↓ 2 callersFunctionload_model
(checkpoint, use_torchscript=False)
LHM/models/encoders/sapiens_warpper.py:121
↓ 2 callersFunctionload_model
Open a checkpoint, build Multi-HMR using saved arguments, load the model weigths.
engine/pose_estimation/model.py:44
↓ 2 callersFunctionload_model_from_huggingface_folder
(file_path, model_names, model_classes, torch_dtype, device)
diffsynth/models/model_manager.py:82
↓ 2 callersMethodload_obj
load an ``obj`` mesh. Args: path (str): path to mesh. albedo_path (str, optional): path to the albedo texture image,
LHM/models/rendering/mesh_utils.py:275
↓ 2 callersFunctionload_patch_model_from_single_file
(state_dict, model_names, model_classes, extra_kwargs, model_manager, torch_dtype, device)
diffsynth/models/model_manager.py:112
↓ 2 callersMethodload_trimesh
load a mesh using ``trimesh.load()``. Can load various formats like ``glb`` and serves as a fallback. Note: We will try
LHM/models/rendering/mesh_utils.py:453
↓ 2 callersMethodload_video
(self, video_file, image_folder, height, width, start_frame_id, end_frame_id)
diffsynth/pipelines/pipeline_runner.py:53
↓ 2 callersFunctionmake_2tuple
(x)
LHM/models/encoders/dinov2/layers/patch_embed.py:16
↓ 2 callersFunctionmake_2tuple
(x)
engine/pose_estimation/dinov2/dinov2/layers/patch_embed.py:16
↓ 2 callersMethodmake_activation
(self, activation)
LHM/models/rendering/utils/utils.py:86
↓ 2 callersFunctionmake_eval_data_loader
(test_dataset_str, batch_size, num_workers, metric_type)
engine/pose_estimation/dinov2/dinov2/eval/linear.py:411
↓ 2 callersMethodmake_res_layer
Make a ResLayer.
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/resnet.py:564
↓ 2 callersMethodmapper
Dictionary mapper. Renames keys according to keymap provided. Args: d (dict): old dict keymap (dict): {'old_
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/pipelines/shared_transform.py:271
↓ 2 callersMethodmask
(self, height, width, border_width)
diffsynth/models/tiler.py:10
↓ 2 callersMethodmask_to_boundary
(self, mask)
engine/BiRefNet/evaluation/metrics.py:749
↓ 2 callersMethodmatch_filenames
(self, file_names_a, file_names_b)
diffsynth/extensions/FastBlend/api.py:143
↓ 2 callersFunctionmatrix_to_axis_angle
Convert rotations given as rotation matrices to axis/angle. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Ret
engine/pose_estimation/pose_utils/rot6d.py:109
↓ 2 callersMethodmix
(self, x, smooth_scale = 0.6)
diffsynth/models/stepvideo_vae.py:1014
↓ 2 callersFunctionmkdir_or_exist
(dir_name, mode=0o777)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/misc.py:67
↓ 2 callersFunctionmodel_forward
()
LHM/models/rendering/smplx/transfer_model/transfer_model.py:71
↓ 2 callersFunctionmodulate
(x: torch.Tensor, shift: torch.Tensor, scale: torch.Tensor)
diffsynth/models/wan_video_dit.py:61
↓ 2 callersFunctionmodulate
(x, scale, shift)
diffsynth/models/stepvideo_dit.py:610
↓ 2 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/resnet.py:92
↓ 2 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/resnet.py:227
↓ 2 callersMethodnorm2
nn.Module: the normalization layer named "norm2"
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/resnet.py:97
↓ 2 callersFunctionnormalize_rgb_tensor
(img, imgenet_normalization=True)
engine/pose_estimation/pose_utils/image.py:46
↓ 2 callersMethodnumel_excluding_first_dim
(self, x)
LHM/losses/tvloss.py:30
↓ 2 callersMethodobtain_facesr
(self, head_image)
LHM/models/modeling_human_lrm.py:574
↓ 2 callersMethodon_exit
This method will be invoked on event `_exit_`. Subclasses should override this method to specifying the exiting behavior.
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/nodes/node.py:293
↓ 2 callersFunctionordered_halving
(val)
diffsynth/pipelines/wan_video_v2v_combine.py:602
↓ 2 callersFunctionordered_halving
(val)
diffsynth/pipelines/wan_video.py:775
↓ 2 callersFunctionparse_require_file
(fpath)
engine/pose_estimation/third-party/ViTPose/setup.py:83
↓ 2 callersFunctionpatch_device
(module)
diffsynth/extensions/ImageQualityMetric/open_clip/openai.py:97
↓ 2 callersFunctionpatch_float
(module)
diffsynth/extensions/ImageQualityMetric/open_clip/openai.py:121
↓ 2 callersMethodpatchify
(self, hidden_states)
diffsynth/models/flux_controlnet.py:46
↓ 2 callersFunctionpost_dark_udp
DARK post-pocessing. Implemented by udp. Paper ref: Huang et al. The Devil is in the Details: Delving into Unbiased Data Processing for Human
engine/pose_estimation/third-party/ViTPose/mmpose/core/evaluation/top_down_eval.py:335
↓ 2 callersMethodpost_process
(self)
LHM/models/ESRGANer_utils.py:233
↓ 2 callersMethodpre_process
Pre-process, such as pre-pad and mod pad, so that the images can be divisible
LHM/models/ESRGANer_utils.py:128
↓ 2 callersMethodprepare_controlnet_input
(self, controlnet_image, controlnet_inpaint_mask, tiler_kwargs)
diffsynth/pipelines/flux_image.py:218
↓ 2 callersFunctionprepare_dataloader
(dataset: torch.utils.data.Dataset, batch_size: int, to_be_distributed=False, is_train=True)
engine/BiRefNet/train.py:71
↓ 2 callersMethodprepare_extra_input
(self, latents=None)
diffsynth/pipelines/sd_image.py:86
↓ 2 callersMethodprepare_rotary_positional_embeddings
( self, height: int, width: int, num_frames: int, device: torch.device
diffsynth/models/cog_dit.py:202
↓ 2 callersMethodprocess
(self)
LHM/models/ESRGANer_utils.py:154
← previousnext →901–1,000 of 5,947, ranked by callers