MCPcopy Create free account

hub / github.com/Tencent/MimicMotion / types & classes

Types & classes12 in github.com/Tencent/MimicMotion

↓ 2 callersClassPoseNet
a tiny conv network for introducing pose sequence as the condition
mimicmotion/modules/pose_net.py:10
↓ 2 callersClassTransformerTemporalModelOutput
The output of [`TransformerTemporalModel`]. Args: sample (`torch.FloatTensor` of shape `(batch_size x num_frames, num_channels, heig
mimicmotion/modules/attention.py:15
↓ 1 callersClassDWposeDetector
A pose detect method for image-like data. Parameters: model_det: (str) serialized ONNX format model path, such
mimicmotion/dwpose/dwpose_detector.py:11
↓ 1 callersClassMimicMotionModel
mimicmotion/utils/loader.py:15
↓ 1 callersClassMimicMotionPipeline
r""" Pipeline to generate video from an input image using Stable Video Diffusion. This model inherits from [`DiffusionPipeline`]. Check the s
mimicmotion/pipelines/pipeline_mimicmotion.py:69
↓ 1 callersClassMimicMotionPipelineOutput
r""" Output class for mimicmotion pipeline. Args: frames (`[List[List[PIL.Image.Image]]`, `np.ndarray`, `torch.Tensor`]):
mimicmotion/pipelines/pipeline_mimicmotion.py:56
↓ 1 callersClassUNetSpatioTemporalConditionOutput
The output of [`UNetSpatioTemporalConditionModel`]. Args: sample (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channel
mimicmotion/modules/unet.py:19
↓ 1 callersClassWholebody
detect human pose by dwpose
mimicmotion/dwpose/wholebody.py:8
ClassPredictor
predict.py:46
ClassTransformerSpatioTemporalModel
A Transformer model for video-like data. Parameters: num_attention_heads (`int`, *optional*, defaults to 16): The number of heads to
mimicmotion/modules/attention.py:188
ClassTransformerTemporalModel
A Transformer model for video-like data. Parameters: num_attention_heads (`int`, *optional*, defaults to 16): The number of heads to
mimicmotion/modules/attention.py:27
ClassUNetSpatioTemporalConditionModel
r""" A conditional Spatio-Temporal UNet model that takes a noisy video frames, conditional state, and a timestep and returns a sample shaped o
mimicmotion/modules/unet.py:31