MCPcopy Create free account

hub / github.com/cangcz/AnchorCrafter / functions

Functions139 in github.com/cangcz/AnchorCrafter

⨍ Functions 139◇ Types & classes 25
Method__len__
(self)
anchor_dataset.py:299
Function_add_variant
(weights_name: str, variant: Optional[str] = None)
anchor_crafter/modules/pose_hand_net.py:13
Function_add_variant
(weights_name: str, variant: Optional[str] = None)
anchor_crafter/modules/track_net.py:13
Method_set_gradient_checkpointing
(self, module, value=False)
anchor_crafter/modules/unet.py:330
Methodattn_processors
r""" Returns: `dict` of attention processors: A dictionary containing all attention processors used in the model with
anchor_crafter/modules/unet.py:255
Methoddo_classifier_free_guidance
(self)
anchor_crafter/pipelines/pipeline.py:353
Functiondownload_image
(url)
train.py:548
Functiondraw_bodypose
(canvas, candidate, subset)
evalution/preprocess/src/util.py:42
Functiondraw_handpose
(canvas, all_hand_peaks, show_number=False)
evalution/preprocess/src/util.py:77
Functiondraw_handpose_by_opencv
(canvas, peaks, show_number=False)
evalution/preprocess/src/util.py:108
Methodenable_forward_chunking
Sets the attention processor to use [feed forward chunking](https://huggingface.co/blog/reformer#2-chunked-feed-forward-layers).
anchor_crafter/modules/unet.py:335
Functionexport_to_gif
Export a list of frames to a GIF. Args: - frames (list): List of frames (as numpy arrays or PIL Image objects). - output_gif_path (s
train.py:250
Functionexport_to_video
(video_frames, output_video_path, fps)
train.py:240
Methodfn_recursive_add_processors
( name: str, module: torch.nn.Module, processors: Dict[str, At
anchor_crafter/modules/unet.py:264
Methodfn_recursive_attn_processor
(name: str, module: torch.nn.Module, processor)
anchor_crafter/modules/unet.py:303
Methodfn_recursive_feed_forward
(module: torch.nn.Module, chunk_size: int, dim: int)
anchor_crafter/modules/unet.py:354
Functionforward
(self, hidden_states)
anchor_crafter/utils/geglu_patch.py:6
Methodforward
(self, pose_features, inp_noisy_latents, conditional_latents, timesteps, encoder_hidden_states
train.py:64
Methodforward
( self, attn, hidden_states, encoder_hidden_states=None, attention_mas
anchor_crafter/modules/attention_processor.py:30
Methodforward
( self, attn, hidden_states, encoder_hidden_states=None, attention_mas
anchor_crafter/modules/attention_processor.py:116
Methodforward
( self, attn, hidden_states, encoder_hidden_states=None, attention_mas
anchor_crafter/modules/attention_processor.py:245
Methodforward
( self, attn, hidden_states, encoder_hidden_states=None, attention_mas
anchor_crafter/modules/attention_processor.py:347
Methodforward
r""" The [`UNetSpatioTemporalConditionModel`] forward method. Args: sample (`torch.FloatTensor`): The noi
anchor_crafter/modules/unet.py:364
Methodforward
(self, x)
anchor_crafter/modules/pose_hand_net.py:72
Methodforward
(self, x)
anchor_crafter/modules/track_net.py:72
Methodforward
The [`TransformerTemporal`] forward method. Args: hidden_states (`torch.LongTensor` of shape `(batch size, num latent pi
anchor_crafter/modules/attention.py:107
Methodforward
Args: hidden_states (`torch.FloatTensor` of shape `(batch size, channel, height, width)`): Input hidden_states.
anchor_crafter/modules/attention.py:263
Methodforward
(self, embeds)
anchor_crafter/modules/obj_proj_net.py:19
Methodforward
(self, embeds)
anchor_crafter/modules/obj_attn_net.py:33
Methodforward
(self, x)
evalution/preprocess/src/model.py:114
Methodforward
(self, x)
evalution/preprocess/src/model.py:204
Methodguidance_scale
(self)
anchor_crafter/pipelines/pipeline.py:346
FunctionhandDetect
(candidate, subset, oriImg)
evalution/preprocess/src/util.py:128
Functionnpmax
(array)
evalution/preprocess/src/util.py:193
Methodnum_timesteps
(self)
anchor_crafter/pipelines/pipeline.py:359
FunctionpadRightDownCorner
(img, stride, padValue)
evalution/preprocess/src/util.py:12
Methodrelease_memory
(self)
anchor_crafter/dwpose/dwpose_detector.py:28
Methodset_default_attn_processor
Disables custom attention processors and sets the default attention implementation.
anchor_crafter/modules/unet.py:316
Functiontransfer
(model, model_weights)
evalution/preprocess/src/util.py:35
← previous101–139 of 139, ranked by callers