MCPcopy Create free account

hub / github.com/ali-vilab/VACE / functions

Functions491 in github.com/ali-vilab/VACE

⨍ Functions 491◇ Types & classes 128
↓ 116 callersMethodget
(self)
vace/gradios/vace_wan_demo.py:27
↓ 29 callersFunctionsave_one_video
(file_path, videos, fps=8, quality=8, macro_block_size=None)
vace/annotators/utils.py:98
↓ 24 callersFunction_upsample_like
upsample tensor 'src' to have the same spatial size with tensor 'tar'.
vace/annotators/salient.py:29
↓ 24 callersMethodforward
(self, x)
vace/annotators/salient.py:59
↓ 19 callersFunctionconvert_to_numpy
(image)
vace/annotators/utils.py:42
↓ 15 callersMethodforward
(self, image=None, mask=None, bbox=None, label=None, caption=None, mode=None, return_mask=None, return_source=
vace/annotators/inpainting.py:59
↓ 15 callersFunctionread_video_frames
(video_path, use_type='cv2', is_rgb=True, info=False)
vace/annotators/utils.py:186
↓ 10 callersFunctionget_activation
(name)
vace/annotators/midas/vit.py:165
↓ 10 callersMethodload
Load model from file. Args: path (str): file path
vace/annotators/midas/base_model.py:7
↓ 9 callersMethodforward
(self, process_type_1, process_type_2, frames_1, frames_2, masks_1, masks_2)
vace/annotators/composition.py:24
↓ 9 callersFunctionread_video_one_frame
(video_path, use_type='cv2', is_rgb=True)
vace/annotators/utils.py:228
↓ 8 callersMethodapply_seg_mask
(self, image, mask, mask_color, mask_cfg=None)
vace/annotators/inpainting.py:50
↓ 8 callersMethodapply_seg_mask
(self, mask_data, frames, mask_color, mask_cfg=None, return_frame=True)
vace/annotators/inpainting.py:179
↓ 7 callersMethodforward
(self, prompt, system_prompt=None, seed=-1)
vace/annotators/prompt_extend.py:28
↓ 7 callersMethodfrom_pretrained
( cls, pretrained_model_path: Optional[Union[str, os.PathLike]], *args,
vace/models/ltx/models/transformers/transformer3d.py:194
↓ 6 callersMethod__init__
(self, in_ch=3, mid_ch=12, out_ch=3)
vace/annotators/salient.py:37
↓ 6 callersMethodadd
(self, item)
vace/gradios/vace_wan_demo.py:23
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
vace/annotators/depth_anything_v2/util/transform.py:51
↓ 6 callersMethodconstrain_to_multiple_of
(self, x, min_val=0, max_val=None)
vace/annotators/midas/transforms.py:97
↓ 6 callersMethodforward
(self, image, input_box=None, mask=None, task_
vace/annotators/sam2.py:32
↓ 5 callersFunction_make_scratch
(in_shape, out_shape, groups=1, expand=False)
vace/annotators/midas/blocks.py:65
↓ 4 callersMethod__init__
Init. Args: scale_factor (float): scaling mode (str): interpolation mode
vace/annotators/midas/blocks.py:151
↓ 4 callersFunction_make_fusion_block
(features, use_bn, size=None)
vace/annotators/depth_anything_v2/dpt.py:201
↓ 4 callersFunction_make_fusion_block
(features, use_bn)
vace/annotators/midas/dpt_depth.py:11
↓ 4 callersFunction_make_vit_b16_backbone
( model, features=[96, 192, 384, 768], size=[384, 384], hooks=[2, 5, 8, 11], vit_features=
vace/annotators/midas/vit.py:189
↓ 4 callersFunctionconvert_to_pil
(image)
vace/annotators/utils.py:53
↓ 4 callersFunctiondict_to_markdown_table
(d)
vace/gradios/vace_preprocess_demo.py:28
↓ 4 callersFunctiondraw_pose
(pose, H, W, use_hand=False, use_body=False, use_face=False)
vace/annotators/pose.py:17
↓ 4 callersMethodforward
(self, image)
vace/annotators/depth.py:21
↓ 4 callersMethodforward
(self, bbox, frame_size=None, num_frames=None, label=None, color=None)
vace/annotators/layout.py:22
↓ 4 callersMethodforward
(self, mask=None, color=None, label=None, mask_cfg=None)
vace/annotators/layout.py:79
↓ 4 callersMethodforward
(self, image)
vace/annotators/pose.py:48
↓ 4 callersMethodforward
(self, image, classes=None, caption=None)
vace/annotators/gdino.py:31
↓ 4 callersFunctionget_annotator
(config_type='', config_task='', return_dict=True)
vace/annotators/utils.py:354
↓ 4 callersMethodload_video_batch
(self, *data_key_batch, crop_box=None, seed=2024, **kwargs)
vace/models/utils/preprocessor.py:227
↓ 4 callersMethodprepare_tokens_with_masks
(self, x, masks=None)
vace/annotators/depth_anything_v2/dinov2.py:211
↓ 4 callersFunctionsave_one_image
(file_path, image, use_type='cv2')
vace/annotators/utils.py:109
↓ 4 callersFunctiontid_maker
()
vace/gradios/vace_preprocess_demo.py:25
↓ 3 callersMethod__init__
(self, start_index=1)
vace/annotators/midas/vit.py:13
↓ 3 callersFunction_make_encoder
( backbone, features, use_pretrained, groups=1, expand=False, exportable=True, hoo
vace/annotators/midas/blocks.py:10
↓ 3 callersMethodforward
(self, image)
vace/annotators/gray.py:12
↓ 3 callersMethodforward
(self, image, expand_ratio=0.3, mask=None, dir
vace/annotators/outpainting.py:22
↓ 3 callersMethodget_expand_params
(self, mask_kwargs)
vace/annotators/maskaug.py:66
↓ 3 callersFunctionget_mask_box
(mask, threshold=255)
vace/annotators/utils.py:34
↓ 3 callersFunctionhalf
(x)
vace/models/wan/distributed/xdit_context_parallel.py:193
↓ 3 callersMethodrand_expand_mask
(mask, bbox, h, w, expand_ratio=None, expand_iters=None, expand_lrtp=None)
vace/annotators/maskaug.py:126
↓ 3 callersFunctionread_image
(image_path, use_type='cv2', is_rgb=True, info=False)
vace/annotators/utils.py:123
↓ 3 callersFunctionsave_sam2_video
(video_path, video_segments, output_video_path)
vace/annotators/utils.py:307
↓ 3 callersMethodtransfer_data_to_cuda
(self, data, device)
vace/models/wan/wan_vace.py:492
↓ 2 callersMethod__init__
(self, cfg, device=None)
vace/annotators/pose.py:36
↓ 2 callersMethod__init__
(self, cfg, device=None)
vace/annotators/outpainting.py:14
↓ 2 callersMethod__init__
(self, vace_layers=None, vace_in_dim=None, model_type='t2v'
vace/models/wan/modules/model.py:72
↓ 2 callersFunction_get_3rd_point
To calculate the affine matrix, three pairs of points are required. This function is used to get the 3rd point, given 2D points a & b. The 3r
vace/annotators/dwpose/onnxpose.py:184
↓ 2 callersMethodapply_plain_mask
(self, image, mask, mask_color)
vace/annotators/inpainting.py:43
↓ 2 callersMethodapply_plain_mask
(self, frames, mask, mask_color, return_frame=True)
vace/annotators/inpainting.py:166
↓ 2 callersMethoddecode_latent
(self, zs, ref_images=None, vae=None)
vace/models/wan/wan_vace.py:245
↓ 2 callersFunctiondrop_add_residual_stochastic_depth
( x: Tensor, residual_func: Callable[[Tensor], Tensor], sample_drop_ratio: float = 0.0, )
vace/annotators/depth_anything_v2/layers/block.py:110
↓ 2 callersFunctiondrop_add_residual_stochastic_depth_list
( x_list: List[Tensor], residual_func: Callable[[Tensor, Any], Tensor], sample_drop_ratio: float =
vace/annotators/depth_anything_v2/layers/block.py:181
↓ 2 callersFunctionfilter_args
(args: Dict[str, Any], parser: argparse.ArgumentParser)
vace/vace_pipeline.py:12
↓ 2 callersMethodforward
(self, mask, mask_cfg=None)
vace/annotators/maskaug.py:26
↓ 2 callersMethodforward
(self, image)
vace/annotators/scribble.py:117
↓ 2 callersMethodforward
(self, image)
vace/annotators/pose.py:125
↓ 2 callersMethodforward
r""" Forward pass through the diffusion model Args: x (List[Tensor]): List of input video tensors, each w
vace/models/wan/modules/model.py:144
↓ 2 callersMethodforward_vace
( self, x, vace_context, seq_len, kwargs )
vace/models/wan/modules/model.py:120
↓ 2 callersMethodget
(self)
vace/gradios/vace_ltx_demo.py:25
↓ 2 callersFunctionget_annotator_instance
(anno_cfg)
vace/annotators/utils.py:345
↓ 2 callersFunctionget_attn_bias_and_cat
this will perform the index select, cat the tensors, and provide the attn_bias from cache
vace/annotators/depth_anything_v2/layers/block.py:157
↓ 2 callersFunctionget_readout_oper
(vit_features, features, use_readout, start_index=1)
vace/annotators/midas/vit.py:172
↓ 2 callersMethodload_image_batch
(self, *data_key_batch, normalize=True, seq_len=None, **kwargs)
vace/models/utils/preprocessor.py:70
↓ 2 callersFunctionload_parser
(module_name: str)
vace/vace_pipeline.py:6
↓ 2 callersFunctionmake_2tuple
(x)
vace/annotators/depth_anything_v2/layers/patch_embed.py:18
↓ 2 callersMethodprepare_source
(self, src_video, src_mask, src_ref_images, num_frames, image_size, device)
vace/models/wan/wan_vace.py:194
↓ 2 callersFunctionresize_image
(input_image, resolution)
vace/annotators/utils.py:76
↓ 2 callersFunctionrope_apply
x: [B, L, N, C]. grid_sizes: [B, 3]. freqs: [M, C // 2].
vace/models/wan/distributed/xdit_context_parallel.py:27
↓ 2 callersMethodset_seq_len
(self, seq_len)
vace/models/utils/preprocessor.py:104
↓ 2 callersFunctionsingle_rle_to_mask
(rle)
vace/annotators/utils.py:21
↓ 2 callersMethodvace_encode_frames
(self, frames, ref_images, masks=None, vae=None)
vace/models/wan/wan_vace.py:127
↓ 2 callersMethodvace_encode_masks
(self, masks, ref_images=None, vae_stride=None)
vace/models/wan/wan_vace.py:157
↓ 2 callersMethodvace_latent
(self, z, m)
vace/models/wan/wan_vace.py:191
↓ 1 callersFunctionDINOv2
(model_name)
vace/annotators/depth_anything_v2/dinov2.py:397
↓ 1 callersFunctionHWC3
(x)
vace/annotators/dwpose/wholebody.py:9
↓ 1 callersMethod__init__
(self, in_features)
vace/annotators/scribble.py:14
↓ 1 callersMethod__init__
( self, in_channels, features=256, use_bn=False, o
vace/annotators/depth_anything_v2/dpt.py:87
↓ 1 callersMethod__init__
( self, in_features: int, hidden_features: Optional[int] = None, out_features:
vace/annotators/depth_anything_v2/layers/swiglu_ffn.py:15
↓ 1 callersMethod__init__
Init. Args: features (int): number of features
vace/annotators/depth_anything_v2/util/blocks.py:37
↓ 1 callersMethod__init__
( self, head, features=256, backbone='vitb_rn50_384', readout='project
vace/annotators/midas/dpt_depth.py:23
↓ 1 callersMethod__init__
(self, *args, **kwargs)
vace/models/ltx/models/transformers/attention.py:13
↓ 1 callersMethod_convexhull
(image, clockwise)
vace/annotators/maskaug.py:154
↓ 1 callersFunction_fix_aspect_ratio
Extend the scale to match the given aspect ratio. Args: scale (np.ndarray): The image scale (w, h) in shape (2, ) aspect_ratio (f
vace/annotators/dwpose/onnxpose.py:151
↓ 1 callersMethod_get_frameid_bbox
(self, fps, frame_timestamps, h, w, crop_box, rng)
vace/models/utils/preprocessor.py:215
↓ 1 callersMethod_get_frameid_bbox_adjust_last
(self, fps, frame_timestamps, h, w, crop_box, rng)
vace/models/utils/preprocessor.py:182
↓ 1 callersMethod_get_frameid_bbox_default
(self, fps, frame_timestamps, h, w, crop_box, rng)
vace/models/utils/preprocessor.py:150
↓ 1 callersMethod_get_intermediate_layers_chunked
(self, x, n=1)
vace/annotators/depth_anything_v2/dinov2.py:282
↓ 1 callersMethod_get_intermediate_layers_not_chunked
(self, x, n=1)
vace/annotators/depth_anything_v2/dinov2.py:270
↓ 1 callersMethod_image_preprocess
(self, img, oh, ow, normalize=True, **kwargs)
vace/models/utils/preprocessor.py:61
↓ 1 callersFunction_init_logging
(rank)
vace/vace_wan_inference.py:184
↓ 1 callersMethod_load_image
(self, img_path)
vace/models/utils/preprocessor.py:29
↓ 1 callersFunction_make_efficientnet_backbone
(effnet)
vace/annotators/midas/blocks.py:118
↓ 1 callersFunction_make_pretrained_efficientnet_lite3
(use_pretrained, exportable=False)
vace/annotators/midas/blocks.py:110
↓ 1 callersFunction_make_pretrained_resnext101_wsl
(use_pretrained)
vace/annotators/midas/blocks.py:142
next →1–100 of 491, ranked by callers