MCPcopy Create free account

hub / github.com/cangcz/AnchorCrafter / types & classes

Types & classes25 in github.com/cangcz/AnchorCrafter

↓ 2 callersClassAnchorCrafterModel
anchor_crafter/utils/loader.py:28
↓ 2 callersClassBody
evalution/preprocess/src/body.py:14
↓ 2 callersClassHand
evalution/preprocess/src/hand.py:15
↓ 2 callersClassIPAttnProcessor
r""" Attention processor for IP-Adapater. Args: hidden_size (`int`): The hidden size of the attention layer. cross
anchor_crafter/modules/attention_processor.py:91
↓ 2 callersClassTransformerTemporalModelOutput
The output of [`TransformerTemporalModel`]. Args: sample (`torch.FloatTensor` of shape `(batch_size x num_frames, num_channels, heig
anchor_crafter/modules/attention.py:15
↓ 2 callersClasshandpose_model
evalution/preprocess/src/model.py:143
↓ 1 callersClassAnchorCrafterPipeline
r""" Pipeline to generate video from an input image using Stable Video Diffusion. This model inherits from [`DiffusionPipeline`]. Check the s
anchor_crafter/pipelines/pipeline.py:78
↓ 1 callersClassAnchorDataset
anchor_dataset.py:248
↓ 1 callersClassAttnProcessor
r""" Default processor for performing attention-related computations.
anchor_crafter/modules/attention_processor.py:19
↓ 1 callersClassDWposeDetector
A pose detect method for image-like data. Parameters: model_det: (str) serialized ONNX format model path, such
anchor_crafter/dwpose/dwpose_detector.py:12
↓ 1 callersClassMimicMotionPipelineOutput
r""" Output class for mimicmotion pipeline. Args: frames (`[List[List[PIL.Image.Image]]`, `np.ndarray`, `torch.Tensor`]):
anchor_crafter/pipelines/pipeline.py:65
↓ 1 callersClassNet
train.py:55
↓ 1 callersClassObjAttnNet
anchor_crafter/modules/obj_attn_net.py:9
↓ 1 callersClassObjProjNet
anchor_crafter/modules/obj_proj_net.py:8
↓ 1 callersClassPoseNet
a tiny conv network for introducing pose sequence as the condition
anchor_crafter/modules/pose_hand_net.py:22
↓ 1 callersClassRegionControler
anchor_crafter/modules/attention_processor.py:14
↓ 1 callersClassTrackNet
a tiny conv network for introducing pose sequence as the conditionpose_net.py
anchor_crafter/modules/track_net.py:22
↓ 1 callersClassUNetSpatioTemporalConditionOutput
The output of [`UNetSpatioTemporalConditionModel`]. Args: sample (`torch.FloatTensor` of shape `(batch_size, num_frames, num_channel
anchor_crafter/modules/unet.py:19
↓ 1 callersClassWholebody
detect human pose by dwpose
anchor_crafter/dwpose/wholebody.py:9
↓ 1 callersClassbodypose_model
evalution/preprocess/src/model.py:24
ClassAttnProcessor2_0
r""" Processor for implementing scaled dot-product attention (enabled by default if you're using PyTorch 2.0).
anchor_crafter/modules/attention_processor.py:232
ClassIPAttnProcessor2_0
r""" Attention processor for IP-Adapater for PyTorch 2.0. Args: hidden_size (`int`): The hidden size of the attention laye
anchor_crafter/modules/attention_processor.py:319
ClassTransformerSpatioTemporalModel
A Transformer model for video-like data. Parameters: num_attention_heads (`int`, *optional*, defaults to 16): The number of heads to
anchor_crafter/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
anchor_crafter/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
anchor_crafter/modules/unet.py:31