MCPcopy Create free account

hub / github.com/TMElyralab/MusePose / functions

Functions214 in github.com/TMElyralab/MusePose

⨍ Functions 214◇ Types & classes 48
↓ 85 callersMethodto
(self, device)
pose/script/wholebody.py:60
↓ 15 callersMethodupdate
(self, writer, dtype=torch.float16)
musepose/models/mutual_self_attention.py:302
↓ 14 callersFunctionwarpAffine_kps
(kps, M)
pose/script/util.py:145
↓ 12 callersFunctiontorch_dfs
(model: torch.nn.Module)
musepose/models/mutual_self_attention.py:12
↓ 10 callersMethodclear
(self)
musepose/models/mutual_self_attention.py:341
↓ 6 callersMethod__init__
( self, in_channels: int, prev_output_channel: int, out_channels: int,
musepose/models/unet_2d_blocks.py:963
↓ 5 callersFunctionget_motion_module
(in_channels, motion_module_type: str, motion_module_kwargs: dict)
musepose/models/motion_module.py:34
↓ 4 callersMethod__init__
( self, in_channels: int, prev_output_channel: int, out_channels: int,
musepose/models/unet_3d_blocks.py:758
↓ 4 callersMethod__init__
(self, d_model, dropout=0.0, max_len=24)
musepose/models/motion_module.py:263
↓ 4 callersMethodfrom_pretrained_2d
( cls, pretrained_model_path: PathLike, motion_module_path: PathLike, subfolde
musepose/models/unet_3d.py:578
↓ 3 callersMethodaugmentation
(self, image, transform, state=None)
musepose/dataset/dance_image.py:74
↓ 3 callersMethodaugmentation
(self, images, transform, state=None)
musepose/dataset/dance_video.py:75
↓ 3 callersFunctiondraw_pose
(pose, H, W, draw_face)
pose/script/dwpose.py:48
↓ 3 callersFunctionsave_checkpoint
(model, save_dir, prefix, ckpt_num, total_limit=None)
train_stage_1_multiGPU.py:595
↓ 2 callersFunctionHWC3
(x)
pose/script/dwpose.py:30
↓ 2 callersMethod__init__
( self, channels, use_conv=False, use_conv_transpose=False, out_channe
musepose/models/resnet.py:32
↓ 2 callersMethodforward
(self, hidden_states)
musepose/models/resnet.py:251
↓ 2 callersFunctionget_down_block
( down_block_type: str, num_layers: int, in_channels: int, out_channels: int, temb_channel
musepose/models/unet_2d_blocks.py:20
↓ 2 callersFunctionget_up_block
( up_block_type: str, num_layers: int, in_channels: int, out_channels: int, prev_output_ch
musepose/models/unet_2d_blocks.py:102
↓ 2 callersFunctionimport_filename
(filename)
musepose/utils/util.py:27
↓ 2 callersFunctionordered_halving
(val)
musepose/pipelines/context.py:7
↓ 2 callersFunctionresize_image
(input_image, resolution)
pose/script/dwpose.py:18
↓ 2 callersFunctionsave_videos_from_pil
(pil_images, path, fps)
pose/script/tool.py:51
↓ 2 callersFunctionsave_videos_grid
(videos: torch.Tensor, path: str, rescale=False, n_rows=6, fps=8)
musepose/utils/util.py:91
↓ 2 callersFunctionscale_video
(video,width,height)
test_stage_2.py:58
↓ 2 callersFunctionseed_everything
(seed)
musepose/utils/util.py:16
↓ 2 callersFunctionsize_calculate
(h, w, resolution)
pose/script/util.py:128
↓ 2 callersFunctionsmart_width
(d)
pose/script/util.py:8
↓ 2 callersFunctionzero_module
(module)
musepose/models/motion_module.py:15
↓ 1 callersMethod__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, d
musepose/models/attention.py:47
↓ 1 callersFunctionalign_img
(img, pose_ori, scales, detect_resolution, image_resolution)
pose_align.py:20
↓ 1 callersFunctioncompute_snr
Computes SNR as per https://github.com/TiankaiHang/Min-SNR-Diffusion-Training/blob/521b624bd70c67cee4bdf49225915f5945a872e3/guided_diffusion/
train_stage_1_multiGPU.py:101
↓ 1 callersFunctioncompute_snr
Computes SNR as per https://github.com/TiankaiHang/Min-SNR-Diffusion-Training/blob/521b624bd70c67cee4bdf49225915f5945a872e3/guided_diffusion/
train_stage_2_multiGPU.py:108
↓ 1 callersMethoddecode_latents
(self, latents)
musepose/pipelines/pipeline_pose2vid.py:104
↓ 1 callersMethoddecode_latents
(self, latents)
musepose/pipelines/pipeline_pose2img.py:102
↓ 1 callersMethoddecode_latents
(self, latents)
musepose/pipelines/pipeline_pose2vid_long.py:112
↓ 1 callersFunctiondraw_dwpose
(video_path, pose_path, out_path, draw_face)
draw_dwpose.py:13
↓ 1 callersFunctionget_context_scheduler
(name: str)
musepose/pipelines/context.py:45
↓ 1 callersFunctionget_fps
(video_path)
musepose/utils/util.py:128
↓ 1 callersFunctionget_tensor_interpolation_method
()
musepose/pipelines/utils.py:6
↓ 1 callersFunctionhandle_single
(ref_image_path, pose_path,seed)
test_stage_1.py:123
↓ 1 callersFunctionhandle_single
(ref_image_path,pose_video_path)
test_stage_2.py:134
↓ 1 callersMethodinterpolate_latents
( self, latents: torch.Tensor, interpolation_factor: int, device )
musepose/pipelines/pipeline_pose2vid_long.py:292
↓ 1 callersFunctionmain
()
pose_align.py:519
↓ 1 callersFunctionmain
()
test_stage_1.py:45
↓ 1 callersFunctionmain
(cfg)
train_stage_1_multiGPU.py:132
↓ 1 callersFunctionmain
()
test_stage_2.py:66
↓ 1 callersFunctionmain
(cfg)
train_stage_2_multiGPU.py:139
↓ 1 callersFunctionparse_args
()
test_stage_1.py:29
↓ 1 callersFunctionparse_args
()
test_stage_2.py:29
↓ 1 callersMethodprepare_extra_step_kwargs
(self, generator, eta)
musepose/pipelines/pipeline_pose2vid.py:119
↓ 1 callersMethodprepare_extra_step_kwargs
(self, generator, eta)
musepose/pipelines/pipeline_pose2img.py:117
↓ 1 callersMethodprepare_extra_step_kwargs
(self, generator, eta)
musepose/pipelines/pipeline_pose2vid_long.py:127
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, width, height, video
musepose/pipelines/pipeline_pose2vid.py:140
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, width, height, dtype
musepose/pipelines/pipeline_pose2img.py:138
↓ 1 callersMethodprepare_latents
( self, batch_size, num_channels_latents, width, height, video
musepose/pipelines/pipeline_pose2vid_long.py:148
↓ 1 callersFunctionprocess_batch_videos
(video_list, detector, root_dir, save_dir)
extract_dwpose_keypoints.py:33
↓ 1 callersFunctionprocess_single_video
(video_path, detector, root_dir, save_dir)
extract_dwpose_keypoints.py:14
↓ 1 callersFunctionread_frames
(video_path)
musepose/utils/util.py:111
↓ 1 callersFunctionread_frames
(video_path)
pose/script/tool.py:108
↓ 1 callersMethodregister_reference_hooks
( self, mode, do_classifier_free_guidance, attention_auto_machine_weight,
musepose/models/mutual_self_attention.py:52
↓ 1 callersFunctionrun_align_video_with_filterPose_translate_smooth
(args)
pose_align.py:248
↓ 1 callersFunctionsave_checkpoint
(model, save_dir, prefix, ckpt_num)
train_stage_2_multiGPU.py:608
↓ 1 callersFunctionsave_videos_from_pil
(pil_images, path, fps=8)
musepose/utils/util.py:51
↓ 1 callersMethodset_attention_slice
r""" Enable sliced attention computation. When this option is enabled, the attention module will split the input tensor in slices, to
musepose/models/unet_3d.py:283
↓ 1 callersMethodset_attention_slice
r""" Enable sliced attention computation. When this option is enabled, the attention module splits the input tensor in slices to comp
musepose/models/unet_2d_condition.py:761
↓ 1 callersMethodset_attn_processor
r""" Sets the attention processor to use to compute attention. Parameters: processor (`dict` of `AttentionProcessor` or o
musepose/models/unet_2d_condition.py:700
↓ 1 callersMethodto
(self, device)
pose/script/dwpose.py:76
Method__call__
( self, ref_image, pose_images, width, height, video_length,
musepose/pipelines/pipeline_pose2vid.py:285
Method__call__
( self, ref_image, pose_image, width, height, num_inference_st
musepose/pipelines/pipeline_pose2img.py:193
Method__call__
( self, ref_image, pose_images, width, height, video_length,
musepose/pipelines/pipeline_pose2vid_long.py:338
Method__call__
(self, input_image, detect_resolution=1024, image_resolution=768, output_type="pil", **kwargs)
pose/script/dwpose.py:88
Method__call__
(self, oriImg)
pose/script/wholebody.py:65
Method__getitem__
(self, index)
musepose/dataset/dance_image.py:79
Method__getitem__
(self, index)
musepose/dataset/dance_video.py:85
Method__init__
( self, reference_unet: UNet2DConditionModel, denoising_unet: UNet3DConditionModel,
train_stage_1_multiGPU.py:54
Method__init__
( self, reference_unet: UNet2DConditionModel, denoising_unet: UNet3DConditionModel,
train_stage_2_multiGPU.py:61
Method__init__
( self, img_size, img_scale=(1.0, 1.0), img_ratio=(0.9, 1.0), drop_rat
musepose/dataset/dance_image.py:13
Method__init__
( self, sample_rate, n_sample_frames, width, height, img_scale
musepose/dataset/dance_video.py:16
Method__init__
( self, vae, image_encoder, reference_unet, denoising_unet, po
musepose/pipelines/pipeline_pose2vid.py:30
Method__init__
( self, vae, image_encoder, reference_unet, denoising_unet, po
musepose/pipelines/pipeline_pose2img.py:34
Method__init__
( self, vae, image_encoder, reference_unet, denoising_unet, po
musepose/pipelines/pipeline_pose2vid_long.py:38
Method__init__
( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe
musepose/models/transformer_3d.py:31
Method__init__
( self, unet, mode="write", do_classifier_free_guidance=False, attenti
musepose/models/mutual_self_attention.py:20
Method__init__
( self, num_attention_heads: int = 16, attention_head_dim: int = 88, in_channe
musepose/models/transformer_2d.py:62
Method__init__
( self, sample_size: Optional[int] = None, in_channels: int = 4, out_channels:
musepose/models/unet_3d.py:34
Method__init__
( self, sample_size: Optional[int] = None, in_channels: int = 4, out_channels:
musepose/models/unet_2d_condition.py:160
Method__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, d
musepose/models/attention.py:299
Method__init__
(self, in_channels: int, out_channels: int, act_fn: str)
musepose/models/unet_2d_blocks.py:202
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
musepose/models/unet_2d_blocks.py:254
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
musepose/models/unet_2d_blocks.py:357
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
musepose/models/unet_2d_blocks.py:511
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
musepose/models/unet_2d_blocks.py:682
Method__init__
( self, in_channels: int, out_channels: int, prev_output_channel: int,
musepose/models/unet_2d_blocks.py:780
Method__init__
( self, in_channels: int, temb_channels: int, dropout: float = 0.0, nu
musepose/models/unet_3d_blocks.py:172
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
musepose/models/unet_3d_blocks.py:297
Method__init__
( self, in_channels: int, out_channels: int, temb_channels: int, dropo
musepose/models/unet_3d_blocks.py:477
Method__init__
( self, in_channels: int, out_channels: int, prev_output_channel: int,
musepose/models/unet_3d_blocks.py:596
Method__init__
( self, conditioning_embedding_channels: int, conditioning_channels: int = 3,
musepose/models/pose_guider.py:13
Method__init__
( self, channels, use_conv=False, out_channels=None, padding=1, name="conv" )
musepose/models/resnet.py:94
next →1–100 of 214, ranked by callers