MCPcopy Create free account

hub / github.com/MyNiuuu/AniCrafter / functions

Functions5,947 in github.com/MyNiuuu/AniCrafter

↓ 463 callersFunctionprint
(*args, **kwargs)
engine/pose_estimation/dinov2/dinov2/distributed/__init__.py:82
↓ 360 callersMethodto
(self,device)
diffsynth/controlnets/processors.py:42
↓ 319 callersMethodget
(self, propId)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/misc.py:335
↓ 291 callersMethodfloat
(self)
engine/SegmentAPI/base.py:28
↓ 268 callersMethoditems
(self)
LHM/models/rendering/smplx/smplx/utils.py:52
↓ 243 callersMethodcpu
(self)
engine/SegmentAPI/base.py:24
↓ 213 callersMethodto
(self, device)
diffsynth/models/model_manager.py:794
↓ 148 callersMethodevaluate
(self, results, res_folder=None, logger=None)
engine/pose_estimation/third-party/ViTPose/tests/test_backward_compatibility/test_eval_hook_compatibility.py:25
↓ 136 callersMethodpop
(self, *args, **kwargs)
engine/ouputs.py:74
↓ 132 callersMethodto
move all tensor attributes to device. Args: device (torch.device): target device. Returns: Mesh: self.
LHM/models/rendering/mesh_utils.py:690
↓ 126 callersMethodregister_buffer
(self, name, maxsize=0)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/buffer.py:72
↓ 122 callersMethodkeys
(self)
LHM/models/rendering/smplx/smplx/utils.py:44
↓ 122 callersMethodmax
(self)
engine/pose_estimation/dinov2/dinov2/logging/helpers.py:180
↓ 117 callersMethodcuda
(self)
engine/SegmentAPI/base.py:20
↓ 109 callersMethodinfo
Returns a dictionary of info about the object.
engine/pose_estimation/dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:82
↓ 108 callersMethodeval
(self)
engine/SegmentAPI/base.py:36
↓ 107 callersMethodfetch_model
(self, model_name, file_path=None, require_model_path=False)
diffsynth/models/model_manager.py:772
↓ 104 callersMethodinterpolate
(self, images, scale=1.0, batch_size=4, num_iter=1, progress_bar=lambda x:x)
diffsynth/extensions/RIFE/__init__.py:171
↓ 101 callersMethodto
(self, device)
engine/SegmentAPI/base.py:32
↓ 95 callersMethodget
(self, filepath, client_key='default')
ProPainter/utils/file_client.py:157
↓ 93 callersMethodload_models_to_device
(self, loadmodel_names=[])
diffsynth/pipelines/base.py:95
↓ 89 callersMethodclone
(self)
LHM/models/rendering/gs_renderer.py:338
↓ 86 callersMethodload
load mesh from path. Args: path (str): path to mesh file, supports ply, obj, glb. clean (bool, optional): perform mes
LHM/models/rendering/mesh_utils.py:111
↓ 77 callersMethodpad
(self, *inputs)
ProPainter/RAFT/utils/utils.py:18
↓ 65 callersFunctionbuild_norm_layer
( dim, norm_layer, in_format="channels_last", out_format="channels_last", eps=1e-6 )
engine/BiRefNet/models/modules/utils.py:15
↓ 62 callersMethoddevice
(self)
LHM/runners/infer/base_inferrer.py:46
↓ 61 callersMethodload
(self, *args, **kwargs)
engine/pose_estimation/dinov2/dinov2/fsdp/__init__.py:114
↓ 59 callersFunctionconvert_db_to_output
(db, batch_size=2, keys=None, is_3d=False)
engine/pose_estimation/third-party/ViTPose/tests/utils/data_utils.py:5
↓ 59 callersMethodto
Change the device of the data inplace. Example: >>> self = SamplingResult.random() >>> print(f'self = {self.to(None)}
engine/pose_estimation/dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:58
↓ 58 callersMethodget
(self, key, default=None)
LHM/models/rendering/smplx/smplx/utils.py:38
↓ 55 callersFunction_pcfg
(url='', hf_hub='', mean=None, std=None)
diffsynth/extensions/ImageQualityMetric/open_clip/pretrained.py:21
↓ 55 callersMethodnorm
(self)
engine/pose_estimation/dinov2/dinov2/hub/depth/decode_heads.py:453
↓ 55 callersMethodupdate
Update a new data sample.
engine/pose_estimation/third-party/ViTPose/mmpose/utils/timer.py:21
↓ 54 callersFunctionbuild_loss
Build loss.
engine/pose_estimation/third-party/ViTPose/mmpose/models/builder.py:32
↓ 52 callersMethodinfo
(self, txt)
engine/BiRefNet/utils.py:56
↓ 52 callersMethodnorm1
nn.Module: the normalization layer named "norm1"
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/hrnet.py:385
↓ 49 callersMethodnorm2
nn.Module: the normalization layer named "norm2"
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/hrnet.py:390
↓ 48 callersMethodwrite
write the mesh to a path. Args: path (str): path to write, supports ply, obj and glb.
LHM/models/rendering/mesh_utils.py:706
↓ 47 callersMethodinit_weights
Initialize the weights in backbone. Args: pretrained (str, optional): Path to pre-trained weights. Defaults to Non
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/vit.py:286
↓ 46 callersMethodcpu
(self)
diffsynth/controlnets/controlnet_unit.py:27
↓ 43 callersMethodimread
(path, idx)
ProPainter/core/utils.py:77
↓ 42 callersMethodread
(self)
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/misc.py:329
↓ 40 callersMethodtrain
Convert the model into training mode.
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/vit.py:338
↓ 37 callersMethodcheckpoint
(self)
engine/pose_estimation/dinov2/dinov2/run/eval/knn.py:28
↓ 37 callersFunctionenable_vram_management
(model: torch.nn.Module, module_map: dict, module_config: dict, max_num_param=None, overflow_module_config: di
diffsynth/vram_management/layers.py:92
↓ 37 callersMethodtile
(self, model_input, tile_size, tile_stride, tile_device, tile_dtype)
diffsynth/models/tiler.py:20
↓ 36 callersMethodnorm
(self, x)
diffsynth/models/wan_video_dit.py:103
↓ 35 callersFunctionconstant_init
(module, val, bias=0)
ProPainter/model/misc.py:11
↓ 30 callersFunctioninit_pose_model
Initialize a pose model from config file. Args: config (str or :obj:`mmcv.Config`): Config file path or the config object.
engine/pose_estimation/third-party/ViTPose/mmpose/apis/inference.py:22
↓ 30 callersMethodpreprocess_image
(self, image)
diffsynth/pipelines/base.py:30
↓ 29 callersMethod_xywh2cs
This encodes bbox(x,y,w,h) into (center, scale) Args: x, y, w, h (float): left, top, width and height padding (float)
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/base/kpt_2d_sview_rgb_vid_top_down_dataset.py:125
↓ 29 callersMethodvalues
(self)
LHM/models/rendering/smplx/smplx/utils.py:48
↓ 28 callersMethodgenerate_noise
(self, shape, seed=None, device="cpu", dtype=torch.float16)
diffsynth/pipelines/base.py:124
↓ 26 callersFunction_n2p
(w, t=True)
diffsynth/extensions/ImageQualityMetric/BLIP/vit.py:203
↓ 24 callersMethodconvert
Multiple with alpha and add beta with clip.
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/pipelines/shared_transform.py:334
↓ 23 callersMethodstep
(self, model_output, timestep, sample, to_final=False)
diffsynth/schedulers/ddim.py:67
↓ 22 callersMethodclose
(self)
engine/BiRefNet/utils.py:59
↓ 22 callersMethodfrom_pretrained
(cls, model_name)
diffsynth/prompters/omnigen_prompter.py:61
↓ 22 callersMethodfrom_pretrained
(file_path, torch_dtype=torch.bfloat16)
diffsynth/models/cog_dit.py:333
↓ 21 callersMethodcriterion
Criterion of wingloss. Note: - batch_size: N - num_keypoints: K - dimension of keypoints: D (D=2 or D=3)
engine/pose_estimation/third-party/ViTPose/mmpose/models/losses/regression_loss.py:79
↓ 21 callersMethodencode
(self, s: str, bos: bool = False, eos: bool = False)
diffsynth/prompters/kolors_prompter.py:50
↓ 21 callersMethodnorm
(self)
LHM/models/encoders/dinov2/hub/depth/decode_heads.py:453
↓ 21 callersMethodupdate
(self, **kwargs)
engine/pose_estimation/dinov2/dinov2/logging/helpers.py:26
↓ 20 callersMethodadd_noise
(self, original_samples, noise, timestep)
diffsynth/schedulers/ddim.py:87
↓ 20 callersFunctioncreate_custom_forward
(module)
diffsynth/distributed/xdit_context_parallel.py:70
↓ 20 callersMethodrandom
Args: rng (None | int | numpy.random.RandomState): seed or state. kwargs (keyword arguments): - num_p
engine/pose_estimation/dinov2/dinov2/eval/segmentation_m2f/core/box/samplers/sampling_result.py:95
↓ 20 callersMethodset_timesteps
(self, num_inference_steps, denoising_strength=1.0, **kwargs)
diffsynth/schedulers/ddim.py:41
↓ 19 callersMethod__init__
(self, in_channels )
diffsynth/models/stepvideo_vae.py:467
↓ 19 callersMethodapply
(module, name, n_power_iterations, dim, eps)
ProPainter/model/modules/spectral_norm.py:122
↓ 19 callersMethodbackward
(ctx, grad_corr)
ProPainter/RAFT/corr.py:75
↓ 19 callersMethodget_image
Get the frame image. Returns: array: The frame image.
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/utils/message.py:103
↓ 19 callersFunctioninference_top_down_pose_model
Inference a single image with a list of person bounding boxes. Note: - num_people: P - num_keypoints: K - bbox height: H
engine/pose_estimation/third-party/ViTPose/mmpose/apis/inference.py:423
↓ 18 callersMethod_sort_and_unique_bboxes
sort kpts and remove the repeated ones.
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/animal/animal_pose_dataset.py:350
↓ 18 callersMethod_write_keypoint_results
Write results into a json file.
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/datasets/mesh/mesh_h36m_dataset.py:55
↓ 18 callersFunctionbuild_dataset
Build a dataset from config dict. Args: cfg (dict): Config dict. It should at least contain the key "type". default_args (dict, o
engine/pose_estimation/third-party/ViTPose/mmpose/datasets/builder.py:62
↓ 18 callersMethodexport
(self, *, overwrite: bool)
engine/pose_estimation/dinov2/dinov2/distributed/__init__.py:219
↓ 18 callersFunctionget_expansion
Get the expansion of a residual block. The block expansion will be obtained by the following order: 1. If ``expansion`` is given, just retur
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/resnet.py:275
↓ 18 callersMethodload
Load netG (and netD).
ProPainter/core/trainer.py:207
↓ 18 callersMethodsave
Dump model and checkpointables to a file. Args: name (str): name of the file. kwargs (dict): extra arbitrary
engine/pose_estimation/dinov2/dinov2/fsdp/__init__.py:86
↓ 17 callersMethodcheck_resize_height_width
(self, height, width)
diffsynth/pipelines/base.py:20
↓ 17 callersFunctionto_tensor
( array: Union[Array, Tensor], dtype=torch.float32 )
LHM/models/rendering/smplx/smplx/utils.py:99
↓ 16 callersFunction_demo_inputs
Create a superset of inputs needed to run backbone. Args: input_shape (tuple): input batch dimensions. Default: (1, 3, 64, 64
engine/pose_estimation/third-party/ViTPose/tests/test_models/test_top_down_head.py:507
↓ 16 callersMethoddecode
Decode hand keypoint, relative root depth and hand type. Args: img_metas (list(dict)): Information about data augmentation
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/interhand_3d_head.py:443
↓ 16 callersMethodinference_model
Inference function. Returns: output (list[np.ndarray]): list of output hand keypoint heatmaps, relative root depth an
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/interhand_3d_head.py:394
↓ 16 callersFunctionload_state_dict
(file_path, torch_dtype=None)
diffsynth/models/utils.py:65
↓ 16 callersMethodstep
(self, pred: np.ndarray, gt: np.ndarray)
engine/BiRefNet/evaluation/metrics.py:151
↓ 15 callersMethod__init__
(self, dim_in: int, dim_out: int, approximate: str = "none", bias: bool = True)
diffsynth/models/stepvideo_dit.py:571
↓ 15 callersFunctionaxis_angle_to_matrix
Convert rotations given as axis/angle to rotation matrices. Args: axis_angle: Rotations given as a vector in axis angle form,
LHM/utils/rot6d.py:238
↓ 15 callersMethodload
(self, file_path="", state_dict={}, device="cuda", torch_dtype=torch.float16, **kwargs)
diffsynth/models/model_manager.py:143
↓ 15 callersMethodmatch
(self, file_path="", state_dict={})
diffsynth/models/model_manager.py:140
↓ 15 callersMethodrun
Method representing the Node's activity. This method override the standard run() method of Thread. Users should not override this met
engine/pose_estimation/third-party/ViTPose/tools/webcam/webcam_apis/nodes/node.py:300
↓ 15 callersFunctionto_tensor
( array: Union[Array, Tensor], dtype=torch.float32 )
LHM/models/rendering/smplx_gsavatar/utils.py:101
↓ 14 callersMethodget_loss
Calculate bottom-up masked mse loss. Note: - batch_size: N - num_channels: C - heatmaps height: H
engine/pose_estimation/third-party/ViTPose/mmpose/models/heads/deconv_head.py:248
↓ 14 callersMethodload
(self, model, state_dict_lora, lora_prefix, alpha=1.0, model_resource=None)
diffsynth/models/lora.py:76
↓ 14 callersFunctionmatrix_to_quaternion
Convert rotations given as rotation matrices to quaternions. Args: matrix: Rotation matrices as tensor of shape (..., 3, 3). Re
LHM/utils/rot6d.py:38
↓ 14 callersMethodnorm3
nn.Module: the normalization layer named "norm3"
engine/pose_estimation/third-party/ViTPose/mmpose/models/backbones/resnet.py:237
↓ 14 callersMethodupdate
(self, hidden_states)
diffsynth/pipelines/wan_video.py:387
↓ 13 callersMethod__init__
(self, config, layer_num)
diffsynth/extensions/ImageQualityMetric/BLIP/med.py:313
↓ 13 callersMethod__init__
(self, patch_size=(1, 2, 2), in_channels=16, embed_dim=3072)
diffsynth/models/hunyuan_video_dit.py:197
↓ 13 callersMethod__init__
(self, config: ChatGLMConfig, device=None)
diffsynth/models/kolors_text_encoder.py:725
next →1–100 of 5,947, ranked by callers