MCPcopy Create free account

hub / github.com/cvlab-kaist/ControlFace / functions

Functions408 in github.com/cvlab-kaist/ControlFace

Methodforward
(self, src, cond)
src/models/pointemb.py:119
Methodforward
The [`Transformer2DModel`] forward method. Args: hidden_states (`torch.LongTensor` of shape `(batch size, num latent pix
src/models/transformer_2d.py:213
Methodforward
r""" The [`UNet2DConditionModel`] forward method. Args: sample (`torch.FloatTensor`): The noisy input ten
src/models/unet_2d_condition.py:871
Methodforward
( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.FloatTensor]
src/models/attention.py:178
Methodforward
( self, hidden_states, encoder_hidden_states=None, timestep=None, atte
src/models/attention.py:381
Methodforward
(self, x: torch.FloatTensor)
src/models/unet_2d_blocks.py:219
Methodforward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None )
src/models/unet_2d_blocks.py:344
Methodforward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,
src/models/unet_2d_blocks.py:442
Methodforward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,
src/models/unet_2d_blocks.py:586
Methodforward
( self, hidden_states: torch.FloatTensor, temb: Optional[torch.FloatTensor] = None,
src/models/unet_2d_blocks.py:717
Methodforward
( self, hidden_states: torch.FloatTensor, res_hidden_states_tuple: Tuple[torch.FloatTe
src/models/unet_2d_blocks.py:842
Methodforward
( self, hidden_states: torch.FloatTensor, res_hidden_states_tuple: Tuple[torch.FloatTe
src/models/unet_2d_blocks.py:984
Methodforward
(self, conditioning)
src/models/pose_guider.py:47
Methodforward
(self, x)
src/models/resnet.py:10
Methodforward
(self, x)
src/models/resnet.py:21
Methodforward
(self, hidden_states, output_size=None)
src/models/resnet.py:53
Methodforward
(self, hidden_states)
src/models/resnet.py:112
Methodforward
(self, input_tensor, temb)
src/models/resnet.py:217
Methodforward
( self, input_tensor, temb, encoder_hidden_states, attention_mask=None
src/models/motion_module.py:77
Methodforward
(self, hidden_states, encoder_hidden_states=None, attention_mask=None)
src/models/motion_module.py:146
Methodforward
( self, hidden_states, encoder_hidden_states=None, attention_mask=None,
src/models/motion_module.py:236
Methodforward
(self, x)
src/models/motion_module.py:275
Methodforward
( self, hidden_states, encoder_hidden_states=None, attention_mask=None,
src/models/motion_module.py:351
Methodforward
(self, vertices, faces, attributes=None, h=None, w=None)
decalib/utils/renderer.py:61
Methodforward
(self, vertices, faces, attributes=None, h=None, w=None, return_bary=False)
decalib/utils/renderer.py:130
Methodforward
-- Texture Rendering vertices: [batch_size, V, 3], vertices in world space, for calculating normals, then shading transformed
decalib/utils/renderer.py:222
Methodforward
(self, x)
decalib/utils/lossfunc.py:346
Methodforward
(self, gen, tar)
decalib/utils/lossfunc.py:440
Methodforward
Pytorch forward Args: x: input image (224x224) Returns: class logits
decalib/utils/lossfunc.py:515
Methodforward
(self, gen, tar)
decalib/utils/lossfunc.py:621
Methodforward
(self, gen, tar, is_crop=True)
decalib/utils/lossfunc.py:667
Methodforward
(self, inputs)
decalib/models/encoders.py:35
Methodforward
(self, noise)
decalib/models/decoders.py:52
Methodforward
(self, x)
decalib/models/frnet.py:28
Methodforward
(self, x)
decalib/models/frnet.py:62
Methodforward
(self, x)
decalib/models/frnet.py:129
Methodforward
Input: shape_params: N X number of shape parameters expression_params: N X number of expression parameter
decalib/models/FLAME.py:175
Methodforward
texcode: [batchsize, n_tex] texture: [bz, 3, 256, 256], range: 0-1
decalib/models/FLAME.py:255
Methodforward
(self, x)
decalib/models/resnet.py:64
Methodforward
(self, x)
decalib/models/resnet.py:98
Methodforward
(self, x)
decalib/models/resnet.py:138
Methodforward
(self, x)
decalib/models/resnet.py:207
Methodforward
(self, x)
decalib/models/resnet.py:221
Methodforward
(self, x1, x2)
decalib/models/resnet.py:239
Methodforward
(self, x)
decalib/models/resnet.py:259
Functiongaussian_blur
(x, kernel_size=(3,3), sigma=(0.8,0.8))
decalib/utils/util.py:373
Functiongenerate_triangles
(h, w, margin_x=2, margin_y=5, mask = None)
decalib/utils/util.py:211
Functionget_context_scheduler
(name: str)
src/pipelines/context.py:45
Functionget_fps
(video_path)
src/utils/util.py:158
Methodget_image
(self, image)
decalib/datasets/datasets.py:96
Functionget_motion_module
(in_channels, motion_module_type: str, motion_module_kwargs: dict)
src/models/motion_module.py:34
Functionget_tensor_interpolation_method
()
src/pipelines/utils.py:6
Functionget_total_steps
( scheduler, timesteps: List[int], num_steps: Optional[int] = None, num_frames: int = ...,
src/pipelines/context.py:52
Functiongradient_dif_loss
(prediction, gt)
decalib/utils/lossfunc.py:278
Methodhacked_basic_transformer_inner_forward
( self, hidden_states: torch.FloatTensor, attention_mask: Optional[torch.F
src/models/mutual_self_attention_point.py:160
Functionimport_filename
(filename)
src/utils/util.py:27
Functionkl_loss
recon_x: generating images x: origin images mu: latent mean logvar: latent log variance
decalib/utils/lossfunc.py:19
Functionl2_distance
(verts1, verts2)
decalib/utils/lossfunc.py:15
Functionlandmark_loss
(predicted_landmarks, landmarks_gt, weight=1.)
decalib/utils/lossfunc.py:145
Functionlandmark_loss_tensor
(predicted_landmarks, landmarks_gt, weight=1.)
decalib/utils/lossfunc.py:217
Functionlaplacian
(x)
decalib/utils/util.py:438
Functionlaplacian_hq_loss
(prediction, gt)
decalib/utils/lossfunc.py:325
Functionlinear
(v1, v2, t)
src/pipelines/utils.py:15
Functionlipd_loss
(predicted_landmarks, landmarks_gt, weight=1.)
decalib/utils/lossfunc.py:184
Functionload_ResNet101Model
()
decalib/models/resnet.py:178
Functionload_ResNet152Model
()
decalib/models/resnet.py:183
Functionload_ResNet50Model
()
decalib/models/resnet.py:173
Functionload_local_mask
(image_size=256, mode='bbx')
decalib/utils/util.py:663
Methodload_mask
(self, maskpath, h, w)
decalib/datasets/train_datasets.py:285
Methodload_mask
(self, maskpath, h, w)
decalib/datasets/train_datasets.py:378
Functionmedian_blur
(x, kernel_size=(3,3))
decalib/utils/util.py:390
Functionparse_args
()
decalib/utils/config.py:126
Functionprepare_anyone
()
tools/download_weights.py:61
Functionprepare_base_model
()
tools/download_weights.py:6
Methodprepare_condition
( self, cond_image, width, height, device, dtype, do_c
src/pipelines/pipeline_point.py:201
Functionprepare_image_encoder
()
tools/download_weights.py:23
Functionprepare_vae
()
tools/download_weights.py:40
Functionrad2deg
Function that converts angles from radians to degrees. See :class:`~torchgeometry.RadToDeg` for details. Args: tensor (Tensor): Tens
decalib/utils/rotation_converter.py:23
Functionremove_module
(state_dict)
decalib/utils/util.py:563
Methodrender_colors
-- rendering colors: could be rgb color/ normals, etc colors: [bz, num of vertices, 3]
decalib/utils/renderer.py:446
Methodrender_depth
-- rendering depth
decalib/utils/renderer.py:425
Functionretrieve_latents
( encoder_output: torch.Tensor, generator: Optional[torch.Generator] = None, sample_mode: str = "sample" )
src/pipelines/pipeline_point.py:45
Functionring_loss
computes ring loss for ring_outputs before FLAME decoder Inputs: ring_outputs = a list containing N streams of the ring;
decalib/utils/lossfunc.py:223
Methodrun
An api for running deca given an image path
decalib/deca.py:285
Methodrun
image: 0-255, uint8, rgb, [h, w, 3] return: detected box
decalib/datasets/detectors.py:47
Methodsave_obj
vertices: [nv, 3], tensor texture: [3, h, w], tensor
decalib/deca.py:256
Functionsave_videos_from_pil
(pil_images, path, fps=8)
src/utils/util.py:51
Functionsave_videos_grid
(videos: torch.Tensor, path: str, rescale=False, n_rows=6, fps=12)
src/utils/util.py:121
Functionseed_everything
(seed)
src/utils/util.py:16
Methodset_chunk_feed_forward
(self, chunk_size: Optional[int], dim: int = 0)
src/models/attention.py:173
Methodset_default_attn_processor
Disables custom attention processors and sets the default attention implementation.
src/models/unet_2d_condition.py:740
Functionset_tensor_interpolation_method
(is_slerp)
src/pipelines/utils.py:10
Methodset_use_memory_efficient_attention_xformers
( self, use_memory_efficient_attention_xformers: bool, attention_op: Optional[Callable
src/models/motion_module.py:309
Functionshading_smooth_loss
assume: shading should be smooth ref: Lifting AutoEncoders: Unsupervised Learning of a Fully-Disentangled 3D Morphable Model using Deep Non-R
decalib/utils/lossfunc.py:47
Functionshading_white_loss
regularize lighting: assume lights close to white
decalib/utils/lossfunc.py:37
Functionslerp
( v0: torch.Tensor, v1: torch.Tensor, t: float, DOT_THRESHOLD: float = 0.9995 )
src/pipelines/utils.py:19
Functionstandard_rasterize_colors
decalib/utils/rasterizer/standard_rasterize_cuda.cpp:32
Functiontensor2image
(tensor)
decalib/utils/util.py:538
Functiontensor_vis_landmarks
(images, landmarks, gt_landmarks=None, color = 'g', isScale=True)
decalib/utils/util.py:631
Methodtransform_points
(self, points, tform, points_scale=None, normalize = True)
decalib/utils/tensor_cropper.py:100
← previousnext →301–400 of 408, ranked by callers